[Edited] Refactor code to optimize performance#1325

MayureshMore wants to merge 2 commits intometa-llama:mainmeta-llama/llama:mainfrom

MayureshMore:mainMayureshMore/llama:mainCopy head branch name to clipboard