chore: add changes to support Granite 4 models#599

dushyantbehl merged 2 commits intofoundation-model-stack:mainfoundation-model-stack/fms-hf-tuning:mainfrom

YashasviChaurasia:g4_supportYashasviChaurasia/fms-hf-tuning:g4_supportCopy head branch name to clipboard