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