[CI] Fix Lint error (#2103)

This commit is contained in:
kkscilife 2025-05-16 15:36:45 +08:00 committed by GitHub
parent 6f3b6a5d12
commit 8c0ccf9a6b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions

View File

@ -75,7 +75,7 @@ class GLM130B(BaseModel):
['--vocab-size', '150528'], ['--vocab-size', '150528'],
['--num-attention-heads', '96'], ['--num-attention-heads', '96'],
['--max-sequence-length', '2048'], ['--max-sequence-length', '2048'],
['--tokenizer-type', 'icetk-glm-130B'], #gitleaks:allow ['--tokenizer-type', 'icetk-glm-130B'], # gitleaks:allow
['--layernorm-order', 'post'], ['--layernorm-order', 'post'],
['--load', self.ckpt_path], ['--load', self.ckpt_path],
['--skip-init'], ['--skip-init'],

View File

@ -358,7 +358,7 @@ class InternTrain(BaseModel):
output_text = self.batch_decode( output_text = self.batch_decode(
outputs, outputs,
eos_token_ids=self.generator.eos_token_id, eos_token_ids=self.generator.eos_token_id,
stopping_criteria=stopping_criteria) #gitleaks:allow stopping_criteria=stopping_criteria) # gitleaks:allow
return output_text return output_text