Charlie81 commited on
Commit
bd82f5d
·
1 Parent(s): 255d210

myolmoe name again

Browse files
Files changed (1) hide show
  1. scripts/eval.py +1 -1
scripts/eval.py CHANGED
@@ -207,7 +207,7 @@ def load_custom_model(args) -> HFLM:
207
 
208
  # CRITICAL FIX: Register the custom model class with the correct config
209
  from transformers import AutoConfig, AutoModelForCausalLM
210
- AutoConfig.register("olmoe", OlmoeConfig)
211
  AutoModelForCausalLM.register(OlmoeConfig, MyOlmoeForCausalLM)
212
  logger.info("Registered MyOlmoeForCausalLM with OlmoeConfig")
213
 
 
207
 
208
  # CRITICAL FIX: Register the custom model class with the correct config
209
  from transformers import AutoConfig, AutoModelForCausalLM
210
+ AutoConfig.register("myolmoe", OlmoeConfig)
211
  AutoModelForCausalLM.register(OlmoeConfig, MyOlmoeForCausalLM)
212
  logger.info("Registered MyOlmoeForCausalLM with OlmoeConfig")
213