Update README.md
Browse files
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
|
11 |
|
12 |
```python
|
13 |
import torch
|
14 |
from diffusion.pipelines import FuseDiTPipeline
|
15 |
-
pipeline = FuseDiTPipeline.from_pretrained("
|
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,
|