fix: align sage_attn default hyperparameter return type with refactor#540

gsprochette merged 1 commit intomainPrunaAI/pruna:mainfrom

fix/sage_attn-model-dependant-hyperparameterPrunaAI/pruna:fix/sage_attn-model-dependant-hyperparameterCopy head branch name to clipboard