jammygrams
/

bart-qa

text2text-generation

Model card Files Files and versions

jammygrams commited on Nov 20, 2022

Commit

0c9e069

·

1 Parent(s): bbefeb7

update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -12,7 +12,7 @@ class EndpointHandler():
         path: str,
     ):
         # self.tagger = SequenceTagger.load(os.path.join(path,"pytorch_model.bin"))
-        self.tokenizer = AutoTokenizer.from_file(os.path.join(path, "tokenizer.json"))
         self.model = AutoModelForSeq2SeqLM.from_pretrained(os.path.join(path, "pytorch_model.bin"))
         self.model.set_active_adapters(ADAPTER_NAME)

         path: str,
     ):
         # self.tagger = SequenceTagger.load(os.path.join(path,"pytorch_model.bin"))
+        self.tokenizer = AutoTokenizer.from_pretrained(os.path.join(path, "tokenizer.json"))
         self.model = AutoModelForSeq2SeqLM.from_pretrained(os.path.join(path, "pytorch_model.bin"))
         self.model.set_active_adapters(ADAPTER_NAME)