prithivMLmods commited on
Commit
6ae9433
1 Parent(s): 82e6520

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -37,11 +37,8 @@ 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 = "strangerzonehf/Flux-Midjourney-Mix-LoRA"
41
- trigger_word = "midjourney mix" # Leave trigger_word blank if not used.
42
-
43
- #lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
44
- #trigger_word = "Super Realism" # Leave trigger_word blank if not used.
45
 
46
  pipe.load_lora_weights(lora_repo)
47
  pipe.to("cuda")
 
37
  base_model = "black-forest-labs/FLUX.1-dev"
38
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
39
 
40
+ lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
41
+ trigger_word = "Super Realism" # Leave trigger_word blank if not used.
 
 
 
42
 
43
  pipe.load_lora_weights(lora_repo)
44
  pipe.to("cuda")