mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-10-29 01:11:18 +08:00
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 = extra.to(dtype)
|
||||||
extra_conds[o] = extra
|
extra_conds[o] = extra
|
||||||
|
|
||||||
print(t)
|
|
||||||
model_output = self.diffusion_model(xc, t, context=context, control=control, transformer_options=transformer_options, **extra_conds).float()
|
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)
|
return self.model_sampling.calculate_denoised(sigma, model_output, x)
|
||||||
|
|
||||||
|
|||||||
@ -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,)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user