github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[chatgpt] type miss of kwargs (#3107)

pull/3105/head
hiko2MSP 2023-03-13 01:00:02 +09:00 committed by GitHub
parent 145ccfd7d1
commit 191daf7411
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
applications/ChatGPT/chatgpt/models/opt/opt_critic.py
View File

@ -25,7 +25,7 @@ class OPTCritic(Critic):
checkpoint: bool = False, checkpoint: bool = False,
lora_rank: int = 0, lora_rank: int = 0,
lora_train_bias: str = 'none', lora_train_bias: str = 'none',
**kargs) -> None: **kwargs) -> None:
if pretrained is not None: if pretrained is not None:
model = OPTModel.from_pretrained(pretrained) model = OPTModel.from_pretrained(pretrained)
elif config is not None: elif config is not None: