diff --git a/unitree_g1_pack_camera/case1/run_world_model_interaction.sh b/unitree_g1_pack_camera/case1/run_world_model_interaction.sh index 899fae9..36b0458 100644 --- a/unitree_g1_pack_camera/case1/run_world_model_interaction.sh +++ b/unitree_g1_pack_camera/case1/run_world_model_interaction.sh @@ -2,7 +2,7 @@ res_dir="unitree_g1_pack_camera/case1" dataset="unitree_g1_pack_camera" { - time CUDA_VISIBLE_DEVICES=1 python3 scripts/evaluation/world_model_interaction.py \ + time CUDA_VISIBLE_DEVICES=0 python3 scripts/evaluation/world_model_interaction.py \ --seed 123 \ --ckpt_path ckpts/unifolm_wma_dual_mixbf16.ckpt \ --config configs/inference/world_model_interaction.yaml \ @@ -23,5 +23,7 @@ dataset="unitree_g1_pack_camera" --perframe_ae \ --diffusion_dtype bf16 \ --projector_mode bf16_full \ - --encoder_mode bf16_full + --encoder_mode bf16_full \ + --vae_dtype bf16 \ + --vae_channels_last } 2>&1 | tee "${res_dir}/output.log"