Files
unifolm-world-model-action/scripts/run_base_model_inference.sh
2025-09-12 21:53:41 +08:00

24 lines
595 B
Bash

#!/bin/bash
model_name=base_model
ckpt=/path/to/base/model
config=configs/inference/base_model_inference.yaml
res_dir="/path/to/result/directory"
seed=123
CUDA_VISIBLE_DEVICES=0 python3 scripts/evaluation/base_model_inference.py \
--seed ${seed} \
--ckpt_path $ckpt \
--config $config \
--savedir "${res_dir}/videos" \
--bs 1 --height 320 --width 512 \
--unconditional_guidance_scale 1.0 \
--ddim_steps 16 \
--ddim_eta 1.0 \
--prompt_dir "/path/to/examples/base_model_prompts" \
--text_input \
--video_length 16 \
--timestep_spacing 'uniform_trailing' \
--guidance_rescale 0.7 \
--perframe_ae