Spaces:
Runtime error
Runtime error
File size: 951 Bytes
cbece1d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
#!/bin/bash
# Single GPU Gradio Launch Script
CUDA_VISIBLE_DEVICES=7
export NCCL_DEBUG=WARN
export NCCL_DEBUG_SUBSYS=OFF
echo "=========================================="
echo "Starting Gradio Web UI in Single-GPU mode"
echo "=========================================="
CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES torchrun \
--nproc_per_node=1 \
--master_port=29501 \
minimal_inference/gradio_app.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 \
--infer_frames 48 \
--load_lora \
--lora_path_dmd "Quark-Vision/Live-Avatar" \
--sample_steps 4 \
--sample_guide_scale 0 \
--num_clip 100 \
--num_gpus_dit 1 \
--sample_solver euler \
--single_gpu \
--ckpt_dir ckpt/Wan2.2-S2V-14B/ \
--server_port 7860 \
--server_name "0.0.0.0"
|