Change a comment

oobabooga · Jul 26, 2023 · d6314fd · d6314fd
1 parent de5de04
commit d6314fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/Extensions.md b/docs/Extensions.md
@@ -128,7 +128,7 @@ params = {
 class MyLogits(LogitsProcessor):
     """
     Manipulates the probabilities for the next token before it gets sampled.
-    Used in the custom_logits_processor function below.
+    Used in the logits_processor_modifier function below.
     """
     def __init__(self):
         pass

diff --git a/extensions/example/script.py b/extensions/example/script.py
@@ -24,7 +24,7 @@
 class MyLogits(LogitsProcessor):
     """
     Manipulates the probabilities for the next token before it gets sampled.
-    Used in the custom_logits_processor function below.
+    Used in the logits_processor_modifier function below.
     """
     def __init__(self):
         pass