VampNet

Running on Zero

initial commit

a189727 about 1 year ago

195 Bytes

import audiotools

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

from .transformer import VampNet

	import audiotools

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

	from .transformer import VampNet