move expansion to managed device (#364)

* move expansion to managed device * move expansion to managed device * move expansion to managed device * move expansion to managed device * move expansion to managed device * move expansion to managed device
2026-08-16 13:13:16 +02:00 · 2023-09-13 12:48:27 -07:00
parent 53beede21d
commit e32f04da34
4 changed files with 17 additions and 3 deletions
@@ -89,7 +89,8 @@ def download_models():
 def clear_comfy_args():
    argv = sys.argv
    sys.argv = [sys.argv[0]]
-    import comfy.cli_args
+    from comfy.cli_args import args as comfy_args
+    comfy_args.disable_cuda_malloc = True
    sys.argv = argv