Spaces:
Runtime error
Runtime error
Fix revision tag.
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ class DropOutput(Callback):
|
|
| 62 |
# is_mega=True,
|
| 63 |
# is_reusable=True
|
| 64 |
# )
|
| 65 |
-
pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", custom_pipeline="stable_diffusion_mega", torch_dtype=torch.float32,
|
| 66 |
|
| 67 |
# pipeline.to("cuda")
|
| 68 |
|
|
|
|
| 62 |
# is_mega=True,
|
| 63 |
# is_reusable=True
|
| 64 |
# )
|
| 65 |
+
pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", custom_pipeline="stable_diffusion_mega", torch_dtype=torch.float32, use_auth_token=AUTH_TOKEN)
|
| 66 |
|
| 67 |
# pipeline.to("cuda")
|
| 68 |
|