Spaces:
Running
on
Zero
Running
on
Zero
tori29umai
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -54,9 +54,11 @@ def load_model(lora_model):
|
|
54 |
pipe.load_lora_weights(lora_dir, weight_name="tcd-animaginexl-3_1.safetensors", adapter_name="tcd-animaginexl-3_1")
|
55 |
pipe.load_lora_weights(lora_dir, weight_name="tori29umai_line.safetensors", adapter_name="tori29umai_line")
|
56 |
pipe.set_adapters(["tcd-animaginexl-3_1", "tori29umai_line"], adapter_weights=[1.0, 1.0])
|
|
|
57 |
elif lora_model == "プレーン":
|
58 |
pipe.load_lora_weights(lora_dir, weight_name="tcd-animaginexl-3_1.safetensors", adapter_name="tcd-animaginexl-3_1")
|
59 |
pipe.set_adapters(["tcd-animaginexl-3_1"], adapter_weights=[1.0])
|
|
|
60 |
|
61 |
# 現在のLoRAモデルを保存
|
62 |
current_lora_model = lora_model
|
|
|
54 |
pipe.load_lora_weights(lora_dir, weight_name="tcd-animaginexl-3_1.safetensors", adapter_name="tcd-animaginexl-3_1")
|
55 |
pipe.load_lora_weights(lora_dir, weight_name="tori29umai_line.safetensors", adapter_name="tori29umai_line")
|
56 |
pipe.set_adapters(["tcd-animaginexl-3_1", "tori29umai_line"], adapter_weights=[1.0, 1.0])
|
57 |
+
pipe.fuse_lora()
|
58 |
elif lora_model == "プレーン":
|
59 |
pipe.load_lora_weights(lora_dir, weight_name="tcd-animaginexl-3_1.safetensors", adapter_name="tcd-animaginexl-3_1")
|
60 |
pipe.set_adapters(["tcd-animaginexl-3_1"], adapter_weights=[1.0])
|
61 |
+
pipe.fuse_lora()
|
62 |
|
63 |
# 現在のLoRAモデルを保存
|
64 |
current_lora_model = lora_model
|