# 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