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')
|