Accept mx.array type for prompt argument for stream_generate

ml-explore / mlx-examples

Examples in the MLX framework

MIT License

6.3k stars 898 forks source link

Closed neilmehta24 closed 5 days ago

neilmehta24 commented 5 days ago

This change allows mlx_lm.utils.stream_generate to accept an mx.array for its prompt argument