mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
[CI] Fix Lint error (#2103)
This commit is contained in:
parent
6f3b6a5d12
commit
8c0ccf9a6b
@ -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'],
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user