kaytoo2022 commited on
Commit
89097fd
·
verified ·
1 Parent(s): 9502eb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,11 +15,11 @@ 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": 0.9})
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, 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