把混和精度模型权重导出至本地文件,减少dtype开销

--export_casted_ckpt ckpts/unifolm_wma_dual_mixbf16.ckpt \
        --export_only
This commit is contained in:
2026-01-19 15:14:01 +08:00
parent cb334f308b
commit 7e501b17fd
20 changed files with 245 additions and 55 deletions

View File

@@ -4,7 +4,7 @@ dataset="unitree_g1_pack_camera"
{
time CUDA_VISIBLE_DEVICES=1 python3 scripts/evaluation/world_model_interaction.py \
--seed 123 \
--ckpt_path ckpts/unifolm_wma_dual.ckpt \
--ckpt_path ckpts/unifolm_wma_dual_mixbf16.ckpt \
--config configs/inference/world_model_interaction.yaml \
--savedir "${res_dir}/output" \
--bs 1 --height 320 --width 512 \
@@ -22,7 +22,6 @@ dataset="unitree_g1_pack_camera"
--guidance_rescale 0.7 \
--perframe_ae \
--diffusion_dtype bf16 \
--projector_mode autocast \
--encoder_mode bf16_full \
--vae_dtype bf16
--projector_mode bf16_full \
--encoder_mode bf16_full
} 2>&1 | tee "${res_dir}/output.log"