diff --git a/warpers.py b/warpers.py index 2eac074e..488a901e 100644 --- a/warpers.py +++ b/warpers.py @@ -28,7 +28,7 @@ SOFTWARE. ''' import torch -from transformers import LogitsWarper, LogitsProcessor +from transformers import LogitsWarper class AdvancedRepetitionPenaltyLogitsProcessor(LogitsWarper):