Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ transformer = FluxTransformer2DModel.from_pretrained(
|
|
21 |
print("Start loading LoRA weights")
|
22 |
state_dict, network_alphas = FluxFillPipeline.lora_state_dict(
|
23 |
pretrained_model_name_or_path_or_dict="blanchon/FluxFillFurniture",
|
24 |
-
weight_name="
|
25 |
return_alphas=True
|
26 |
)
|
27 |
is_correct_format = all("lora" in key or "dora_scale" in key for key in state_dict.keys())
|
|
|
21 |
print("Start loading LoRA weights")
|
22 |
state_dict, network_alphas = FluxFillPipeline.lora_state_dict(
|
23 |
pretrained_model_name_or_path_or_dict="blanchon/FluxFillFurniture",
|
24 |
+
weight_name="pytorch_lora_weights3.safetensors",
|
25 |
return_alphas=True
|
26 |
)
|
27 |
is_correct_format = all("lora" in key or "dora_scale" in key for key in state_dict.keys())
|