Skip to content

Commit

Permalink
10s
Browse files Browse the repository at this point in the history
  • Loading branch information
HavenCross committed Oct 12, 2024
1 parent 3d2c788 commit dd79562
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions configs/inference/run_768_10s.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@

ckpt=/home/netzone22/mira-models/768-v1-10s.pt



HOST_GPU_NUM=1
HOST_NUM=1
CHIEF_IP=127.0.0.1
INDEX=0


torchrun --nproc_per_node=$HOST_GPU_NUM --nnodes=$HOST_NUM --master_addr=$CHIEF_IP --master_port=12594 --node_rank=$INDEX \
mira/scripts/evaluation/ddp_wrapper.py \
--module 'inference' \
--seed 2000 \
--ckpt_path $ckpt \
--base "./configs/Mira/config_768v1_10s_mira.yaml" \
--savedir ./Saved_Text2Video-Test/test-mira-768-10s \
--n_samples 1 \
--bs 1 \
--height 480 \
--width 768 \
--unconditional_guidance_scale 12 \
--ddim_steps 50 --ddim_eta 1.0 \
--prompt_file ./prompts/test_prompt.txt

0 comments on commit dd79562

Please sign in to comment.