fix: handle nonfloat dtype during ckpt consolidation#996

HuiyingLi wants to merge 1 commit intomainNVIDIA-NeMo/Automodel:mainfrom

huiyingl/fix_consolidation_dtypeinfoNVIDIA-NeMo/Automodel:huiyingl/fix_consolidation_dtypeinfoCopy head branch name to clipboard