John6666 commited on
Commit
64f7574
1 Parent(s): d34ac72

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ with open('loras.json', 'r') as f:
14
  loras = json.load(f)
15
 
16
  # Initialize the base model
17
- models = ["camenduru/FLUX.1-dev-diffusers", "black-forest-labs/FLUX.1-schnell"]
 
18
  base_model = models[0]
19
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
20
 
 
14
  loras = json.load(f)
15
 
16
  # Initialize the base model
17
+ models = ["camenduru/FLUX.1-dev-diffusers", "black-forest-labs/FLUX.1-schnell",
18
+ "sayakpaul/FLUX.1-merged", "John6666/blue-pencil-flux1-v001-fp8-flux"]
19
  base_model = models[0]
20
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
21