yresearch
/

swd-large-4-steps

Model card Files Files and versions

quickjkee commited on Mar 23

Commit

ad9788e

·

verified ·

1 Parent(s): de7a612

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.
                                                 torch_dtype=torch.float16,
                                                 custom_pipeline='quickjkee/swd_pipeline')
 pipe = pipe.to("cuda")
-lora_path = 'yresearch/swd-large-6-steps'
 pipe.transformer = PeftModel.from_pretrained(
     pipe.transformer,
     lora_path,

                                                 torch_dtype=torch.float16,
                                                 custom_pipeline='quickjkee/swd_pipeline')
 pipe = pipe.to("cuda")
+lora_path = 'yresearch/swd-large-4-steps'
 pipe.transformer = PeftModel.from_pretrained(
     pipe.transformer,
     lora_path,