predictor is a better name

This commit is contained in:
layerdiffusion
2024-08-01 21:29:34 -07:00
parent bc9977a305
commit f6981339b0
10 changed files with 17 additions and 17 deletions

View File

@@ -38,7 +38,7 @@ class DynamicThresholdingNode:
cond = input - args["cond"]
uncond = input - args["uncond"]
cond_scale = args["cond_scale"]
time_step = model.model.prediction.timestep(args["sigma"])
time_step = model.model.predictor.timestep(args["sigma"])
time_step = time_step[0].item()
dynamic_thresh.step = 999 - time_step