From 0268305cfe09b2198f871d730ec4719b884d0fed Mon Sep 17 00:00:00 2001 From: onesome Date: Tue, 25 Apr 2023 18:26:49 -0500 Subject: [PATCH] Change fallback notifications to warnings --- modeling/inference_models/generic_hf_torch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modeling/inference_models/generic_hf_torch.py b/modeling/inference_models/generic_hf_torch.py index a9fc6370..8722c722 100644 --- a/modeling/inference_models/generic_hf_torch.py +++ b/modeling/inference_models/generic_hf_torch.py @@ -81,11 +81,11 @@ class GenericHFTorchInferenceModel(HFTorchInferenceModel): try: metamodel = AutoModelForCausalLM.from_config(self.model_config) except Exception as e: - logger.error(f"Fell back to neo for metamodel due to {e}") + logger.warning(f"Fell back to neo for metamodel due to {e}") try: metamodel = GPTNeoForCausalLM.from_config(self.model_config) except Exception as e: - logger.error(f"Falling back again due to {e}") + logger.warning(f"Falling back again due to {e}") raise SuperLegacyModelError utils.layers_module_names = utils.get_layers_module_names(metamodel)