Update README.md
Browse files
README.md
CHANGED
|
@@ -525,10 +525,14 @@ torchrun --nproc_per_node=8 generate.py --task vace-14B --size 1280*720 --ckpt_d
|
|
| 525 |
```
|
| 526 |
|
| 527 |
##### (3) Running local gradio
|
|
|
|
|
|
|
|
|
|
|
|
|
| 528 |
|
|
|
|
| 529 |
```sh
|
| 530 |
-
|
| 531 |
-
python vace_1.3B_singleGPU.py --ckpt_dir_720p ./Wan2.1-VACE-1.3B # use 480P model in gradio
|
| 532 |
```
|
| 533 |
|
| 534 |
#### Run Text-to-Image Generation
|
|
|
|
| 525 |
```
|
| 526 |
|
| 527 |
##### (3) Running local gradio
|
| 528 |
+
- Single-GPU inference
|
| 529 |
+
```sh
|
| 530 |
+
python gradio/vace.py --ckpt_dir ./Wan2.1-VACE-1.3B
|
| 531 |
+
```
|
| 532 |
|
| 533 |
+
- Multi-GPU inference using FSDP + xDiT USP
|
| 534 |
```sh
|
| 535 |
+
python gradio/vace.py --mp --ulysses_size 8 --ckpt_dir ./Wan2.1-VACE-14B/
|
|
|
|
| 536 |
```
|
| 537 |
|
| 538 |
#### Run Text-to-Image Generation
|