VAE 也做 BF16
这个权重不做修改更好精度
This commit is contained in:
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -22,6 +22,7 @@ dataset="unitree_g1_pack_camera"
|
||||
--guidance_rescale 0.7 \
|
||||
--perframe_ae \
|
||||
--diffusion_dtype bf16 \
|
||||
--projector_dtype bf16 \
|
||||
--encoder_mode autocast #fp32/autocast/bf16_full
|
||||
--projector_mode autocast \
|
||||
--encoder_mode bf16_full \
|
||||
--vae_dtype bf16
|
||||
} 2>&1 | tee "${res_dir}/output.log"
|
||||
|
||||
Reference in New Issue
Block a user