From 5cb54ba9ccc9afd623eb9cc13fd819cb8938f998 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Thu, 16 Jul 2026 07:40:25 -0600 Subject: [PATCH] Allow setting weight saving flag on hidream_o1 --- .../diffusion_models/hidream/hidream_o1_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions_built_in/diffusion_models/hidream/hidream_o1_model.py b/extensions_built_in/diffusion_models/hidream/hidream_o1_model.py index cf58ee0..4542bef 100644 --- a/extensions_built_in/diffusion_models/hidream/hidream_o1_model.py +++ b/extensions_built_in/diffusion_models/hidream/hidream_o1_model.py @@ -140,7 +140,7 @@ class HidreamO1Model(BaseModel): print(f"Using noise scale: {self.noise_scale}") global _GLOBAL_NOISE_SCALE _GLOBAL_NOISE_SCALE = self.noise_scale - self.is_comfy_weight = False # save as single file if true + self.is_comfy_weight = self.model_config.model_kwargs.get("is_comfy_weight", False) # static method to get the noise scheduler @staticmethod