CUDA_VISIBLE_DEVICES=0 xvfb-run python run_agent/run_agent.py \
    --in_model ../data/weights/vpt/2x.model \
    --in_weights ../data/weights/steve1/steve1.weights \
    --prior_weights ../data/weights/steve1/steve1_prior.pt \
    --text_cond_scale 6.0 \
    --visual_cond_scale 7.0 \
    --gameplay_length 1000 \
    --save_dirpath ../data/generated_videos/custom_text_prompt \
    --custom_text_prompt "look at the sky"