Deploy TDATR ZeroGPU Space

TDATR/tokenizers/bbox_tokenizer.py

@@ -13,7 +13,7 @@ from torch import Tensor
 import json
 
 from TDATR_utils.dataclass import HulkDataclass
-from TDATR.tokenizers.sp_tokenizer import SPTokenizerConfig, SPMiniGPT4Tokenizer
+from TDATR.tokenizers.sp_tokenizer import SPTokenizerConfig, SPTokenizer
 
 logger= logging.getLogger(__name__)
 
@@ -87,7 +87,7 @@ class BboxTokenConfig(SPTokenizerConfig):
     )
 
 
-class BboxTokenizer(SPMiniGPT4Tokenizer):
+class BboxTokenizer(SPTokenizer):
     def __init__(self, cfg: BboxTokenConfig):
         super().__init__(cfg)
         self.patch_num = cfg.patch_num
@@ -378,4 +378,3 @@ def get_cell_ranges(
         ranges.append(range_list)
     
     return ranges
-