Commit
•
07b481c
1
Parent(s):
200283f
Update app.py
Browse files
app.py
CHANGED
|
@@ -55,7 +55,6 @@ if torch.cuda.is_available():
|
|
| 55 |
vae=vae,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
use_safetensors=True,
|
| 58 |
-
variant="fp16",
|
| 59 |
)
|
| 60 |
|
| 61 |
if ENABLE_CPU_OFFLOAD:
|
|
|
|
| 55 |
vae=vae,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
use_safetensors=True,
|
|
|
|
| 58 |
)
|
| 59 |
|
| 60 |
if ENABLE_CPU_OFFLOAD:
|