prithivMLmods commited on
Commit
cb944f6
·
verified ·
1 Parent(s): f64f4f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ if not torch.cuda.is_available():
26
  base_model = "black-forest-labs/FLUX.1-dev"
27
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
28
 
29
- lora_repo = "strangerzonehf/Flux-Enrich-Art-LoRA"
30
- trigger_word = "enrich art" # Leave trigger_word blank if not used.
31
 
32
  pipe.load_lora_weights(lora_repo)
33
  pipe.to("cuda")
 
26
  base_model = "black-forest-labs/FLUX.1-dev"
27
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
28
 
29
+ lora_repo = "strangerzonehf/Flux-St-Shot"
30
+ trigger_word = "st portrait" # Leave trigger_word blank if not used.
31
 
32
  pipe.load_lora_weights(lora_repo)
33
  pipe.to("cuda")