jianchen0311 commited on
Commit
92d8d54
·
verified ·
1 Parent(s): 034c56f

Upload model

Browse files
Files changed (1) hide show
  1. modeling_dflash.py +0 -1
modeling_dflash.py CHANGED
@@ -179,7 +179,6 @@ def extract_context_feature(
179
  class DFlashDraftModel(Qwen3PreTrainedModel):
180
  config_class = Qwen3Config
181
  _no_split_modules = ["Qwen3DFlashDecoderLayer"]
182
- _keys_to_ignore_on_load_missing = ["lm_head.weight", "embed_tokens.weight"]
183
  def __init__(self, config) -> None:
184
  super().__init__(config)
185
  self.config = config
 
179
  class DFlashDraftModel(Qwen3PreTrainedModel):
180
  config_class = Qwen3Config
181
  _no_split_modules = ["Qwen3DFlashDecoderLayer"]
 
182
  def __init__(self, config) -> None:
183
  super().__init__(config)
184
  self.config = config