Modalities / modalities

Modalities, a PyTorch-native framework for distributed and reproducible foundation model training.
MIT License
61 stars 5 forks source link

Include mamba state space model dependency #185

Open fromm-m opened 3 months ago

fromm-m commented 3 months ago

System Info

modalities==0.2 ubuntu python==3.10

🐛 Describe the bug

In https://github.com/Modalities/modalities/pull/182 we had to remove the mamba dependency because it caused issues by executing the tests as "packaging" in the mamba build was error-prone. In the mamba repository, there are problems that the library is at the moment not pip-installable https://github.com/state-spaces/mamba/pull/402