Proposal
|
def save_model(self, path: str, only_rank0: bool = False, tokenizer: Optional[PreTrainedTokenizerBase] = None) -> None: |
|
def save_model(self, path: str, only_rank0: bool = False, tokenizer: Optional[PreTrainedTokenizerBase] = None) -> None: |
Probably a simple mistake, it need to clear one.
Self-service
Proposal
ColossalAI/applications/Chat/coati/trainer/ppo.py
Line 210 in d7bf284
ColossalAI/applications/Chat/coati/trainer/ppo.py
Line 213 in d7bf284
Probably a simple mistake, it need to clear one.
Self-service