Spaces:

fffiloni
/

B-LoRa-Inference

Sleeping

fffiloni commited on May 29, 2024

Commit

0adf250

verified ·

1 Parent(s): 67d94fb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def load_b_lora_to_unet(pipe, content_lora_model_id: str = '', style_lora_model_
         try:
             # Get Content B-LoRA SD
             if content_lora_model_id:
-                content_B_LoRA_sd, _ = pipe.lora_state_dict(content_lora_model_id)
                 content_B_LoRA = filter_lora(content_B_LoRA_sd, BLOCKS['content'])
                 content_B_LoRA = scale_lora(content_B_LoRA, content_alpha)
             else:
@@ -30,7 +30,7 @@ def load_b_lora_to_unet(pipe, content_lora_model_id: str = '', style_lora_model_
             # Get Style B-LoRA SD
             if style_lora_model_id:
-                style_B_LoRA_sd, _ = pipe.lora_state_dict(style_lora_model_id)
                 style_B_LoRA = filter_lora(style_B_LoRA_sd, BLOCKS['style'])
                 style_B_LoRA = scale_lora(style_B_LoRA, style_alpha)
             else:

         try:
             # Get Content B-LoRA SD
             if content_lora_model_id:
+                content_B_LoRA_sd, _ = pipe.lora_state_dict(content_lora_model_id, use_auth_token=True)
                 content_B_LoRA = filter_lora(content_B_LoRA_sd, BLOCKS['content'])
                 content_B_LoRA = scale_lora(content_B_LoRA, content_alpha)
             else:
             # Get Style B-LoRA SD
             if style_lora_model_id:
+                style_B_LoRA_sd, _ = pipe.lora_state_dict(style_lora_model_id, use_auth_token=True)
                 style_B_LoRA = filter_lora(style_B_LoRA_sd, BLOCKS['style'])
                 style_B_LoRA = scale_lora(style_B_LoRA, style_alpha)
             else: