Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
handle both tensor and numpy input in init_sigmas_for_each_stage
from upstream: jy0205/Pyramid-Flow@bf10e73#diff-8e1c08bf8ab5d587ee253dccfb631b170202adf0664272785a9b377975ffb8a5R145
- Loading branch information