diff --git a/torch_brain/models/__init__.py b/torch_brain/models/__init__.py index dcb1f62..e32b2c7 100644 --- a/torch_brain/models/__init__.py +++ b/torch_brain/models/__init__.py @@ -1,2 +1,2 @@ -from .poyo_plus import POYOPlus, POYOPlusTokenizer from .poyo import POYO, poyo_mp, POYOTokenizer +from .poyo_plus import POYOPlus, POYOPlusTokenizer