Fix regression.

comfyanonymous · Aug 9, 2024 · 6678d5c · 6678d5c
1 parent e172564
commit 6678d5c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py
@@ -396,9 +396,10 @@ def lowvram_load(self, device_to=None, lowvram_model_memory=0, force_patch_weigh
  if hasattr(m, "weight"):
  mem_counter += comfy.model_management.module_size(m)
  param = list(m.parameters())
- weight = param[0]
- if weight.device == device_to:
- continue
+ if len(param) > 0:
+ weight = param[0]
+ if weight.device == device_to:
+ continue
 
  self.patch_weight_to_device(weight_key) #TODO: speed this up without OOM
  self.patch_weight_to_device(bias_key)