alxndrTL / mamba.py

A simple and efficient Mamba implementation in pure PyTorch and MLX.
MIT License
960 stars 86 forks source link

Fix MLX Generation script for Python 3.11 #31

Closed michal1000w closed 5 months ago

michal1000w commented 5 months ago

When using python 3.11 and mlx generate script there is an error "TypeError: can only concatenate tuple (not "list") to tuple". This pull request fixes that allowing the script to execute properly.

alxndrTL commented 5 months ago

Thank you for the PR!