From a9be5fd6b0f81126f29e76e6ab65db03a9e5bdc9 Mon Sep 17 00:00:00 2001 From: anton-l Date: Tue, 27 Sep 2022 19:09:14 +0200 Subject: [PATCH] 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 23bd8fbc367a..07e9c1d9acd6 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 @@ 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