Spaces:
Sleeping
Sleeping
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
|