vall-e/vall_e/models/arch/mamba_vasqu/mamba2_hf.py

4 lines
236 B
Python
Raw Normal View History

# https://github.com/vasqu/mamba2-torch
# NOTE: edit `src/mamba2_torch/__init__.py` to remove reference to .src. because of how pip treats packages
from mamba2_torch import Mamba2Model as Mamba2Model_HF, Mamba2Config as Mamba2Config_HF