tori29umai commited on
Commit
0a4dd2d
·
verified ·
1 Parent(s): 0c98b49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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