Update __init__.py
Browse files- __init__.py +2 -2
__init__.py
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
from .configuration_gator import GatorConfig
|
| 2 |
-
from .modeling_gator import GatorModel
|
| 3 |
-
__all__ = ["GatorConfig", "GatorModel"]
|
|
|
|
| 1 |
from .configuration_gator import GatorConfig
|
| 2 |
+
from .modeling_gator import GatorModel, GatorModelForCasualLM
|
| 3 |
+
__all__ = ["GatorConfig", "GatorModel", "GatorModelForCasualLM"]
|