From d6314fd5394bbfe2edd9030d953892dcfc4de105 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 26 Jul 2023 09:37:48 -0700
Subject: [PATCH] Change a comment

---
 docs/Extensions.md           | 2 +-
 extensions/example/script.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/Extensions.md b/docs/Extensions.md
index 74189efd..c7d1aa36 100644
--- 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
index f81a652b..669749c0 100644
--- a/extensions/example/script.py
+++ b/extensions/example/script.py
@@ -24,7 +24,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