Upload deepseek_tinystories/__init__.py
Browse files
deepseek_tinystories/__init__.py
CHANGED
|
@@ -6,14 +6,14 @@ architecture, trained on the TinyStories dataset.
|
|
| 6 |
"""
|
| 7 |
|
| 8 |
from .modeling_deepseek import DeepseekInspiredModel, DeepSeekModelConfig
|
| 9 |
-
from .processor import
|
| 10 |
from .utils import load_model, generate_text
|
| 11 |
|
| 12 |
__version__ = "1.0.0"
|
| 13 |
__all__ = [
|
| 14 |
"DeepseekInspiredModel",
|
| 15 |
"DeepSeekModelConfig",
|
| 16 |
-
"
|
| 17 |
"load_model",
|
| 18 |
"generate_text"
|
| 19 |
]
|
|
|
|
| 6 |
"""
|
| 7 |
|
| 8 |
from .modeling_deepseek import DeepseekInspiredModel, DeepSeekModelConfig
|
| 9 |
+
from .processor import TinyStoriesProcesssor
|
| 10 |
from .utils import load_model, generate_text
|
| 11 |
|
| 12 |
__version__ = "1.0.0"
|
| 13 |
__all__ = [
|
| 14 |
"DeepseekInspiredModel",
|
| 15 |
"DeepSeekModelConfig",
|
| 16 |
+
"TinyStoriesProcesssor",
|
| 17 |
"load_model",
|
| 18 |
"generate_text"
|
| 19 |
]
|