Update type annotation and expand hint in attention function#1

tenosel merged 1 commit intomainMachine-Learning-Foundations/exercise_15_language_models:mainfrom

suggestionsMachine-Learning-Foundations/exercise_15_language_models:suggestionsCopy head branch name to clipboard