vall-e/vall_e/models/retnet.py

3 lines
142 B
Python
Executable File

from torchscale.architecture.config import RetNetConfig
from torchscale.architecture.retnet import RetNetDecoder
# from retnet import RetNet