Remove prints.

This commit is contained in:
comfyanonymous 2024-11-22 10:51:31 -05:00
parent 4004280bba
commit 2f708b29c9

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,)