File size: 195 Bytes
41b9d24
 
 
 
 
 
1
2
3
4
5
6
import audiotools

audiotools.ml.BaseModel.INTERN += ["vampnet.modules.**"]
audiotools.ml.BaseModel.EXTERN += ["einops", "flash_attn.flash_attention", "loralib"]

from .transformer import VampNet