Fixed issue #181 - Githubissues

kyegomez / zeta

Build high-performance AI models with modular building blocks

https://zeta.apac.ai

Apache License 2.0

320 stars 28 forks source link

Closed Bool1020 closed 2 months ago

Bool1020 commented 2 months ago

Description: Fixed issue where tensors in the SSM of the mambablock module were not on the same device.
Issue: #181
Dependencies: None
Tag maintainer: kye@apac.ai

kyegomez commented 2 months ago

@Bool1020 Appreciate you!