Remove prints.

This commit is contained in:
comfyanonymous 2024-11-22 10:51:31 -05:00
parent 5818f6cf51
commit 94323a26a7

View File

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