From 38fbcb6e54989cdb6006d7cfca226ef29e2fe473 Mon Sep 17 00:00:00 2001 From: Luke Kumar Date: Thu, 12 Jun 2025 18:57:35 +0000 Subject: [PATCH] quick fix ssm checkpoint formats --- fast_llm/models/ssm/config.py | 1 - 1 file changed, 1 deletion(-) 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"]: