waifu-research-department
/

long-prompt-weighting-pipeline

stable-diffusion

Model card Files Files and versions Community

skytnt commited on Oct 17, 2022

Commit

bff2f6d

•

1 Parent(s): 6ef7cee

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ import torch
 pipe = DiffusionPipeline.from_pretrained(
     'hakurei/waifu-diffusion',
-    custom_pipeline="waifu-research-department/stable-diffusion-long-prompt-weighting-pipeline",
     revision="fp16",
     torch_dtype=torch.float16
 )

 pipe = DiffusionPipeline.from_pretrained(
     'hakurei/waifu-diffusion',
+    custom_pipeline="waifu-research-department/long-prompt-weighting-pipeline",
     revision="fp16",
     torch_dtype=torch.float16
 )