@@ -73,7 +73,7 @@ def __call__(
             prompt,
             padding="max_length",
             max_length=self.tokenizer.model_max_length,
-            return_tensors="pt",
+            return_tensors="np",
         )
         text_input_ids = text_inputs.input_ids
Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ def __call__(`
`73`	`73`	`prompt,`
`74`	`74`	`padding="max_length",`
`75`	`75`	`max_length=self.tokenizer.model_max_length,`
`76`		`- return_tensors="pt",`
	`76`	`+ return_tensors="np",`
`77`	`77`	`)`
`78`	`78`	`text_input_ids = text_inputs.input_ids`
`79`	`79`