Merge branch 'master' into patch_hooks_improved_memory
This commit is contained in:
commit
815c6f36e1
|
@ -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)
|
||||
|
||||
|
|
|
@ -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,)
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue