Update MLX integration to use new generate_step function signature#3021

merrymercy merged 2 commits intolm-sys:mainlm-sys/FastChat:mainfrom

aliasaria:mainaliasaria/FastChat:mainCopy head branch name to clipboard