Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,6 @@ def load_pipeline(model_name):
|
|
| 58 |
use_safetensors=True,
|
| 59 |
add_watermarker=False,
|
| 60 |
use_auth_token=HF_TOKEN,
|
| 61 |
-
variant="fp16",
|
| 62 |
)
|
| 63 |
|
| 64 |
pipe.to(device)
|
|
|
|
| 58 |
use_safetensors=True,
|
| 59 |
add_watermarker=False,
|
| 60 |
use_auth_token=HF_TOKEN,
|
|
|
|
| 61 |
)
|
| 62 |
|
| 63 |
pipe.to(device)
|