myolmoe name again
Browse files- 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("
|
| 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 |
|