ford442 commited on
Commit
912ccea
·
1 Parent(s): dd83fff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -112,9 +112,9 @@ def load_and_prepare_model(model_id):
112
  torch_dtype=torch.bfloat16,
113
  use_safetensors=True,
114
  add_watermarker=False,
115
- vae=vae,variant="bf16",
116
- torch_dtype=torch.bfloat16,
117
- safety_checker=None
118
  )
119
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
120
  pipe.to('cuda')
 
112
  torch_dtype=torch.bfloat16,
113
  use_safetensors=True,
114
  add_watermarker=False,
115
+ vae=vae,
116
+ variant="bf16",
117
+ safety_checker=None,
118
  )
119
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
120
  pipe.to('cuda')