From cdd507de43325db24b690a33acd77060443f36f1 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 8 Jun 2024 12:49:16 +0300
Subject: [PATCH] Update nodes.py

---
 nodes/nodes.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/nodes/nodes.py b/nodes/nodes.py
index 440934b..78e7c13 100644
--- a/nodes/nodes.py
+++ b/nodes/nodes.py
@@ -837,10 +837,11 @@ SVD:
     def customsigmas(self, sigmas_string, interpolate_to_steps):
         sigmas_list = sigmas_string.split(', ')
         sigmas_float_list = [float(sigma) for sigma in sigmas_list]
-        sigmas_tensor = torch.tensor(sigmas_float_list)
-        if len(sigmas_tensor) != interpolate_to_steps:
-            sigmas_tensor = self.loglinear_interp(sigmas_tensor, interpolate_to_steps)
-        return (sigmas_tensor,)
+        sigmas_tensor = torch.FloatTensor(sigmas_float_list)
+        if len(sigmas_tensor) != interpolate_to_steps + 1:
+            sigmas_tensor = self.loglinear_interp(sigmas_tensor, interpolate_to_steps + 1)
+        sigmas_tensor[-1] = 0
+        return (sigmas_tensor.float(),)
      
     def loglinear_interp(self, t_steps, num_steps):
         """