prithivMLmods commited on
Commit
19ec9bc
·
verified ·
1 Parent(s): dfc556c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -37,10 +37,15 @@ if not torch.cuda.is_available():
37
  base_model = "black-forest-labs/FLUX.1-dev"
38
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
39
 
40
- lora_repo = "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
41
- trigger_word = "Ultra realistic" # Leave trigger_word blank if not used.
 
42
  pipe.load_lora_weights(lora_repo)
43
 
 
 
 
 
44
  pipe.to("cuda")
45
 
46
  style_list = [
 
37
  base_model = "black-forest-labs/FLUX.1-dev"
38
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
39
 
40
+
41
+ lora_repo = "prithivMLmods/Red-Undersea-Flux-LoRA"
42
+ trigger_word = "Red Undersea" # Leave trigger_word blank if not used.
43
  pipe.load_lora_weights(lora_repo)
44
 
45
+ #lora_repo = "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
46
+ #trigger_word = "Ultra realistic" # Leave trigger_word blank if not used.
47
+ #pipe.load_lora_weights(lora_repo)
48
+
49
  pipe.to("cuda")
50
 
51
  style_list = [