use local imports

model_4m.py → modeling_ss4m.py

@@ -1,6 +1,6 @@
 from transformers import PreTrainedModel
-from simple_stories_4m_model.config_4m import SimpleStories4MConfig
-from simple_stories_4m_model.nano_gpt_model import NanoGPT
+from .configuration_ss4m import SimpleStories4MConfig
+from .nano_gpt_model import NanoGPT
 
 class SimpleStories4MModel(PreTrainedModel):
     config_class = SimpleStories4MConfig