Merge branch 'master' into patch_hooks_improved_memory

This commit is contained in:
Jedrzej Kosinski 2024-11-22 10:13:28 -06:00
commit 815c6f36e1
2 changed files with 0 additions and 3 deletions

View File

@ -154,7 +154,6 @@ class BaseModel(torch.nn.Module):
extra = extra.to(dtype)
extra_conds[o] = extra
print(t)
model_output = self.diffusion_model(xc, t, context=context, control=control, transformer_options=transformer_options, **extra_conds).float()
return self.model_sampling.calculate_denoised(sigma, model_output, x)

View File

@ -152,7 +152,6 @@ class LTXVScheduler:
mm = (max_shift - base_shift) / (x2 - x1)
b = base_shift - mm * x1
sigma_shift = (tokens) * mm + b
print(sigma_shift)
power = 1
sigmas = torch.where(
@ -170,7 +169,6 @@ class LTXVScheduler:
stretched = 1.0 - (one_minus_z / scale_factor)
sigmas[non_zero_mask] = stretched
print(sigmas)
return (sigmas,)