ooutlierr commited on
Commit
fb40f98
·
verified ·
1 Parent(s): 75c8eea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -7,12 +7,12 @@ license: apache-2.0
7
  - [GitHub: Code](https://github.com/tang-bd/fuse-dit)
8
 
9
  ## Quick Start
10
- You can use `FuseDiTPipeline` in our codebase to run the model:
11
 
12
  ```python
13
  import torch
14
  from diffusion.pipelines import FuseDiTPipeline
15
- pipeline = FuseDiTPipeline.from_pretrained("ooutlierr/fuse-dit").to("cuda")
16
  image = pipeline(
17
  "your prompt",
18
  width=512,
 
7
  - [GitHub: Code](https://github.com/tang-bd/fuse-dit)
8
 
9
  ## Quick Start
10
+ You can download the pre-trained model and then use `FuseDiTPipeline` in our codebase to run inference:
11
 
12
  ```python
13
  import torch
14
  from diffusion.pipelines import FuseDiTPipeline
15
+ pipeline = FuseDiTPipeline.from_pretrained("/path/to/pipeline/").to("cuda")
16
  image = pipeline(
17
  "your prompt",
18
  width=512,