Latte-1 / sample /sky_ddp.sh
maxin-cn's picture
Upload folder using huggingface_hub
94bafa8 verified
raw
history blame contribute delete
212 Bytes
#!/bin/bash
export CUDA_VISIBLE_DEVICES=6,7
torchrun --nnodes=1 --nproc_per_node=2 sample/sample_ddp.py \
--config ./configs/sky/sky_sample.yaml \
--ckpt ./share_ckpts/skytimelapse.pt \
--save_video_path ./test