diff --git a/main.py b/main.py index 3a2206834..ceaa9d809 100644 --- a/main.py +++ b/main.py @@ -86,8 +86,9 @@ if __name__ == "__main__": import cuda_malloc if args.windows_standalone_build: + # TODO: Convert fix_torch to a function. try: - pass + import fix_torch # noqa: F401 except: pass