On decode, strip section tokens from start and end

Author: AustinMroz

nodes.py

@@ -63,6 +63,15 @@ def INPUT_TYPES(s):
     RETURN_TYPES = "AUDIO",
     CATEOGRY = "Orpheus"
     def decode(self, tokens, vae):
+        #trim any section tokens from start or end
+        start_index = 0
+        while tokens[start_index] < tokeniser_length + 10:
+            start_index += 1
+        end_index = 0
+        while tokens[end_index - 1] < tokeniser_length + 10:
+            end_index -= 1
+        if start_index != 0 or end_index != 0:
+            tokens = tokens[start_index:end_index]
         #assert len(tokens) % 7 == 0
         tokens = tokens[:len(tokens)//7*7]
         t = torch.tensor(tokens).reshape((-1,7))

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-orpheus"
 description = "Nodes for using Orpheus-TTS in ComfyUI"
-version = "0.1.3"
+version = "0.1.4"
 classifiers = [
     "License :: OSI Approved :: GNU General Public License v3 (GPLv3)"
 ]