apolinario commited on
Commit
b243c18
1 Parent(s): 7d40bb0

Bump to v1-3

Browse files
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