Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|
116 |
-
|
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')
|