prithivMLmods commited on
Commit
0ce8ff5
1 Parent(s): 9f675c8

Update app.py

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