hakurei commited on
Commit
fabf1b4
1 Parent(s): 04dfd3f

please work, im begging you

Browse files
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 = "hakurei/waifu-diffusion"
7
- pipe = StableDiffusionPipeline.from_pretrained("hakurei/waifu-diffusion")
8
  pipe = pipe.to("cuda")
9
  #torch.backends.cudnn.benchmark = True
10
  num_samples = 2
 
4
  from diffusers import StableDiffusionPipeline
5
 
6
  #model_id = "hakurei/waifu-diffusion"
7
+ pipe = StableDiffusionPipeline.from_pretrained("hakurei/waifu-diffusion", torch_type=torch.float16, revision="fp16")
8
  pipe = pipe.to("cuda")
9
  #torch.backends.cudnn.benchmark = True
10
  num_samples = 2