Watay commited on
Commit
4dda852
·
verified ·
1 Parent(s): 9807adc

Update model card

Browse files
Files changed (1) hide show
  1. README.md +2 -14
README.md CHANGED
@@ -126,40 +126,28 @@ huggingface-cli download \
126
  Run from the repository root. This command generates an 81-frame video from an input image with the `1step` checkpoint on a single GPU.
127
 
128
  ```bash
129
- TORCH_COMPILE_DISABLE=1 TORCHDYNAMO_DISABLE=1 \
130
- CUDA_VISIBLE_DEVICES=0 \
131
  python aad1/inference.py \
132
  --prompt "two people scuba diving in the ocean" \
133
- --image_path "assets/examples/two people scuba diving in the ocean.jpg" \
134
  --output_path outputs/aad1_scuba_1step.mp4 \
135
  --checkpoint_path checkpoints/14b_i2v_1step_transformer/self_forcing_generator_bf16.index.json \
136
  --wan_model_dir wan_models/Wan2.1-T2V-14B \
137
  --num_frames 81 \
138
- --height 480 \
139
- --width 832 \
140
  --seed 1000 \
141
- --local_attn_size 9 \
142
- --sink_size 1 \
143
  --denoising_timestep_list 1000
144
  ```
145
 
146
  Example `2step` command:
147
 
148
  ```bash
149
- TORCH_COMPILE_DISABLE=1 TORCHDYNAMO_DISABLE=1 \
150
- CUDA_VISIBLE_DEVICES=0 \
151
  python aad1/inference.py \
152
  --prompt "two people scuba diving in the ocean" \
153
- --image_path "assets/examples/two people scuba diving in the ocean.jpg" \
154
  --output_path outputs/aad1_scuba_2step.mp4 \
155
  --checkpoint_path checkpoints/14b_i2v_2step_transformer/self_forcing_generator_bf16.index.json \
156
  --wan_model_dir wan_models/Wan2.1-T2V-14B \
157
  --num_frames 81 \
158
- --height 480 \
159
- --width 832 \
160
  --seed 1000 \
161
- --local_attn_size 9 \
162
- --sink_size 1 \
163
  --denoising_timestep_list 1000,500
164
  ```
165
 
 
126
  Run from the repository root. This command generates an 81-frame video from an input image with the `1step` checkpoint on a single GPU.
127
 
128
  ```bash
 
 
129
  python aad1/inference.py \
130
  --prompt "two people scuba diving in the ocean" \
131
+ --image_path assets/examples/scuba_diving_ocean.jpg \
132
  --output_path outputs/aad1_scuba_1step.mp4 \
133
  --checkpoint_path checkpoints/14b_i2v_1step_transformer/self_forcing_generator_bf16.index.json \
134
  --wan_model_dir wan_models/Wan2.1-T2V-14B \
135
  --num_frames 81 \
 
 
136
  --seed 1000 \
 
 
137
  --denoising_timestep_list 1000
138
  ```
139
 
140
  Example `2step` command:
141
 
142
  ```bash
 
 
143
  python aad1/inference.py \
144
  --prompt "two people scuba diving in the ocean" \
145
+ --image_path assets/examples/scuba_diving_ocean.jpg \
146
  --output_path outputs/aad1_scuba_2step.mp4 \
147
  --checkpoint_path checkpoints/14b_i2v_2step_transformer/self_forcing_generator_bf16.index.json \
148
  --wan_model_dir wan_models/Wan2.1-T2V-14B \
149
  --num_frames 81 \
 
 
150
  --seed 1000 \
 
 
151
  --denoising_timestep_list 1000,500
152
  ```
153