Fix onnx tensor format (#654)

fix np onnx
2022-09-27 19:09:13 +02:00 · 2022-09-27 19:09:13 +02:00 · d8572f20c7
parent c0c98df9a1
commit d8572f20c7
1 changed files with 1 additions and 1 deletions
--- 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 @@ class StableDiffusionOnnxPipeline(DiffusionPipeline):
            prompt,
            padding="max_length",
            max_length=self.tokenizer.model_max_length,
-            return_tensors="pt",
+            return_tensors="np",
        )
        text_input_ids = text_inputs.input_ids