Remove unused import from warpers.py
This commit is contained in:
parent
74922966bd
commit
9eecb61fea
|
@ -28,7 +28,7 @@ SOFTWARE.
|
||||||
'''
|
'''
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
from transformers import LogitsWarper, LogitsProcessor
|
from transformers import LogitsWarper
|
||||||
|
|
||||||
|
|
||||||
class AdvancedRepetitionPenaltyLogitsProcessor(LogitsWarper):
|
class AdvancedRepetitionPenaltyLogitsProcessor(LogitsWarper):
|
||||||
|
|
Loading…
Reference in New Issue