hakurei
/

waifu-diffusion

StableDiffusionPipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

hakurei commited on Oct 10, 2022

Commit

cc452ee

•

1 Parent(s): 2dff9da

i need more sleep

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -49,7 +49,7 @@ from torch import autocast
 from diffusers import StableDiffusionPipeline
 pipe = StableDiffusionPipeline.from_pretrained(
-    'waifu-diffusion',
     torch_dtype=torch.float32
 ).to('cuda')

 from diffusers import StableDiffusionPipeline
 pipe = StableDiffusionPipeline.from_pretrained(
+    'hakurei/waifu-diffusion',
     torch_dtype=torch.float32
 ).to('cuda')