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