diff --git a/auto_transformers.py b/auto_transformers.py index e5255de..e3fb185 100644 --- a/auto_transformers.py +++ b/auto_transformers.py @@ -104,7 +104,7 @@ class AutoTransformers(NNOperator): try: inputs = self.tokenizer(txt, padding=True, truncation=True, return_tensors="pt").to(self.device) except Exception: - inputs = self.tokenizer(dummy_input, truncation=True, return_tensors='pt').to(self.device) + inputs = self.tokenizer(txt, truncation=True, return_tensors='pt').to(self.device) try: outs = self.model(**inputs) except Exception as e: