fix wrong name, for #89

2024-01-05 22:34:47 -08:00 · 2024-01-05 22:34:47 -08:00 · e9d6c02438
parent 0d017bcf44
commit e9d6c02438
1 changed files with 1 additions and 1 deletions
--- a/dynthres_comfyui.py
+++ b/dynthres_comfyui.py
@ -70,7 +70,7 @@ class DynamicThresholdingSimpleComfyNode:
            uncond = input - args["uncond"]
            cond_scale = args["cond_scale"]
            time_step = model.model.model_sampling.timestep(args["sigma"])
-            time_step = time_step[0].value()
+            time_step = time_step[0].item()
            dynamic_thresh.step = 999 - time_step

            return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)