diff --git a/comfy/weight_adapter/lora.py b/comfy/weight_adapter/lora.py index 9e00e9e6..89f56a45 100644 --- a/comfy/weight_adapter/lora.py +++ b/comfy/weight_adapter/lora.py @@ -90,7 +90,7 @@ class LoRAAdapter(WeightAdapterBase): intermediate_dtype=torch.float32, original_weight=None, ): - v = self.weights[1] + v = self.weights mat1 = comfy.model_management.cast_to_device( v[0], weight.device, intermediate_dtype )