Commit Graph

4 Commits

Author SHA1 Message Date
Adrian Corduneanu
4d12b3da87
Merge fea47e70f7 into ec902046f6 2025-05-28 14:31:49 +08:00
Ang Wang
76e9427657
Format the code (#402)
* isort the code

* format the code

* Add yapf config file

* Remove torch cuda memory profiler
2025-05-16 12:35:38 +08:00
Adrian Corduneanu
fea47e70f7
Fix assertion error in UniPC scheduler for high step counts
This fixes an edge case in the FlowUniPCMultistepScheduler where using high sampling step counts (> 50) would cause an assertion error in the last step. The issue was that with lower_order_final=True, the order calculation could become 0 when step_index equals len(timesteps), causing 'assert self.this_order > 0' to fail.

The fix ensures this_order is always at least 1, maintaining stability while allowing higher quality generation with increased step counts.

🤖 Generated with Claude Code
Co-Authored-By: Claude noreply@anthropic.com
2025-02-28 22:31:58 -08:00
WanX-Video-1
65386b2e03 init upload 2025-02-25 22:07:47 +08:00