ooutlierr
/

fuse-dit

FuseDiTPipeline

Model card Files Files and versions Community

ooutlierr commited on 28 days ago

Commit

871beb8

·

verified ·

1 Parent(s): 0c27423

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -12,7 +12,7 @@ You can use `FuseDiTPipeline` in our codebase to run the model:
 ```python
 import torch
 from diffusion.pipelines import FuseDiTPipeline
-pipeline = FuseDiTPipeline.from_pretrained("ooutlierr/fusedit").to("cuda")
 image = pipeline(
     "your prompt",
     width=512,

 ```python
 import torch
 from diffusion.pipelines import FuseDiTPipeline
+pipeline = FuseDiTPipeline.from_pretrained("ooutlierr/fuse-dit").to("cuda")
 image = pipeline(
     "your prompt",
     width=512,