From 6dba61e0333f269a0dccca20e397f6e5e4c46aa0 Mon Sep 17 00:00:00 2001 From: blepping Date: Mon, 2 Sep 2024 19:11:11 -0600 Subject: [PATCH] Fix fadeout in Deep Shrink node --- py/nodes/deepShrink.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/nodes/deepShrink.py b/py/nodes/deepShrink.py index e19fed8..d11cb30 100644 --- a/py/nodes/deepShrink.py +++ b/py/nodes/deepShrink.py @@ -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