From 10e08b05544652700a9ca38a710e236fffec2abf Mon Sep 17 00:00:00 2001 From: huchenlei Date: Sun, 8 Dec 2024 20:58:28 -0500 Subject: [PATCH] revert fix_torch import --- main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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