Fix onnx tensor format (#654)

anton-l · web-flow · commit d8572f20c700 · 2022-09-27T19:09:13.000+02:00
fix np onnx
diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_onnx.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_onnx.py
@@ -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`