kaytoo2022 commited on
Commit
8ac5d04
·
verified ·
1 Parent(s): 89097fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -15,11 +15,13 @@ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
15
 
16
  trigger_word = "" # Leave trigger_word blank if not used.
17
 
18
- # lora_ghibli = 'alvarobartt/ghibli-characters-flux-lora'
19
- # pipe.load_lora_weights(lora_ghibli, cross_attention_kwargs={"scale": 0.1})
20
 
21
  lora_repo = "kaytoo2022/mtsd-cat-flux"
22
- pipe.load_lora_weights(lora_repo, cross_attention_kwargs={"scale": 1.0})
 
 
23
 
24
  pipe.to("cuda")
25
 
 
15
 
16
  trigger_word = "" # Leave trigger_word blank if not used.
17
 
18
+ lora_ghibli = 'alvarobartt/ghibli-characters-flux-lora'
19
+ pipe.load_lora_weights(lora_ghibli, adapter_name="ghibli")
20
 
21
  lora_repo = "kaytoo2022/mtsd-cat-flux"
22
+ pipe.load_lora_weights(lora_repo, adapter_name="mtsd")
23
+
24
+ pipe.set_adapters(["ghibli", "mtsd"], adapter_weights=[0.7, 0.8])
25
 
26
  pipe.to("cuda")
27