Hugging Face's logo

Spaces:
hakurei
/
waifu-diffusion-demo
Runtime error

App Files Community
30

Bump to v1-3

#9
by apolinario - opened
base: refs/heads/main
←
from: refs/pr/9
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  from torch import autocast
4
  from diffusers import StableDiffusionPipeline
5
 
6
- model_id = "hakurei/waifu-diffusion"
7
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision='fp16')
8
  pipe = pipe.to("cuda")
9
  torch.backends.cudnn.benchmark = True
 
3
  from torch import autocast
4
  from diffusers import StableDiffusionPipeline
5
 
6
+ model_id = "hakurei/waifu-diffusion-v1-3"
7
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision='fp16')
8
  pipe = pipe.to("cuda")
9
  torch.backends.cudnn.benchmark = True