gstaff commited on
Commit
411cb25
1 Parent(s): ad18cb1

Use fp32 for CPU diffusion pipeline.

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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", torch_dtype=torch.float32,
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