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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -13,15 +13,15 @@ torch.backends.cuda.matmul.allow_tf32 = True
13
  base_model = "black-forest-labs/FLUX.1-dev"
14
  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, 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
 
 
13
  base_model = "black-forest-labs/FLUX.1-dev"
14
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
15
 
16
+ trigger_word = "mtsd cat" # 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