Skip to content

GitLab

Commit 258dbc06 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix some memory related issues.

parent 744ac944
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
comfy/model_management.py
View file @ 258dbc06
...@@ -617,7 +617,8 @@ def supports_dtype(device, dtype): #TODO ...@@ -617,7 +617,8 @@ def supports_dtype(device, dtype): #TODO
def device_supports_non_blocking(device): def device_supports_non_blocking(device):
if is_device_mps(device): if is_device_mps(device):
return False #pytorch bug? mps doesn't support non blocking 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): def cast_to_device(tensor, device, dtype, copy=False):
device_supports_cast = False device_supports_cast = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment