Skip to content

GitLab

Unverified Commit df309fc6 authored by ddobokki's avatar ddobokki Committed by GitHub
Browse files

[Chat] Remove duplicate functions (#3625)

parent 179558a8
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
applications/Chat/coati/trainer/ppo.py
View file @ df309fc6
......@@ -210,9 +210,6 @@ class PPOTrainer(Trainer):
def save_model(self, path: str, only_rank0: bool = False, tokenizer: Optional[PreTrainedTokenizerBase] = None) -> None:
self.strategy.save_model(model=self.actor, path=path, only_rank0=only_rank0, tokenizer=tokenizer)
def save_model(self, path: str, only_rank0: bool = False, tokenizer: Optional[PreTrainedTokenizerBase] = None) -> None:
self.strategy.save_model(model=self.actor, path=path, only_rank0=only_rank0, tokenizer=tokenizer)
def _set_default_generate_kwargs(strategy: Strategy, generate_kwargs: dict, actor: Actor) -> None:
origin_model = strategy._unwrap_actor(actor)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment