github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Allow setting weight saving flag on hidream_o1

Browse Source
This commit is contained in:
Jaret Burkett
2026-07-16 07:40:25 -06:00
parent 741aeb9ce0
commit 5cb54ba9cc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/diffusion_models/hidream/hidream_o1_model.py
View File

@@ -140,7 +140,7 @@ class HidreamO1Model(BaseModel):
print(f"Using noise scale: {self.noise_scale}") print(f"Using noise scale: {self.noise_scale}")
global _GLOBAL_NOISE_SCALE global _GLOBAL_NOISE_SCALE
_GLOBAL_NOISE_SCALE = self.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 # static method to get the noise scheduler
@staticmethod @staticmethod