Spaces:
Running
on
Zero
Running
on
Zero
tori29umai
commited on
Commit
•
774f35c
1
Parent(s):
f7ce1b7
Update app.py
Browse files
app.py
CHANGED
@@ -45,8 +45,8 @@ def load_model(lora_model):
|
|
45 |
|
46 |
# LoRAモデルの設定
|
47 |
if lora_model == "少女漫画風":
|
48 |
-
pipe.load_lora_weights(lora_dir, weight_name="
|
49 |
-
pipe.set_adapters(["
|
50 |
elif lora_model == "プレーン":
|
51 |
pass # プレーンの場合はLoRAを読み込まない
|
52 |
|
|
|
45 |
|
46 |
# LoRAモデルの設定
|
47 |
if lora_model == "少女漫画風":
|
48 |
+
pipe.load_lora_weights(lora_dir, weight_name="syoujomannga_line.safetensors", adapter_name="syoujomannga_line")
|
49 |
+
pipe.set_adapters(["syoujomannga_line"], adapter_weights=[1.0])
|
50 |
elif lora_model == "プレーン":
|
51 |
pass # プレーンの場合はLoRAを読み込まない
|
52 |
|