prithivMLmods commited on
Commit
bf46246
1 Parent(s): ed7e269

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ dtype = torch.bfloat16
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14
  # Load the base model pipeline
15
- pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
16
 
17
  # Load the Flux Realism LoRA model
18
  lora_repo = "XLabs-AI/flux-RealismLora"
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
 
14
  # Load the base model pipeline
15
+ pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype).to(device)
16
 
17
  # Load the Flux Realism LoRA model
18
  lora_repo = "XLabs-AI/flux-RealismLora"