Files
LiveAvatar/infinite_inference_single_gpu.sh
T
2025-12-04 15:20:01 +08:00

26 lines
1.3 KiB
Bash

CUDA_VISIBLE_DEVICES=3
export NCCL_DEBUG=WARN
export NCCL_DEBUG_SUBSYS=OFF
CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES /primus_xpfs_workspace_T04/huangshijie/miniconda3/envs/wan_s2v/bin/torchrun --nproc_per_node=1 --master_port=29101 minimal_inference/s2v_streaming_interact.py \
--ulysses_size 1 \
--task s2v-14B \
--size "704*384" \
--base_seed 420 \
--training_config liveavatar/configs/s2v_causal_sft.yaml \
--offload_model True \
--convert_model_dtype \
--prompt "A stout, cheerful dwarf with a magnificent braided beard adorned with metal rings, wearing a heavy leather apron. He's standing in his fiery, cluttered forge, laughing heartily as he explains the mastery of his craft, holding up a glowing hammer. Style of Blizzard Entertainment cinematics (like World of Warcraft), warm, dynamic lighting from the forge." \
--image "examples/dwarven_blacksmith.jpg" \
--audio "examples/dwarven_blacksmith.wav" \
--infer_frames 48 \
--load_lora \
--lora_path_dmd "/primus_xpfs_workspace_T04/huangyubo/LongLive/1027_s2v_selfforcing_1_13_2/ckptcopy/step_2500_generator_lora.pt" \
--sample_steps 4 \
--sample_guide_scale 0 \
--num_clip 10 \
--num_gpus_dit 1 \
--sample_solver euler \
--single_gpu \
--ckpt_dir ckpt/Wan2.2-S2V-14B/