Apply suggestions from code review

Commit

1 year ago

Apply suggestions from code review Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>