Spaces:

hakurei
/

waifu-diffusion-demo

Runtime error

hakurei commited on Oct 8, 2022

Commit

9baeee9

•

1 Parent(s): 06987b8

oops

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from torch import autocast
 from diffusers import StableDiffusionPipeline
 model_id = "hakurei/waifu-diffusion"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 torch.backends.cudnn.benchmark = True
 num_samples = 2

 from diffusers import StableDiffusionPipeline
 model_id = "hakurei/waifu-diffusion"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision='fp16')
 pipe = pipe.to("cuda")
 torch.backends.cudnn.benchmark = True
 num_samples = 2