[Bugfix] fix OmniGen2 offload and dtype mismatch #2560
+14
−8
Merged
Loading