Use fp32 for CPU diffusion pipeline.
Browse files
app.py
CHANGED
@@ -33,8 +33,7 @@ if gpu:
|
|
33 |
pipeline.to("cuda")
|
34 |
else:
|
35 |
pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5",
|
36 |
-
custom_pipeline="stable_diffusion_mega",
|
37 |
-
revision="fp32", use_auth_token=AUTH_TOKEN)
|
38 |
|
39 |
# Huggingface Spaces have 16GB RAM and 8 CPU cores
|
40 |
# See https://huggingface.co/docs/hub/spaces-overview#hardware-resources
|
|
|
33 |
pipeline.to("cuda")
|
34 |
else:
|
35 |
pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5",
|
36 |
+
custom_pipeline="stable_diffusion_mega", use_auth_token=AUTH_TOKEN)
|
|
|
37 |
|
38 |
# Huggingface Spaces have 16GB RAM and 8 CPU cores
|
39 |
# See https://huggingface.co/docs/hub/spaces-overview#hardware-resources
|