Anurag Bhardwaj commited on
Commit
c34f45c
·
verified ·
1 Parent(s): b994d40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def load_pipeline():
17
  base_model = "black-forest-labs/FLUX.1-dev"
18
  pipe = DiffusionPipeline.from_pretrained(
19
  base_model,
20
- torch_dtype=torch.bfloat16
21
  )
22
 
23
  # Load LoRA weights
 
17
  base_model = "black-forest-labs/FLUX.1-dev"
18
  pipe = DiffusionPipeline.from_pretrained(
19
  base_model,
20
+ torch_dtype=torch.float32
21
  )
22
 
23
  # Load LoRA weights