diff --git a/DeepSeek-OCR-master/DeepSeek-OCR-vllm/process/ngram_norepeat.py b/DeepSeek-OCR-master/DeepSeek-OCR-vllm/process/ngram_norepeat.py
index c130c2f8c..a17303541 100644
--- a/DeepSeek-OCR-master/DeepSeek-OCR-vllm/process/ngram_norepeat.py
+++ b/DeepSeek-OCR-master/DeepSeek-OCR-vllm/process/ngram_norepeat.py
@@ -1,6 +1,5 @@
 import torch
 from transformers import LogitsProcessor
-from transformers.generation.logits_process import _calc_banned_ngram_tokens
 from typing import List, Set
 
 
@@ -37,4 +36,4 @@ def __call__(self, input_ids: List[int], scores: torch.FloatTensor) -> torch.Flo
             for token in banned_tokens:
                 scores[token] = -float("inf")
         
-        return scores
\ No newline at end of file
+        return scores