UNet from Scratch

Now backend rewrite is about 50% finished.
Estimated finish is in 72 hours.
After that, many newer features will land.
This commit is contained in:
layerdiffusion
2024-08-01 21:19:41 -07:00
parent e3522c8919
commit bc9977a305
20 changed files with 1393 additions and 56 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.model_sampling.timestep(args["sigma"])
time_step = model.model.prediction.timestep(args["sigma"])
time_step = time_step[0].item()
dynamic_thresh.step = 999 - time_step