ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-11 08:20:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix saving for hidream-o1

Browse Source
This commit is contained in:
Jaret Burkett
2026-05-10 12:10:02 -06:00
parent 05e85d18b5
commit bbdb05d744
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/src/hidream_o1/qwen3_vl_transformers.py
View File

@@ -2104,7 +2104,7 @@ class Qwen3VLCausalLMOutputWithPast(ModelOutput):
class Qwen3VLForConditionalGeneration(Qwen3VLPreTrainedModel, GenerationMixin):
_checkpoint_conversion_mapping = {}
_tied_weights_keys = ["lm_head.weight"]
_tied_weights_keys = {"lm_head.weight": "model.language_model.embed_tokens.weight"}
# Reference: fix gemma3 grad acc #37208
accepts_loss_kwargs = False
config: Qwen3VLConfig