fix: image build failure due to flash attn#629

ashokponkumar merged 3 commits intofoundation-model-stack:mainfoundation-model-stack/fms-hf-tuning:mainfrom

dushyantbehl:fix-image-failuredushyantbehl/fms-hf-tuning:fix-image-failureCopy head branch name to clipboard