wehub-resource-sync
52 lines
998 B
YAML
52 lines
998 B
YAML
# Ulysses sequence-parallel inference config for Wan2.2-TI2V-5B.
|
|
#
|
|
# Example launch:
|
|
# torchrun --nproc_per_node=4 inference_sp.py --config_path configs/inference_sp.yaml
|
|
|
|
model_kwargs:
|
|
model_name: Wan2.2-TI2V-5B
|
|
timestep_shift: 5.0
|
|
num_frame_per_block: 8
|
|
local_attn_size: 32
|
|
|
|
sp_size: 4
|
|
dp_size: 1
|
|
auto_sp_remainder: false
|
|
model_num_heads: 24
|
|
|
|
use_ema: false
|
|
output_folder: videos/longlive2_sp
|
|
num_samples: 1
|
|
save_latents_only: false
|
|
|
|
data:
|
|
data_path: /path/to/longlive2/inference_prompts
|
|
image_or_video_shape:
|
|
- 1
|
|
- 128
|
|
- 48
|
|
- 44
|
|
- 80
|
|
|
|
inference:
|
|
sampling_steps: 4
|
|
sink_size: 8
|
|
guidance_scale: 1.0
|
|
multi_shot_sink: true
|
|
multi_shot_rope_offset: 8
|
|
|
|
checkpoints:
|
|
generator_ckpt: /path/to/longlive2/generator_checkpoint.pt
|
|
lora_ckpt: /path/to/longlive2/lora_checkpoint.pt
|
|
|
|
# Presence of this section enables LoRA inference; remove it for full-generator inference.
|
|
adapter:
|
|
type: lora
|
|
rank: 128
|
|
alpha: 128
|
|
dropout: 0.0
|
|
verbose: true
|
|
|
|
logging:
|
|
seed: 0
|