Skip to content

GitLab

Unverified Commit d8572f20 authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Fix onnx tensor format (#654) 

fix np onnx
parent c0c98df9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_onnx.py
View file @ d8572f20
...@@ -73,7 +73,7 @@ class StableDiffusionOnnxPipeline(DiffusionPipeline): ...@@ -73,7 +73,7 @@ class StableDiffusionOnnxPipeline(DiffusionPipeline):
prompt, prompt,
padding="max_length", padding="max_length",
max_length=self.tokenizer.model_max_length, max_length=self.tokenizer.model_max_length,
return_tensors="pt", return_tensors="np",
) )
text_input_ids = text_inputs.input_ids text_input_ids = text_inputs.input_ids
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment