Fix some memory related issues.

jn-jairo · Apr 14, 2024 · 258dbc0 · 258dbc0
1 parent 744ac94
commit 258dbc0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -617,7 +617,8 @@ def supports_dtype(device, dtype): #TODO
 def device_supports_non_blocking(device):
     if is_device_mps(device):
         return False #pytorch bug? mps doesn't support non blocking
-    return True
+    return False
+    # return True #TODO: figure out why this causes issues
 
 def cast_to_device(tensor, device, dtype, copy=False):
     device_supports_cast = False