kaytoo2022 commited on
Commit
ea8ca16
·
verified ·
1 Parent(s): 0cb6010

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -14,9 +14,12 @@ base_model = "black-forest-labs/FLUX.1-dev"
14
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
15
 
16
  lora_repo = "kaytoo2022/cara-the-cavapoo-flux"
17
- trigger_word = "Cara the Cavapoo" # Leave trigger_word blank if not used.
18
  pipe.load_lora_weights(lora_repo)
19
 
 
 
 
20
  pipe.to("cuda")
21
 
22
  MAX_SEED = 2**32-1
 
14
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
15
 
16
  lora_repo = "kaytoo2022/cara-the-cavapoo-flux"
17
+ trigger_word = "" # Leave trigger_word blank if not used.
18
  pipe.load_lora_weights(lora_repo)
19
 
20
+ lora_cat = 'kaytoo2022/mtsd-cat-flux'
21
+ pipe.load_lora_weights(lora_cat)
22
+
23
  pipe.to("cuda")
24
 
25
  MAX_SEED = 2**32-1