Logic/DDT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bugs(admas timedeltas)

Browse Source
This commit is contained in:
wangshuai6
2025-05-20 12:25:50 +08:00
parent 45d5e8db5b
commit 4d4b3262c3
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusion/flow_matching/adam_sampling.py
View File

@@ -59,7 +59,7 @@ class AdamLMSampler(BaseSampler):
timesteps = torch.linspace(0.0, 1 - self.last_step, self.num_steps) timesteps = torch.linspace(0.0, 1 - self.last_step, self.num_steps)
timesteps = torch.cat([timesteps, torch.tensor([1.0])], dim=0) timesteps = torch.cat([timesteps, torch.tensor([1.0])], dim=0)
self.timesteps = shift_respace_fn(timesteps, timeshift) self.timesteps = shift_respace_fn(timesteps, timeshift)
self.timedeltas = timesteps[1:] - self.timesteps[:-1] self.timedeltas = self.timesteps[1:] - self.timesteps[:-1]
self._reparameterize_coeffs() self._reparameterize_coeffs()
def _reparameterize_coeffs(self): def _reparameterize_coeffs(self):

2
src/diffusion/stateful_flow_matching/adam_sampling.py
View File

@@ -61,7 +61,7 @@ class AdamLMSampler(BaseSampler):
timesteps = torch.linspace(0.0, 1 - self.last_step, self.num_steps) timesteps = torch.linspace(0.0, 1 - self.last_step, self.num_steps)
timesteps = torch.cat([timesteps, torch.tensor([1.0])], dim=0) timesteps = torch.cat([timesteps, torch.tensor([1.0])], dim=0)
self.timesteps = shift_respace_fn(timesteps, timeshift) self.timesteps = shift_respace_fn(timesteps, timeshift)
self.timedeltas = timesteps[1:] - self.timesteps[:-1] self.timedeltas = self.timesteps[1:] - self.timesteps[:-1]
self._reparameterize_coeffs() self._reparameterize_coeffs()
def _reparameterize_coeffs(self): def _reparameterize_coeffs(self):