Quick fix ssm checkpoint formats (#301)

nitsanluke · web-flow · commit e2f36d8111cf · 2025-06-12T14:59:50.000-04:00
diff --git 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"]:

Original file line number	Diff line number	Diff line change
`@@ -179,7 +179,6 @@ class HybridSSMModelConfig(FastLLMModelConfig):`
`179`	`179`	`AprielSSMHHybridHuggingfaceCheckpointFormat,`
`180`	`180`	`AprielThinkerSSMHHybridHuggingfaceCheckpointFormat,`
`181`	`181`	`)`
`182`		`- checkpoint_formats = FastLLMModelConfig.checkpoint_formats + (LLambaHuggingfaceCheckpointFormat,)`
`183`	`182`
`184`	`183`	`@classmethod`
`185`	`184`	`def get_model_class(cls) -> type["HybridSSMModel"]:`