diff --git a/llmfoundry/models/hf/hf_causal_lm.py b/llmfoundry/models/hf/hf_causal_lm.py index 498133262b..cdfa142420 100644 --- a/llmfoundry/models/hf/hf_causal_lm.py +++ b/llmfoundry/models/hf/hf_causal_lm.py @@ -71,9 +71,7 @@ def __init__(self, om_model_config: Union[DictConfig, InContextLearningLMAccuracy(), InContextLearningMultipleChoiceAccuracy(), InContextLearningQAAccuracy(), - InContextLearningCodeEvalAccuracy(), - InContextLearningLMExpectedCalibrationError(), - InContextLearningMCExpectedCalibrationError() + InContextLearningCodeEvalAccuracy() ] # if we are passed a DictConfig, we need to instantiate the model diff --git a/llmfoundry/models/mpt/modeling_mpt.py b/llmfoundry/models/mpt/modeling_mpt.py index 029ae78375..45bc4eb6d1 100644 --- a/llmfoundry/models/mpt/modeling_mpt.py +++ b/llmfoundry/models/mpt/modeling_mpt.py @@ -1025,8 +1025,6 @@ def __init__( InContextLearningMultipleChoiceAccuracy(), InContextLearningQAAccuracy(), InContextLearningCodeEvalAccuracy(), - InContextLearningLMExpectedCalibrationError(), - InContextLearningMCExpectedCalibrationError(), ] super().__init__(