broskicodes
/

simple-stories-4M

Text Generation

simple_stories_4m

text-generation-inference

Model card Files Files and versions Community

broskicodes commited on Jan 6

Commit

0e0495d

•

1 Parent(s): 5889dba

Update model_4m.py

Files changed (1) hide show

model_4m.py +1 -1

model_4m.py CHANGED Viewed

@@ -4,7 +4,7 @@ from simple_stories_4m_model.nano_gpt_model import NanoGPT
 import transformer_modules
 class SimpleStories4MModel(PreTrainedModel):
-    config_class = transformers_modules.broskicodes.simple-stories-4M.34bd6eabc4ee5a55bfb57b0048155fd1521ba2cb.config_4m.SimpleStories4MConfig
     def __init__(self, config):
         super().__init__(config)

 import transformer_modules
 class SimpleStories4MModel(PreTrainedModel):
+    config_class = SimpleStories4MConfig
     def __init__(self, config):
         super().__init__(config)