Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix onnx tensor format (#654) 

fix np onnx
This commit is contained in:
Anton Lozhkov 2022-09-27 19:09:13 +02:00 committed by GitHub
parent c0c98df9a1
commit d8572f20c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_onnx.py
View File

@ -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