Hugging Face's logo

Spaces:
andr290606
/
HD-test-run
Runtime error

App Files Community
andr290606 commited on
Commit
3f9bd49
·
1 Parent(s): 5a3bea7

Fix app.py

Browse files

Fix revision attribute

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from torch import autocast
4
  from diffusers import StableDiffusionPipeline
5
 
6
  #model_id = "Deltaadams/Hentai-Diffusion"
7
- pipe = StableDiffusionPipeline.from_pretrained("Deltaadams/Hentai-Diffusion", torch_type=torch.float16, revision="fp16")
8
  pipe = pipe.to("cuda")
9
  #torch.backends.cudnn.benchmark = True
10
  num_samples = 2
 
4
  from diffusers import StableDiffusionPipeline
5
 
6
  #model_id = "Deltaadams/Hentai-Diffusion"
7
+ pipe = StableDiffusionPipeline.from_pretrained("Deltaadams/Hentai-Diffusion", torch_type=torch.float16)
8
  pipe = pipe.to("cuda")
9
  #torch.backends.cudnn.benchmark = True
10
  num_samples = 2