sky-2002 commited on
Commit
43088b5
·
verified ·
1 Parent(s): 2314e2b

Upload deepseek_tinystories/__init__.py

Browse files
Files changed (1) hide show
  1. deepseek_tinystories/__init__.py +19 -0
deepseek_tinystories/__init__.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Deepseek-inspired TinyStories Model Package
3
+
4
+ This package contains a Deepseek-inspired language model with Mixture of Experts
5
+ architecture, trained on the TinyStories dataset.
6
+ """
7
+
8
+ from .modeling_deepseek import DeepseekInspiredModel, DeepSeekModelConfig
9
+ from .processor import TinyStoriesProcessor
10
+ from .utils import load_model, generate_text
11
+
12
+ __version__ = "1.0.0"
13
+ __all__ = [
14
+ "DeepseekInspiredModel",
15
+ "DeepSeekModelConfig",
16
+ "TinyStoriesProcessor",
17
+ "load_model",
18
+ "generate_text"
19
+ ]