Fix fadeout in Deep Shrink node

2024-09-02 19:11:11 -06:00
parent 68a94d0c00
commit 6dba61e033
1 changed files with 1 additions and 1 deletions
@@ -164,7 +164,7 @@ class DeepShrinkBleh:
                or block_num not in block_numbers
            ):
                return h
-            pct = 1.0 - (ms.timestep(sigma_tensor).detach().cpu() / 999)
+            pct = 1.0 - (ms.timestep(sigma_tensor).detach().cpu().item() / 999)
            if (
                pct < start_fadeout_percent
                or start_fadeout_percent > end_percent