Fix onnx tensor format (#654)

fix np onnx

Fix onnx tensor format (#654)
fix np onnx
d8572f20 · Anton Lozhkov · GitHub · c0c98df9 · d8572f20
Unverified Commit d8572f20 authored Sep 27, 2022 by Anton Lozhkov Committed by GitHub Sep 27, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_onnx.py ...elines/stable_diffusion/pipeline_stable_diffusion_onnx.py +1 -1

No files found.
--- 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