[fix bug] PixArt inference_steps=1#11079

Merged

hlky merged 3 commits intohuggingface:mainfrom

lawrence-cj:main

Mar 20, 2025

Commits on Mar 17, 2025

fix bug when pixart-dmd inference with `num_inference_steps=1`
lawrence-cj
committed

Commits on Mar 20, 2025

use return_dict=False and return [1] element for 1-step pixart model, which works for both lcm and dmd
lawrence-cj
committed
Merge branch 'main' into main
lawrence-cj
authored