Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -131,6 +131,7 @@ def load_and_prepare_model():
|
|
| 131 |
#torch_dtype=torch.bfloat16,
|
| 132 |
add_watermarker=False,
|
| 133 |
low_cpu_mem_usage = False,
|
|
|
|
| 134 |
)
|
| 135 |
#pipe.vae = vaeRV #.to(torch.bfloat16)
|
| 136 |
pipe.scheduler = sched
|
|
|
|
| 131 |
#torch_dtype=torch.bfloat16,
|
| 132 |
add_watermarker=False,
|
| 133 |
low_cpu_mem_usage = False,
|
| 134 |
+
token=HF_TOKEN,
|
| 135 |
)
|
| 136 |
#pipe.vae = vaeRV #.to(torch.bfloat16)
|
| 137 |
pipe.scheduler = sched
|