diff --git a/fast_llm/models/ssm/config.py b/fast_llm/models/ssm/config.py index 75bb2a3be..e27e52802 100644 --- a/fast_llm/models/ssm/config.py +++ b/fast_llm/models/ssm/config.py @@ -179,7 +179,6 @@ class HybridSSMModelConfig(FastLLMModelConfig): AprielSSMHHybridHuggingfaceCheckpointFormat, AprielThinkerSSMHHybridHuggingfaceCheckpointFormat, ) - checkpoint_formats = FastLLMModelConfig.checkpoint_formats + (LLambaHuggingfaceCheckpointFormat,) @classmethod def get_model_class(cls) -> type["HybridSSMModel"]: