tori29umai commited on
Commit
774f35c
1 Parent(s): f7ce1b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="tori29umai_line.safetensors", adapter_name="tori29umai_line")
49
- pipe.set_adapters(["tori29umai_line"], adapter_weights=[1.0])
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