Merge pull request nebuly-ai#190 from egrefen/main

Change distillate to distill
SuperSecureHuman · Feb 28, 2023 · f292283 · f292283
2 parents ccc0e23 + 5157a6d
commit f292283
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/apps/accelerate/chatllama/README.md b/apps/accelerate/chatllama/README.md
@@ -34,7 +34,7 @@ from chatllama.rlhf.config import Config
 path = "path_to_config_file.yaml"
 config = Config(path=path)
 trainer = RLTrainer(config.trainer)
-trainer.distillate()
+trainer.distill()
 trainer.train()
 trainer.training_stats.plot()
 ```

diff --git a/apps/accelerate/chatllama/chatllama/rlhf/reward.py b/apps/accelerate/chatllama/chatllama/rlhf/reward.py
@@ -258,7 +258,7 @@ class RewardTrainer:
         train: Train the reward model
         generate_user_input: Generate the user input for the LLM to evaluate a
             couple, (user_input, completion) and assing a score
-        distillate: Parse the dataset and assign scores using LLMs
+        distill: Parse the dataset and assign scores using LLMs
     """
 
     def __init__(self, config: ConfigReward) -> None:
@@ -324,7 +324,7 @@ def __init__(self, config: ConfigReward) -> None:
                 config=self.config.deepspeed_config_path,
             )
 
-    def distillate(
+    def distill(
         self,
     ):
         """Parse the dataset and assign scores using LLMs