From d8572f20c7007d96be07fa80cee41f515f77e092 Mon Sep 17 00:00:00 2001 From: Anton Lozhkov Date: Tue, 27 Sep 2022 19:09:13 +0200 Subject: [PATCH] Fix onnx tensor format (#654) fix np onnx --- .../stable_diffusion/pipeline_stable_diffusion_onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_onnx.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_onnx.py index 23bd8fbc..07e9c1d9 100644 --- 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