Just use the Automodel DPSKv4 branch : https://github.com/NVIDIA-NeMo/Automodel/pull/3252 cc @HuiyingLi
Just use the Automodel DPSKv4 branch : NVIDIA-NeMo/Automodel#3252
cc @HuiyingLi