bug fix

ud.py

@@ -1,5 +1,10 @@
 from transformers import TokenClassificationPipeline,DebertaV2TokenizerFast
 from transformers.models.bert_japanese.tokenization_bert_japanese import MecabTokenizer
+try:
+  from transformers.utils import cached_file
+except:
+  from transformers.file_utils import cached_path,hf_bucket_url
+  cached_file=lambda x,y:cached_path(hf_bucket_url(x,y))
 
 class UniversalDependenciesPipeline(TokenClassificationPipeline):
   def _forward(self,model_inputs):
@@ -78,14 +83,9 @@ class JumanDebertaV2TokenizerFast(DebertaV2TokenizerFast):
     if not (os.path.isdir(d) and os.path.isfile(r)):
       import zipfile
       import tempfile
-      try:
-        from transformers.utils import cached_file
-      except:
-        from transformers.file_utils import cached_path,hf_bucket_url
-        cached_file=lambda x,y:cached_path(hf_bucket_url(x,y))
       self.dicdir=tempfile.TemporaryDirectory()
       d=self.dicdir.name
-      with zipfile.ZipFile(cached_file("KoichiYasuoka/deberta-base-japanese-juman-ud-goeswith","mecab-jumandic-utf8.zip")) as z:
+      with zipfile.ZipFile(cached_file(self.name_or_path,"mecab-jumandic-utf8.zip")) as z:
         z.extractall(d)
       r=os.path.join(d,"mecabrc")
       with open(r,"w",encoding="utf-8") as w:
@@ -101,3 +101,4 @@ class JumanDebertaV2TokenizerFast(DebertaV2TokenizerFast):
     super().save_pretrained(save_directory,**kwargs)
     self._tokenizer.pre_tokenizer=self.custom_pre_tokenizer
     shutil.copy(os.path.abspath(__file__),os.path.join(save_directory,"ud.py"))
+    shutil.copy(cached_file(self.name_or_path,"mecab-jumandic-utf8.zip"),os.path.join(save_directory,"mecab-jumandic-utf8.zip"))