gstaff commited on
Commit
82ebf55
·
1 Parent(s): 2989ea1

Fix revision tag.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, revision="fp32", use_auth_token=AUTH_TOKEN)
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