Fix regression.

7 months ago · 6678d5cf65
parent e172564eea
commit 6678d5cf65
1 changed files with 4 additions and 3 deletions
--- a/comfy/model_patcher.py
+++ b/comfy/model_patcher.py
@ -396,9 +396,10 @@ class ModelPatcher:
                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)