Fix some memory related issues.

7 months ago · 258dbc06c3
1 changed files with 2 additions and 1 deletions
--- 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