theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 01:48:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

hunyuanvideo step_processor

Browse Source
This commit is contained in:
Artiprocher
2024-12-26 10:20:59 +08:00
parent 6a999e1127
commit 4f6d5e7074
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
diffsynth/schedulers/flow_match.py
View File

@@ -46,7 +46,6 @@ class FlowMatchScheduler():
def return_to_timestep(self, timestep, sample, sample_stablized): def return_to_timestep(self, timestep, sample, sample_stablized):
# This scheduler doesn't support this function.
if isinstance(timestep, torch.Tensor): if isinstance(timestep, torch.Tensor):
timestep = timestep.cpu() timestep = timestep.cpu()
timestep_id = torch.argmin((self.timesteps - timestep).abs()) timestep_id = torch.argmin((self.timesteps - timestep).abs())