Merge branch 'avril' into rep-pen-order

This commit is contained in:
vfbd
2022-08-23 14:47:29 -04:00
3 changed files with 23 additions and 29 deletions

View File

@ -31,7 +31,7 @@ import torch
from transformers import LogitsWarper, LogitsProcessor
class AdvancedRepetitionPenaltyLogitsProcessor(LogitsProcessor):
class AdvancedRepetitionPenaltyLogitsProcessor(LogitsWarper):
def __init__(self, *args, **kwargs):
pass