change name casings

myolmoe/__init__.py

@@ -1 +1 @@
-from .modeling_olmoe import OLMoEForCausalLM
+from .modeling_olmoe import MyOlmoeForCausalLM

scripts/eval.py

@@ -227,8 +227,8 @@ def load_custom_model(args) -> HFLM:
     
     try:
         # Import custom model class
-        from modeling_myolmoe import MyOLMoEForCausalLM
-        logger.info("Successfully imported MyOLMoEForCausalLM")
+        from modeling_myolmoe import MyOlmoeForCausalLM
+        logger.info("Successfully imported MyOlmoeForCausalLM")
     except ImportError as e:
         logger.error(f"Failed to import custom model: {e}")
         logger.error("Make sure the custom model code is available in the specified path")
@@ -261,7 +261,7 @@ def load_custom_model(args) -> HFLM:
         }[args.dtype]
     
     # Load the custom model
-    hf_model = MyOLMoEForCausalLM.from_pretrained(
+    hf_model = MyOlmoeForCausalLM.from_pretrained(
         args.model_path,
         config=config,
         torch_dtype=torch_dtype,