Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,8 +26,6 @@ def get_lora_sd_pipeline(
|
|
| 26 |
before_params = pipe.unet.parameters()
|
| 27 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
|
| 28 |
pipe.unet.set_adapter(adapter_name)
|
| 29 |
-
pipe.unet.load_lora_weights(os.path.join(unet_sub_dir, "adapter_model.safetensors"))
|
| 30 |
-
pipe.fuse_lora(lora_scale=0.4)
|
| 31 |
after_params = pipe.unet.parameters()
|
| 32 |
print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))
|
| 33 |
|
|
@@ -75,6 +73,7 @@ def infer(
|
|
| 75 |
lora_scale=0.5,
|
| 76 |
progress=gr.Progress(track_tqdm=True)
|
| 77 |
):
|
|
|
|
| 78 |
generator = torch.Generator(device).manual_seed(seed)
|
| 79 |
|
| 80 |
if model_id != model_id_default:
|
|
|
|
| 26 |
before_params = pipe.unet.parameters()
|
| 27 |
pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
|
| 28 |
pipe.unet.set_adapter(adapter_name)
|
|
|
|
|
|
|
| 29 |
after_params = pipe.unet.parameters()
|
| 30 |
print("Parameters changed:", any(torch.any(b != a) for b, a in zip(before_params, after_params)))
|
| 31 |
|
|
|
|
| 73 |
lora_scale=0.5,
|
| 74 |
progress=gr.Progress(track_tqdm=True)
|
| 75 |
):
|
| 76 |
+
|
| 77 |
generator = torch.Generator(device).manual_seed(seed)
|
| 78 |
|
| 79 |
if model_id != model_id_default:
|