jianchen0311 commited on
Commit
60b575a
·
verified ·
1 Parent(s): 92d8d54

Upload model

Browse files
Files changed (1) hide show
  1. modeling_dflash.py +2 -1
modeling_dflash.py CHANGED
@@ -229,7 +229,8 @@ class DFlashDraftModel(Qwen3PreTrainedModel):
229
  stop_token_ids: list[int],
230
  temperature: float,
231
  ):
232
- self.eval()
 
233
  num_input_tokens = input_ids.shape[1]
234
  max_length = num_input_tokens + max_new_tokens
235
 
 
229
  stop_token_ids: list[int],
230
  temperature: float,
231
  ):
232
+ self.eval()
233
+ target.eval()
234
  num_input_tokens = input_ids.shape[1]
235
  max_length = num_input_tokens + max_new_tokens
236