Files
custom-nodes-security-scan/docs/comfyui-sound-lab-bandit.html
zeekay 6bdea42e64 rebrand: Hanzo AI Inc — full rebrand from ComfyUI/Comfy-Org
- Updated copyright to Hanzo AI, Inc.
- Replaced ComfyUI references with Hanzo Studio
- Updated GitHub org URLs (Comfy-Org → hanzoui)
- Updated domains (comfy.org → hanzo.ai)
- Updated package names and npm scopes
- Updated LICENSE with Hanzo AI copyright header
2026-02-23 10:54:25 -08:00

4279 lines
188 KiB
HTML

<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<title>
Bandit Report
</title>
<style>
html * {
font-family: "Arial", sans-serif;
}
pre {
font-family: "Monaco", monospace;
}
.bordered-box {
border: 1px solid black;
padding-top:.5em;
padding-bottom:.5em;
padding-left:1em;
}
.metrics-box {
font-size: 1.1em;
line-height: 130%;
}
.metrics-title {
font-size: 1.5em;
font-weight: 500;
margin-bottom: .25em;
}
.issue-description {
font-size: 1.3em;
font-weight: 500;
}
.candidate-issues {
margin-left: 2em;
border-left: solid 1px; LightGray;
padding-left: 5%;
margin-top: .2em;
margin-bottom: .2em;
}
.issue-block {
border: 1px solid LightGray;
padding-left: .5em;
padding-top: .5em;
padding-bottom: .5em;
margin-bottom: .5em;
}
.issue-sev-high {
background-color: Pink;
}
.issue-sev-medium {
background-color: NavajoWhite;
}
.issue-sev-low {
background-color: LightCyan;
}
</style>
</head>
<body>
<div id="metrics">
<div class="metrics-box bordered-box">
<div class="metrics-title">
Metrics:<br>
</div>
Total lines of code: <span id="loc">11141</span><br>
Total lines skipped (#nosec): <span id="nosec">0</span>
</div>
</div>
<br>
<div id="results">
<div id="issue-0">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Consider possible security implications associated with the subprocess module.<br>
<b>Test ID:</b> B404<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/78.html" target="_blank">CWE-78</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/ffmpy-0.3.2/build/lib/ffmpy.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/ffmpy-0.3.2/build/lib/ffmpy.py</a><br>
<b>Line number: </b>3<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_imports.html#b404-import-subprocess" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_imports.html#b404-import-subprocess</a><br>
<div class="code">
<pre>
1 import errno
2 import shlex
3 import subprocess
4
5 __version__ = &quot;0.3.2&quot;
6
7
8 class FFmpeg(object):
</pre>
</div>
</div>
</div>
<div id="issue-1">
<div class="issue-block issue-sev-low">
<b>subprocess_without_shell_equals_true: </b> subprocess call - check for execution of untrusted input.<br>
<b>Test ID:</b> B603<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/78.html" target="_blank">CWE-78</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/ffmpy-0.3.2/build/lib/ffmpy.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/ffmpy-0.3.2/build/lib/ffmpy.py</a><br>
<b>Line number: </b>95<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b603_subprocess_without_shell_equals_true.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b603_subprocess_without_shell_equals_true.html</a><br>
<div class="code">
<pre>
91 :raise: `FFRuntimeError` in case FFmpeg command exits with a non-zero code;
92 `FFExecutableNotFoundError` in case the executable path passed was not valid
93 &quot;&quot;&quot;
94 try:
95 self.process = subprocess.Popen(
96 self._cmd,
97 stdin=subprocess.PIPE,
98 stdout=stdout,
99 stderr=stderr,
100 env=env,
101 **kwargs
102 )
103 except OSError as e:
104 if e.errno == errno.ENOENT:
105 raise FFExecutableNotFoundError(
</pre>
</div>
</div>
</div>
<div id="issue-2">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Consider possible security implications associated with the subprocess module.<br>
<b>Test ID:</b> B404<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/78.html" target="_blank">CWE-78</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/ffmpy-0.3.2/ffmpy.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/ffmpy-0.3.2/ffmpy.py</a><br>
<b>Line number: </b>3<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_imports.html#b404-import-subprocess" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_imports.html#b404-import-subprocess</a><br>
<div class="code">
<pre>
1 import errno
2 import shlex
3 import subprocess
4
5 __version__ = &quot;0.3.2&quot;
6
7
8 class FFmpeg(object):
</pre>
</div>
</div>
</div>
<div id="issue-3">
<div class="issue-block issue-sev-low">
<b>subprocess_without_shell_equals_true: </b> subprocess call - check for execution of untrusted input.<br>
<b>Test ID:</b> B603<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/78.html" target="_blank">CWE-78</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/ffmpy-0.3.2/ffmpy.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/ffmpy-0.3.2/ffmpy.py</a><br>
<b>Line number: </b>95<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b603_subprocess_without_shell_equals_true.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b603_subprocess_without_shell_equals_true.html</a><br>
<div class="code">
<pre>
91 :raise: `FFRuntimeError` in case FFmpeg command exits with a non-zero code;
92 `FFExecutableNotFoundError` in case the executable path passed was not valid
93 &quot;&quot;&quot;
94 try:
95 self.process = subprocess.Popen(
96 self._cmd,
97 stdin=subprocess.PIPE,
98 stdout=stdout,
99 stderr=stderr,
100 env=env,
101 **kwargs
102 )
103 except OSError as e:
104 if e.errno == errno.ENOENT:
105 raise FFExecutableNotFoundError(
</pre>
</div>
</div>
</div>
<div id="issue-4">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Consider possible security implications associated with the subprocess module.<br>
<b>Test ID:</b> B404<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/78.html" target="_blank">CWE-78</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>8<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_imports.html#b404-import-subprocess" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_imports.html#b404-import-subprocess</a><br>
<div class="code">
<pre>
4 import os
5 import posixpath
6 import random
7 import re
8 import subprocess
9 import time
10 import torch
11 import torchaudio
</pre>
</div>
</div>
</div>
<div id="issue-5">
<div class="issue-block issue-sev-low">
<b>try_except_pass: </b> Try, Except, Pass detected.<br>
<b>Test ID:</b> B110<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>45<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b110_try_except_pass.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b110_try_except_pass.html</a><br>
<div class="code">
<pre>
41 is_hidden = os.path.basename(f.path).startswith(&quot;.&quot;)
42
43 if file_ext in ext and not is_hidden:
44 files.append(f.path)
45 except:
46 pass
47 except:
48 pass
49
</pre>
</div>
</div>
</div>
<div id="issue-6">
<div class="issue-block issue-sev-low">
<b>try_except_pass: </b> Try, Except, Pass detected.<br>
<b>Test ID:</b> B110<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>47<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b110_try_except_pass.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b110_try_except_pass.html</a><br>
<div class="code">
<pre>
43 if file_ext in ext and not is_hidden:
44 files.append(f.path)
45 except:
46 pass
47 except:
48 pass
49
50 for dir in list(subfolders):
51 sf, f = fast_scandir(dir, ext)
</pre>
</div>
</div>
</div>
<div id="issue-7">
<div class="issue-block issue-sev-low">
<b>try_except_pass: </b> Try, Except, Pass detected.<br>
<b>Test ID:</b> B110<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>83<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b110_try_except_pass.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b110_try_except_pass.html</a><br>
<div class="code">
<pre>
79 has_banned = any(
80 [banned_word in name_lower for banned_word in banned_words])
81 if has_ext and has_keyword and not has_banned and not is_hidden and not os.path.basename(f.path).startswith(&quot;._&quot;):
82 files.append(f.path)
83 except:
84 pass
85 except:
86 pass
87
</pre>
</div>
</div>
</div>
<div id="issue-8">
<div class="issue-block issue-sev-low">
<b>try_except_pass: </b> Try, Except, Pass detected.<br>
<b>Test ID:</b> B110<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>85<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b110_try_except_pass.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b110_try_except_pass.html</a><br>
<div class="code">
<pre>
81 if has_ext and has_keyword and not has_banned and not is_hidden and not os.path.basename(f.path).startswith(&quot;._&quot;):
82 files.append(f.path)
83 except:
84 pass
85 except:
86 pass
87
88 for dir in list(subfolders):
89 sf, f = keyword_scandir(dir, ext, keywords)
</pre>
</div>
</div>
</div>
<div id="issue-9">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>224<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
220 custom_metadata = custom_metadata_fn(info, audio)
221 info.update(custom_metadata)
222
223 if &quot;__reject__&quot; in info and info[&quot;__reject__&quot;]:
224 return self[random.randrange(len(self))]
225
226 return (audio, info)
227 except Exception as e:
</pre>
</div>
</div>
</div>
<div id="issue-10">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>229<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
225
226 return (audio, info)
227 except Exception as e:
228 print(f&#x27;Couldn\&#x27;t load file {audio_filename}: {e}&#x27;)
229 return self[random.randrange(len(self))]
230
231 def group_by_keys(data, keys=wds.tariterators.base_plus_ext, lcase=True, suffixes=None, handler=None):
232 &quot;&quot;&quot;Return function over iterator that groups key, value pairs into samples.
</pre>
</div>
</div>
</div>
<div id="issue-11">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>238<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
234 :param lcase: convert suffixes to lower case (Default value = True)
235 &quot;&quot;&quot;
236 current_sample = None
237 for filesample in data:
238 assert isinstance(filesample, dict)
239 fname, value = filesample[&quot;fname&quot;], filesample[&quot;data&quot;]
240 prefix, suffix = keys(fname)
241 if wds.tariterators.trace:
</pre>
</div>
</div>
</div>
<div id="issue-12">
<div class="issue-block issue-sev-low">
<b>subprocess_without_shell_equals_true: </b> subprocess call - check for execution of untrusted input.<br>
<b>Test ID:</b> B603<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/78.html" target="_blank">CWE-78</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>286<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b603_subprocess_without_shell_equals_true.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b603_subprocess_without_shell_equals_true.html</a><br>
<div class="code">
<pre>
282 # Add the --recursive flag if requested
283 cmd.append(&#x27;--recursive&#x27;)
284
285 # Run the `aws s3 ls` command and capture the output
286 run_ls = subprocess.run(cmd, capture_output=True, check=True)
287 # Split the output into lines and strip whitespace from each line
288 contents = run_ls.stdout.decode(&#x27;utf-8&#x27;).split(&#x27;\n&#x27;)
289 contents = [x.strip() for x in contents if x]
</pre>
</div>
</div>
</div>
<div id="issue-13">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>560<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
556 def create_dataloader_from_config(dataset_config, batch_size, sample_size, sample_rate, audio_channels=2, num_workers=4):
557
558 dataset_type = dataset_config.get(&quot;dataset_type&quot;, None)
559
560 assert dataset_type is not None, &quot;Dataset type must be specified in dataset config&quot;
561
562 if audio_channels == 1:
563 force_channels = &quot;mono&quot;
</pre>
</div>
</div>
</div>
<div id="issue-14">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>571<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
567 if dataset_type == &quot;audio_dir&quot;:
568
569 audio_dir_configs = dataset_config.get(&quot;datasets&quot;, None)
570
571 assert audio_dir_configs is not None, &quot;Directory configuration must be specified in datasets[\&quot;dataset\&quot;]&quot;
572
573 configs = []
574
</pre>
</div>
</div>
</div>
<div id="issue-15">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/dataset.py</a><br>
<b>Line number: </b>577<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
573 configs = []
574
575 for audio_dir_config in audio_dir_configs:
576 audio_dir_path = audio_dir_config.get(&quot;path&quot;, None)
577 assert audio_dir_path is not None, &quot;Path must be set for local audio directory configuration&quot;
578
579 custom_metadata_fn = None
580 custom_metadata_module_path = audio_dir_config.get(&quot;custom_metadata_module&quot;, None)
</pre>
</div>
</div>
</div>
<div id="issue-16">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/utils.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/utils.py</a><br>
<b>Line number: </b>42<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
38
39 # If randomize is False, always start at the beginning of the audio
40 offset = 0
41 if(self.randomize and n_samples &gt; self.n_samples):
42 offset = random.randint(0, upper_bound)
43
44 # Calculate the start and end times of the chunk
45 t_start = offset / (upper_bound + self.n_samples)
</pre>
</div>
</div>
</div>
<div id="issue-17">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/utils.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/data/utils.py</a><br>
<b>Line number: </b>78<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
74 def __init__(self, p=0.5):
75 super().__init__()
76 self.p = p
77 def __call__(self, signal):
78 return -signal if (random.random() &lt; self.p) else signal
79
80 class Mono(nn.Module):
81 def __call__(self, signal):
</pre>
</div>
</div>
</div>
<div id="issue-18">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/inference/generation.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/inference/generation.py</a><br>
<b>Line number: </b>150<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
146 torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
147 torch.backends.cudnn.benchmark = False
148
149 # Conditioning
150 assert conditioning is not None or conditioning_tensors is not None, &quot;Must provide either conditioning or conditioning_tensors&quot;
151 if conditioning_tensors is None:
152 conditioning_tensors = model.conditioner(conditioning, device)
153 conditioning_inputs = model.get_conditioning_inputs(conditioning_tensors)
</pre>
</div>
</div>
</div>
<div id="issue-19">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/inference/generation.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/inference/generation.py</a><br>
<b>Line number: </b>195<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
191 # This is helpful for forward and reverse outpainting
192 cropfrom = math.floor(mask_args[&quot;cropfrom&quot;]/100.0 * sample_size)
193 pastefrom = math.floor(mask_args[&quot;pastefrom&quot;]/100.0 * sample_size)
194 pasteto = math.ceil(mask_args[&quot;pasteto&quot;]/100.0 * sample_size)
195 assert pastefrom &lt; pasteto, &quot;Paste From should be less than Paste To&quot;
196 croplen = pasteto - pastefrom
197 if cropfrom + croplen &gt; sample_size:
198 croplen = sample_size - cropfrom
</pre>
</div>
</div>
</div>
<div id="issue-20">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/interface/gradio.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/interface/gradio.py</a><br>
<b>Line number: </b>659<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
655 return ui
656
657 def create_ui(model_config_path=None, ckpt_path=None, pretrained_name=None, pretransform_ckpt_path=None, model_half=False):
658
659 assert (pretrained_name is not None) ^ (model_config_path is not None and ckpt_path is not None), &quot;Must specify either pretrained name or provide a model config and checkpoint, but not both&quot;
660
661 if model_config_path is not None:
662 # Load config from json file
</pre>
</div>
</div>
</div>
<div id="issue-21">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>103<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
99 If this is the case, we insert extra 0 padding to the right before the reflection happen.
100 &quot;&quot;&quot;
101 length = x.shape[-1]
102 padding_left, padding_right = paddings
103 assert padding_left &gt;= 0 and padding_right &gt;= 0, (padding_left, padding_right)
104 if mode == &#x27;reflect&#x27;:
105 max_pad = max(padding_left, padding_right)
106 extra_pad = 0
</pre>
</div>
</div>
</div>
<div id="issue-22">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>120<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
116
117 def unpad1d(x: torch.Tensor, paddings: tp.Tuple[int, int]):
118 &quot;&quot;&quot;Remove padding from x, handling properly zero padding. Only for 1d!&quot;&quot;&quot;
119 padding_left, padding_right = paddings
120 assert padding_left &gt;= 0 and padding_right &gt;= 0, (padding_left, padding_right)
121 assert (padding_left + padding_right) &lt;= x.shape[-1]
122 end = x.shape[-1] - padding_right
123 return x[..., padding_left: end]
</pre>
</div>
</div>
</div>
<div id="issue-23">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>121<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
117 def unpad1d(x: torch.Tensor, paddings: tp.Tuple[int, int]):
118 &quot;&quot;&quot;Remove padding from x, handling properly zero padding. Only for 1d!&quot;&quot;&quot;
119 padding_left, padding_right = paddings
120 assert padding_left &gt;= 0 and padding_right &gt;= 0, (padding_left, padding_right)
121 assert (padding_left + padding_right) &lt;= x.shape[-1]
122 end = x.shape[-1] - padding_right
123 return x[..., padding_left: end]
124
</pre>
</div>
</div>
</div>
<div id="issue-24">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>177<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
173
174 def Downsample1d(
175 in_channels: int, out_channels: int, factor: int, kernel_multiplier: int = 2
176 ) -&gt; nn.Module:
177 assert kernel_multiplier % 2 == 0, &quot;Kernel multiplier must be even&quot;
178
179 return Conv1d(
180 in_channels=in_channels,
</pre>
</div>
</div>
</div>
<div id="issue-25">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>309<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
305 use_snake=use_snake
306 )
307
308 if self.use_mapping:
309 assert exists(context_mapping_features)
310 self.to_scale_shift = MappingToScaleShift(
311 features=context_mapping_features, channels=out_channels
312 )
</pre>
</div>
</div>
</div>
<div id="issue-26">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>330<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
326 )
327
328 def forward(self, x: Tensor, mapping: Optional[Tensor] = None, causal=False) -&gt; Tensor:
329 assert_message = &quot;context mapping required if context_mapping_features &gt; 0&quot;
330 assert not (self.use_mapping ^ exists(mapping)), assert_message
331
332 h = self.block1(x, causal=causal)
333
</pre>
</div>
</div>
</div>
<div id="issue-27">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>354<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
350 use_snake: bool = False,
351 ):
352 super().__init__()
353 assert_message = f&quot;out_channels must be divisible by patch_size ({patch_size})&quot;
354 assert out_channels % patch_size == 0, assert_message
355 self.patch_size = patch_size
356
357 self.block = ResnetBlock1d(
</pre>
</div>
</div>
</div>
<div id="issue-28">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>382<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
378 use_snake: bool = False
379 ):
380 super().__init__()
381 assert_message = f&quot;in_channels must be divisible by patch_size ({patch_size})&quot;
382 assert in_channels % patch_size == 0, assert_message
383 self.patch_size = patch_size
384
385 self.block = ResnetBlock1d(
</pre>
</div>
</div>
</div>
<div id="issue-29">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>526<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
522 context_mask: Optional[Tensor] = None, # [b, m], false is masked,
523 causal: Optional[bool] = False,
524 ) -&gt; Tensor:
525 assert_message = &quot;You must provide a context when using context_features&quot;
526 assert not self.context_features or exists(context), assert_message
527 # Use context if provided
528 context = default(context, x)
529 # Normalize then compute q from input and k,v from context
</pre>
</div>
</div>
</div>
<div id="issue-30">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>673<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
669 &quot;&quot;&quot;Used for continuous time&quot;&quot;&quot;
670
671 def __init__(self, dim: int):
672 super().__init__()
673 assert (dim % 2) == 0
674 half_dim = dim // 2
675 self.weights = nn.Parameter(torch.randn(half_dim))
676
</pre>
</div>
</div>
</div>
<div id="issue-31">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>749<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
745 ]
746 )
747
748 if self.use_transformer:
749 assert (
750 (exists(attention_heads) or exists(attention_features))
751 and exists(attention_multiplier)
752 )
753
754 if attention_features is None and attention_heads is not None:
755 attention_features = channels // attention_heads
</pre>
</div>
</div>
</div>
<div id="issue-32">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>861<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
857 ]
858 )
859
860 if self.use_transformer:
861 assert (
862 (exists(attention_heads) or exists(attention_features))
863 and exists(attention_multiplier)
864 )
865
866 if attention_features is None and attention_heads is not None:
867 attention_features = channels // attention_heads
</pre>
</div>
</div>
</div>
<div id="issue-33">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>957<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
953 use_snake=use_snake
954 )
955
956 if self.use_transformer:
957 assert (
958 (exists(attention_heads) or exists(attention_features))
959 and exists(attention_multiplier)
960 )
961
962 if attention_features is None and attention_heads is not None:
963 attention_features = channels // attention_heads
</pre>
</div>
</div>
</div>
<div id="issue-34">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1059<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1055 has_context = [c &gt; 0 for c in context_channels]
1056 self.has_context = has_context
1057 self.channels_ids = [sum(has_context[:i]) for i in range(len(has_context))]
1058
1059 assert (
1060 len(factors) == num_layers
1061 and len(attentions) &gt;= num_layers
1062 and len(num_blocks) == num_layers
1063 )
1064
1065 if use_context_time or use_context_features:
1066 context_mapping_features = channels * context_features_multiplier
</pre>
</div>
</div>
</div>
<div id="issue-35">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1076<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1072 nn.GELU(),
1073 )
1074
1075 if use_context_time:
1076 assert exists(context_mapping_features)
1077 self.to_time = nn.Sequential(
1078 TimePositionalEmbedding(
1079 dim=channels, out_features=context_mapping_features
</pre>
</div>
</div>
</div>
<div id="issue-36">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1085<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1081 nn.GELU(),
1082 )
1083
1084 if use_context_features:
1085 assert exists(context_features) and exists(context_mapping_features)
1086 self.to_features = nn.Sequential(
1087 nn.Linear(
1088 in_features=context_features, out_features=context_mapping_features
</pre>
</div>
</div>
</div>
<div id="issue-37">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1095<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1091 )
1092
1093 if use_stft:
1094 stft_kwargs, kwargs = groupby(&quot;stft_&quot;, kwargs)
1095 assert &quot;num_fft&quot; in stft_kwargs, &quot;stft_num_fft required if use_stft=True&quot;
1096 stft_channels = (stft_kwargs[&quot;num_fft&quot;] // 2 + 1) * 2
1097 in_channels *= stft_channels
1098 out_channels *= stft_channels
</pre>
</div>
</div>
</div>
<div id="issue-38">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1100<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1096 stft_channels = (stft_kwargs[&quot;num_fft&quot;] // 2 + 1) * 2
1097 in_channels *= stft_channels
1098 out_channels *= stft_channels
1099 context_channels[0] *= stft_channels if use_stft_context else 1
1100 assert exists(in_channels) and exists(out_channels)
1101 self.stft = STFT(**stft_kwargs)
1102
1103 assert not kwargs, f&quot;Unknown arguments: {&#x27;, &#x27;.join(list(kwargs.keys()))}&quot;
</pre>
</div>
</div>
</div>
<div id="issue-39">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1103<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1099 context_channels[0] *= stft_channels if use_stft_context else 1
1100 assert exists(in_channels) and exists(out_channels)
1101 self.stft = STFT(**stft_kwargs)
1102
1103 assert not kwargs, f&quot;Unknown arguments: {&#x27;, &#x27;.join(list(kwargs.keys()))}&quot;
1104
1105 self.to_in = Patcher(
1106 in_channels=in_channels + context_channels[0],
</pre>
</div>
</div>
</div>
<div id="issue-40">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1183<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1179 &quot;&quot;&quot;Gets context channels at `layer` and checks that shape is correct&quot;&quot;&quot;
1180 use_context_channels = self.use_context_channels and self.has_context[layer]
1181 if not use_context_channels:
1182 return None
1183 assert exists(channels_list), &quot;Missing context&quot;
1184 # Get channels index (skipping zero channel contexts)
1185 channels_id = self.channels_ids[layer]
1186 # Get channels
</pre>
</div>
</div>
</div>
<div id="issue-41">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1189<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1185 channels_id = self.channels_ids[layer]
1186 # Get channels
1187 channels = channels_list[channels_id]
1188 message = f&quot;Missing context for layer {layer} at index {channels_id}&quot;
1189 assert exists(channels), message
1190 # Check channels
1191 num_channels = self.context_channels[layer]
1192 message = f&quot;Expected context with {num_channels} channels at idx {channels_id}&quot;
</pre>
</div>
</div>
</div>
<div id="issue-42">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1193<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1189 assert exists(channels), message
1190 # Check channels
1191 num_channels = self.context_channels[layer]
1192 message = f&quot;Expected context with {num_channels} channels at idx {channels_id}&quot;
1193 assert channels.shape[1] == num_channels, message
1194 # STFT channels if requested
1195 channels = self.stft.encode1d(channels) if self.use_stft_context else channels # type: ignore # noqa
1196 return channels
</pre>
</div>
</div>
</div>
<div id="issue-43">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1206<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1202 items, mapping = [], None
1203 # Compute time features
1204 if self.use_context_time:
1205 assert_message = &quot;use_context_time=True but no time features provided&quot;
1206 assert exists(time), assert_message
1207 items += [self.to_time(time)]
1208 # Compute features
1209 if self.use_context_features:
</pre>
</div>
</div>
</div>
<div id="issue-44">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1211<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1207 items += [self.to_time(time)]
1208 # Compute features
1209 if self.use_context_features:
1210 assert_message = &quot;context_features exists but no features provided&quot;
1211 assert exists(features), assert_message
1212 items += [self.to_features(features)]
1213 # Compute joint mapping
1214 if self.use_context_time or self.use_context_features:
</pre>
</div>
</div>
</div>
<div id="issue-45">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1272<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1268
1269 def forward(self, x: Tensor) -&gt; Tensor:
1270 batch_size, length, device = *x.shape[0:2], x.device
1271 assert_message = &quot;Input sequence length must be &lt;= max_length&quot;
1272 assert length &lt;= self.max_length, assert_message
1273 position = torch.arange(length, device=device)
1274 fixed_embedding = self.embedding(position)
1275 fixed_embedding = repeat(fixed_embedding, &quot;n d -&gt; b n d&quot;, b=batch_size)
</pre>
</div>
</div>
</div>
<div id="issue-46">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1306<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1302
1303 self.use_xattn_time = use_xattn_time
1304
1305 if use_xattn_time:
1306 assert exists(context_embedding_features)
1307 self.to_time_embedding = nn.Sequential(
1308 TimePositionalEmbedding(
1309 dim=kwargs[&quot;channels&quot;], out_features=context_embedding_features
</pre>
</div>
</div>
</div>
<div id="issue-47">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/adp.py</a><br>
<b>Line number: </b>1495<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
1491 def forward(self, x: Union[List[float], Tensor]) -&gt; Tensor:
1492 if not torch.is_tensor(x):
1493 device = next(self.embedding.parameters()).device
1494 x = torch.tensor(x, device=device)
1495 assert isinstance(x, Tensor)
1496 shape = x.shape
1497 x = rearrange(x, &quot;... -&gt; (...)&quot;)
1498 embedding = self.embedding(x)
</pre>
</div>
</div>
</div>
<div id="issue-48">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>369<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
365 Decode discrete tokens to audio
366 Only works with discrete autoencoders
367 &#x27;&#x27;&#x27;
368
369 assert isinstance(self.bottleneck, DiscreteBottleneck), &quot;decode_tokens only works with discrete autoencoders&quot;
370
371 latents = self.bottleneck.decode_tokens(tokens, **kwargs)
372
</pre>
</div>
</div>
</div>
<div id="issue-49">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>399<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
395 &#x27;&#x27;&#x27;
396 batch_size = len(audio_list)
397 if isinstance(in_sr_list, int):
398 in_sr_list = [in_sr_list]*batch_size
399 assert len(in_sr_list) == batch_size, &quot;list of sample rates must be the same length of audio_list&quot;
400 new_audio = []
401 max_length = 0
402 # resample &amp; find the max length
</pre>
</div>
</div>
</div>
<div id="issue-50">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>412<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
408 audio = audio.squeeze(0)
409 elif len(audio.shape) == 1:
410 # Mono signal, channel dimension is missing, unsqueeze it in
411 audio = audio.unsqueeze(0)
412 assert len(audio.shape)==2, &quot;Audio should be shape (Channels x Length) with no batch dimension&quot;
413 # Resample audio
414 if in_sr != self.sample_rate:
415 resample_tf = T.Resample(in_sr, self.sample_rate).to(audio.device)
</pre>
</div>
</div>
</div>
<div id="issue-51">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>613<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
609 # AE factories
610
611 def create_encoder_from_config(encoder_config: Dict[str, Any]):
612 encoder_type = encoder_config.get(&quot;type&quot;, None)
613 assert encoder_type is not None, &quot;Encoder type must be specified&quot;
614
615 if encoder_type == &quot;oobleck&quot;:
616 encoder = OobleckEncoder(
</pre>
</div>
</div>
</div>
<div id="issue-52">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>653<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
649 return encoder
650
651 def create_decoder_from_config(decoder_config: Dict[str, Any]):
652 decoder_type = decoder_config.get(&quot;type&quot;, None)
653 assert decoder_type is not None, &quot;Decoder type must be specified&quot;
654
655 if decoder_type == &quot;oobleck&quot;:
656 decoder = OobleckDecoder(
</pre>
</div>
</div>
</div>
<div id="issue-53">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>697<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
693
694 bottleneck = ae_config.get(&quot;bottleneck&quot;, None)
695
696 latent_dim = ae_config.get(&quot;latent_dim&quot;, None)
697 assert latent_dim is not None, &quot;latent_dim must be specified in model config&quot;
698 downsampling_ratio = ae_config.get(&quot;downsampling_ratio&quot;, None)
699 assert downsampling_ratio is not None, &quot;downsampling_ratio must be specified in model config&quot;
700 io_channels = ae_config.get(&quot;io_channels&quot;, None)
</pre>
</div>
</div>
</div>
<div id="issue-54">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>699<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
695
696 latent_dim = ae_config.get(&quot;latent_dim&quot;, None)
697 assert latent_dim is not None, &quot;latent_dim must be specified in model config&quot;
698 downsampling_ratio = ae_config.get(&quot;downsampling_ratio&quot;, None)
699 assert downsampling_ratio is not None, &quot;downsampling_ratio must be specified in model config&quot;
700 io_channels = ae_config.get(&quot;io_channels&quot;, None)
701 assert io_channels is not None, &quot;io_channels must be specified in model config&quot;
702 sample_rate = config.get(&quot;sample_rate&quot;, None)
</pre>
</div>
</div>
</div>
<div id="issue-55">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>701<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
697 assert latent_dim is not None, &quot;latent_dim must be specified in model config&quot;
698 downsampling_ratio = ae_config.get(&quot;downsampling_ratio&quot;, None)
699 assert downsampling_ratio is not None, &quot;downsampling_ratio must be specified in model config&quot;
700 io_channels = ae_config.get(&quot;io_channels&quot;, None)
701 assert io_channels is not None, &quot;io_channels must be specified in model config&quot;
702 sample_rate = config.get(&quot;sample_rate&quot;, None)
703 assert sample_rate is not None, &quot;sample_rate must be specified in model config&quot;
704
</pre>
</div>
</div>
</div>
<div id="issue-56">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>703<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
699 assert downsampling_ratio is not None, &quot;downsampling_ratio must be specified in model config&quot;
700 io_channels = ae_config.get(&quot;io_channels&quot;, None)
701 assert io_channels is not None, &quot;io_channels must be specified in model config&quot;
702 sample_rate = config.get(&quot;sample_rate&quot;, None)
703 assert sample_rate is not None, &quot;sample_rate must be specified in model config&quot;
704
705 in_channels = ae_config.get(&quot;in_channels&quot;, None)
706 out_channels = ae_config.get(&quot;out_channels&quot;, None)
</pre>
</div>
</div>
</div>
<div id="issue-57">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>756<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
752 elif diffusion_model_type == &quot;dit&quot;:
753 diffusion = DiTWrapper(**diffae_config[&quot;diffusion&quot;][&quot;config&quot;])
754
755 latent_dim = diffae_config.get(&quot;latent_dim&quot;, None)
756 assert latent_dim is not None, &quot;latent_dim must be specified in model config&quot;
757 downsampling_ratio = diffae_config.get(&quot;downsampling_ratio&quot;, None)
758 assert downsampling_ratio is not None, &quot;downsampling_ratio must be specified in model config&quot;
759 io_channels = diffae_config.get(&quot;io_channels&quot;, None)
</pre>
</div>
</div>
</div>
<div id="issue-58">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>758<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
754
755 latent_dim = diffae_config.get(&quot;latent_dim&quot;, None)
756 assert latent_dim is not None, &quot;latent_dim must be specified in model config&quot;
757 downsampling_ratio = diffae_config.get(&quot;downsampling_ratio&quot;, None)
758 assert downsampling_ratio is not None, &quot;downsampling_ratio must be specified in model config&quot;
759 io_channels = diffae_config.get(&quot;io_channels&quot;, None)
760 assert io_channels is not None, &quot;io_channels must be specified in model config&quot;
761 sample_rate = config.get(&quot;sample_rate&quot;, None)
</pre>
</div>
</div>
</div>
<div id="issue-59">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>760<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
756 assert latent_dim is not None, &quot;latent_dim must be specified in model config&quot;
757 downsampling_ratio = diffae_config.get(&quot;downsampling_ratio&quot;, None)
758 assert downsampling_ratio is not None, &quot;downsampling_ratio must be specified in model config&quot;
759 io_channels = diffae_config.get(&quot;io_channels&quot;, None)
760 assert io_channels is not None, &quot;io_channels must be specified in model config&quot;
761 sample_rate = config.get(&quot;sample_rate&quot;, None)
762 assert sample_rate is not None, &quot;sample_rate must be specified in model config&quot;
763
</pre>
</div>
</div>
</div>
<div id="issue-60">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/autoencoders.py</a><br>
<b>Line number: </b>762<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
758 assert downsampling_ratio is not None, &quot;downsampling_ratio must be specified in model config&quot;
759 io_channels = diffae_config.get(&quot;io_channels&quot;, None)
760 assert io_channels is not None, &quot;io_channels must be specified in model config&quot;
761 sample_rate = config.get(&quot;sample_rate&quot;, None)
762 assert sample_rate is not None, &quot;sample_rate must be specified in model config&quot;
763
764 bottleneck = diffae_config.get(&quot;bottleneck&quot;, None)
765
</pre>
</div>
</div>
</div>
<div id="issue-61">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/blocks.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/blocks.py</a><br>
<b>Line number: </b>37<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
33
34 class SelfAttention1d(nn.Module):
35 def __init__(self, c_in, n_head=1, dropout_rate=0.):
36 super().__init__()
37 assert c_in % n_head == 0
38 self.norm = nn.GroupNorm(1, c_in)
39 self.n_head = n_head
40 self.qkv_proj = nn.Conv1d(c_in, c_in * 3, 1)
</pre>
</div>
</div>
</div>
<div id="issue-62">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/blocks.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/blocks.py</a><br>
<b>Line number: </b>87<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
83
84 class FourierFeatures(nn.Module):
85 def __init__(self, in_features, out_features, std=1.):
86 super().__init__()
87 assert out_features % 2 == 0
88 self.weight = nn.Parameter(torch.randn(
89 [out_features // 2, in_features]) * std)
90
</pre>
</div>
</div>
</div>
<div id="issue-63">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/blocks.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/blocks.py</a><br>
<b>Line number: </b>157<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
153
154 def Downsample1d_2(
155 in_channels: int, out_channels: int, factor: int, kernel_multiplier: int = 2
156 ) -&gt; nn.Module:
157 assert kernel_multiplier % 2 == 0, &quot;Kernel multiplier must be even&quot;
158
159 return nn.Conv1d(
160 in_channels=in_channels,
</pre>
</div>
</div>
</div>
<div id="issue-64">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>48<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
44 timesteps: int
45 n_q: int
46
47 def __post_init__(self):
48 assert len(self.layout) &gt; 0
49 self._validate_layout()
50 self._build_reverted_sequence_scatter_indexes = lru_cache(100)(self._build_reverted_sequence_scatter_indexes)
51 self._build_pattern_sequence_scatter_indexes = lru_cache(100)(self._build_pattern_sequence_scatter_indexes)
</pre>
</div>
</div>
</div>
<div id="issue-65">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>68<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
64 qs = set()
65 for coord in seq_coords:
66 qs.add(coord.q)
67 last_q_timestep = q_timesteps[coord.q]
68 assert coord.t &gt;= last_q_timestep, \
69 f&quot;Past timesteps are found in the sequence for codebook = {coord.q} at step {s}&quot;
70 q_timesteps[coord.q] = coord.t
71 # each sequence step contains at max 1 coordinate per codebook
72 assert len(qs) == len(seq_coords), \
</pre>
</div>
</div>
</div>
<div id="issue-66">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>72<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
68 assert coord.t &gt;= last_q_timestep, \
69 f&quot;Past timesteps are found in the sequence for codebook = {coord.q} at step {s}&quot;
70 q_timesteps[coord.q] = coord.t
71 # each sequence step contains at max 1 coordinate per codebook
72 assert len(qs) == len(seq_coords), \
73 f&quot;Multiple entries for a same codebook are found at step {s}&quot;
74
75 @property
76 def num_sequence_steps(self):
</pre>
</div>
</div>
</div>
<div id="issue-67">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>100<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
96 &quot;&quot;&quot;Get codebook coordinates in the layout that corresponds to the specified timestep t
97 and optionally to the codebook q. Coordinates are returned as a tuple with the sequence step
98 and the actual codebook coordinates.
99 &quot;&quot;&quot;
100 assert t &lt;= self.timesteps, &quot;provided timesteps is greater than the pattern&#x27;s number of timesteps&quot;
101 if q is not None:
102 assert q &lt;= self.n_q, &quot;provided number of codebooks is greater than the pattern&#x27;s number of codebooks&quot;
103 coords = []
</pre>
</div>
</div>
</div>
<div id="issue-68">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>102<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
98 and the actual codebook coordinates.
99 &quot;&quot;&quot;
100 assert t &lt;= self.timesteps, &quot;provided timesteps is greater than the pattern&#x27;s number of timesteps&quot;
101 if q is not None:
102 assert q &lt;= self.n_q, &quot;provided number of codebooks is greater than the pattern&#x27;s number of codebooks&quot;
103 coords = []
104 for s, seq_codes in enumerate(self.layout):
105 for code in seq_codes:
</pre>
</div>
</div>
</div>
<div id="issue-69">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>129<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
125 Returns:
126 indexes (torch.Tensor): Indexes corresponding to the sequence, of shape [K, S].
127 mask (torch.Tensor): Mask corresponding to indexes that matches valid indexes, of shape [K, S].
128 &quot;&quot;&quot;
129 assert n_q == self.n_q, f&quot;invalid number of codebooks for the sequence and the pattern: {n_q} != {self.n_q}&quot;
130 assert timesteps &lt;= self.timesteps, &quot;invalid number of timesteps used to build the sequence from the pattern&quot;
131 # use the proper layout based on whether we limit ourselves to valid steps only or not,
132 # note that using the valid_layout will result in a truncated sequence up to the valid steps
</pre>
</div>
</div>
</div>
<div id="issue-70">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>130<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
126 indexes (torch.Tensor): Indexes corresponding to the sequence, of shape [K, S].
127 mask (torch.Tensor): Mask corresponding to indexes that matches valid indexes, of shape [K, S].
128 &quot;&quot;&quot;
129 assert n_q == self.n_q, f&quot;invalid number of codebooks for the sequence and the pattern: {n_q} != {self.n_q}&quot;
130 assert timesteps &lt;= self.timesteps, &quot;invalid number of timesteps used to build the sequence from the pattern&quot;
131 # use the proper layout based on whether we limit ourselves to valid steps only or not,
132 # note that using the valid_layout will result in a truncated sequence up to the valid steps
133 ref_layout = self.valid_layout if keep_only_valid_steps else self.layout
</pre>
</div>
</div>
</div>
<div id="issue-71">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>199<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
195 &quot;&quot;&quot;
196 ref_layout = self.valid_layout if keep_only_valid_steps else self.layout
197 # TODO(jade): Do we want to further truncate to only valid timesteps here as well?
198 timesteps = self.timesteps
199 assert n_q == self.n_q, f&quot;invalid number of codebooks for the sequence and the pattern: {n_q} != {self.n_q}&quot;
200 assert sequence_steps &lt;= len(ref_layout), \
201 f&quot;sequence to revert is longer than the defined pattern: {sequence_steps} &gt; {len(ref_layout)}&quot;
202
</pre>
</div>
</div>
</div>
<div id="issue-72">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>200<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
196 ref_layout = self.valid_layout if keep_only_valid_steps else self.layout
197 # TODO(jade): Do we want to further truncate to only valid timesteps here as well?
198 timesteps = self.timesteps
199 assert n_q == self.n_q, f&quot;invalid number of codebooks for the sequence and the pattern: {n_q} != {self.n_q}&quot;
200 assert sequence_steps &lt;= len(ref_layout), \
201 f&quot;sequence to revert is longer than the defined pattern: {sequence_steps} &gt; {len(ref_layout)}&quot;
202
203 # ensure we take the appropriate indexes to keep the model output from the first special token as well
204 if is_model_output and self.starts_with_special_token():
</pre>
</div>
</div>
</div>
<div id="issue-73">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>288<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
284 cached (bool): if True, patterns for a given length are cached. In general
285 that should be true for efficiency reason to avoid synchronization points.
286 &quot;&quot;&quot;
287 def __init__(self, n_q: int, cached: bool = True):
288 assert n_q &gt; 0
289 self.n_q = n_q
290 self.get_pattern = lru_cache(100)(self.get_pattern) # type: ignore
291
</pre>
</div>
</div>
</div>
<div id="issue-74">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>333<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
329 delays = list(range(n_q))
330 self.delays = delays
331 self.flatten_first = flatten_first
332 self.empty_initial = empty_initial
333 assert len(self.delays) == self.n_q
334 assert sorted(self.delays) == self.delays
335
336 def get_pattern(self, timesteps: int) -&gt; Pattern:
</pre>
</div>
</div>
</div>
<div id="issue-75">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>334<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
330 self.delays = delays
331 self.flatten_first = flatten_first
332 self.empty_initial = empty_initial
333 assert len(self.delays) == self.n_q
334 assert sorted(self.delays) == self.delays
335
336 def get_pattern(self, timesteps: int) -&gt; Pattern:
337 omit_special_token = self.empty_initial &lt; 0
</pre>
</div>
</div>
</div>
<div id="issue-76">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>427<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
423 if flattening is None:
424 flattening = list(range(n_q))
425 if delays is None:
426 delays = [0] * n_q
427 assert len(flattening) == n_q
428 assert len(delays) == n_q
429 assert sorted(flattening) == flattening
430 assert sorted(delays) == delays
</pre>
</div>
</div>
</div>
<div id="issue-77">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>428<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
424 flattening = list(range(n_q))
425 if delays is None:
426 delays = [0] * n_q
427 assert len(flattening) == n_q
428 assert len(delays) == n_q
429 assert sorted(flattening) == flattening
430 assert sorted(delays) == delays
431 self._flattened_codebooks = self._build_flattened_codebooks(delays, flattening)
</pre>
</div>
</div>
</div>
<div id="issue-78">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>429<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
425 if delays is None:
426 delays = [0] * n_q
427 assert len(flattening) == n_q
428 assert len(delays) == n_q
429 assert sorted(flattening) == flattening
430 assert sorted(delays) == delays
431 self._flattened_codebooks = self._build_flattened_codebooks(delays, flattening)
432 self.max_delay = max(delays)
</pre>
</div>
</div>
</div>
<div id="issue-79">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>430<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
426 delays = [0] * n_q
427 assert len(flattening) == n_q
428 assert len(delays) == n_q
429 assert sorted(flattening) == flattening
430 assert sorted(delays) == delays
431 self._flattened_codebooks = self._build_flattened_codebooks(delays, flattening)
432 self.max_delay = max(delays)
433
</pre>
</div>
</div>
</div>
<div id="issue-80">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>445<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
441 if inner_step not in flattened_codebooks:
442 flat_codebook = UnrolledPatternProvider.FlattenedCodebook(codebooks=[q], delay=delay)
443 else:
444 flat_codebook = flattened_codebooks[inner_step]
445 assert flat_codebook.delay == delay, (
446 &quot;Delay and flattening between codebooks is inconsistent: &quot;,
447 &quot;two codebooks flattened to the same position should have the same delay.&quot;
448 )
449 flat_codebook.codebooks.append(q)
450 flattened_codebooks[inner_step] = flat_codebook
451 return flattened_codebooks
</pre>
</div>
</div>
</div>
<div id="issue-81">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>509<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
505 super().__init__(n_q)
506 if delays is None:
507 delays = [0] * (n_q - 1)
508 self.delays = delays
509 assert len(self.delays) == self.n_q - 1
510 assert sorted(self.delays) == self.delays
511
512 def get_pattern(self, timesteps: int) -&gt; Pattern:
</pre>
</div>
</div>
</div>
<div id="issue-82">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/codebook_patterns.py</a><br>
<b>Line number: </b>510<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
506 if delays is None:
507 delays = [0] * (n_q - 1)
508 self.delays = delays
509 assert len(self.delays) == self.n_q - 1
510 assert sorted(self.delays) == self.delays
511
512 def get_pattern(self, timesteps: int) -&gt; Pattern:
513 out: PatternLayout = [[]]
</pre>
</div>
</div>
</div>
<div id="issue-83">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/conditioners.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/conditioners.py</a><br>
<b>Line number: </b>269<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
265 max_length: str = 128,
266 enable_grad: bool = False,
267 project_out: bool = False
268 ):
269 assert t5_model_name in self.T5_MODELS, f&quot;Unknown T5 model name: {t5_model_name}&quot;
270 super().__init__(self.T5_MODEL_DIMS[t5_model_name], output_dim, project_out=project_out)
271
272 from transformers import T5EncoderModel, AutoTokenizer
</pre>
</div>
</div>
</div>
<div id="issue-84">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/conditioners.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/conditioners.py</a><br>
<b>Line number: </b>550<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
546 elif conditioner_type == &quot;lut&quot;:
547 conditioners[id] = TokenizerLUTConditioner(**conditioner_config)
548 elif conditioner_type == &quot;pretransform&quot;:
549 sample_rate = conditioner_config.pop(&quot;sample_rate&quot;, None)
550 assert sample_rate is not None, &quot;Sample rate must be specified for pretransform conditioners&quot;
551
552 pretransform = create_pretransform_from_config(conditioner_config.pop(&quot;pretransform_config&quot;), sample_rate=sample_rate)
553
</pre>
</div>
</div>
</div>
<div id="issue-85">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>529<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
525 rescale_cfg: bool = False,
526 scale_phi: float = 0.0,
527 **kwargs):
528
529 assert batch_cfg, &quot;batch_cfg must be True for DiTWrapper&quot;
530 #assert negative_input_concat_cond is None, &quot;negative_input_concat_cond is not supported for DiTWrapper&quot;
531
532 return self.model(
</pre>
</div>
</div>
</div>
<div id="issue-86">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>575<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
571 model_type = diffusion_uncond_config.get(&#x27;type&#x27;, None)
572
573 diffusion_config = diffusion_uncond_config.get(&#x27;config&#x27;, {})
574
575 assert model_type is not None, &quot;Must specify model type in config&quot;
576
577 pretransform = diffusion_uncond_config.get(&quot;pretransform&quot;, None)
578
</pre>
</div>
</div>
</div>
<div id="issue-87">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>580<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
576
577 pretransform = diffusion_uncond_config.get(&quot;pretransform&quot;, None)
578
579 sample_size = config.get(&quot;sample_size&quot;, None)
580 assert sample_size is not None, &quot;Must specify sample size in config&quot;
581
582 sample_rate = config.get(&quot;sample_rate&quot;, None)
583 assert sample_rate is not None, &quot;Must specify sample rate in config&quot;
</pre>
</div>
</div>
</div>
<div id="issue-88">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>583<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
579 sample_size = config.get(&quot;sample_size&quot;, None)
580 assert sample_size is not None, &quot;Must specify sample size in config&quot;
581
582 sample_rate = config.get(&quot;sample_rate&quot;, None)
583 assert sample_rate is not None, &quot;Must specify sample rate in config&quot;
584
585 if pretransform is not None:
586 pretransform = create_pretransform_from_config(pretransform, sample_rate)
</pre>
</div>
</div>
</div>
<div id="issue-89">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>625<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
621
622 model_type = config[&quot;model_type&quot;]
623
624 diffusion_config = model_config.get(&#x27;diffusion&#x27;, None)
625 assert diffusion_config is not None, &quot;Must specify diffusion config&quot;
626
627 diffusion_model_type = diffusion_config.get(&#x27;type&#x27;, None)
628 assert diffusion_model_type is not None, &quot;Must specify diffusion model type&quot;
</pre>
</div>
</div>
</div>
<div id="issue-90">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>628<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
624 diffusion_config = model_config.get(&#x27;diffusion&#x27;, None)
625 assert diffusion_config is not None, &quot;Must specify diffusion config&quot;
626
627 diffusion_model_type = diffusion_config.get(&#x27;type&#x27;, None)
628 assert diffusion_model_type is not None, &quot;Must specify diffusion model type&quot;
629
630 diffusion_model_config = diffusion_config.get(&#x27;config&#x27;, None)
631 assert diffusion_model_config is not None, &quot;Must specify diffusion model config&quot;
</pre>
</div>
</div>
</div>
<div id="issue-91">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>631<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
627 diffusion_model_type = diffusion_config.get(&#x27;type&#x27;, None)
628 assert diffusion_model_type is not None, &quot;Must specify diffusion model type&quot;
629
630 diffusion_model_config = diffusion_config.get(&#x27;config&#x27;, None)
631 assert diffusion_model_config is not None, &quot;Must specify diffusion model config&quot;
632
633 if diffusion_model_type == &#x27;adp_cfg_1d&#x27;:
634 diffusion_model = UNetCFG1DWrapper(**diffusion_model_config)
</pre>
</div>
</div>
</div>
<div id="issue-92">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>641<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
637 elif diffusion_model_type == &#x27;dit&#x27;:
638 diffusion_model = DiTWrapper(**diffusion_model_config)
639
640 io_channels = model_config.get(&#x27;io_channels&#x27;, None)
641 assert io_channels is not None, &quot;Must specify io_channels in model config&quot;
642
643 sample_rate = config.get(&#x27;sample_rate&#x27;, None)
644 assert sample_rate is not None, &quot;Must specify sample_rate in config&quot;
</pre>
</div>
</div>
</div>
<div id="issue-93">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>644<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
640 io_channels = model_config.get(&#x27;io_channels&#x27;, None)
641 assert io_channels is not None, &quot;Must specify io_channels in model config&quot;
642
643 sample_rate = config.get(&#x27;sample_rate&#x27;, None)
644 assert sample_rate is not None, &quot;Must specify sample_rate in config&quot;
645
646 diffusion_objective = diffusion_config.get(&#x27;diffusion_objective&#x27;, &#x27;v&#x27;)
647
</pre>
</div>
</div>
</div>
<div id="issue-94">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/diffusion.py</a><br>
<b>Line number: </b>683<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
679 extra_kwargs[&quot;diffusion_objective&quot;] = diffusion_objective
680
681 elif model_type == &quot;diffusion_prior&quot;:
682 prior_type = model_config.get(&quot;prior_type&quot;, None)
683 assert prior_type is not None, &quot;Must specify prior_type in diffusion prior model config&quot;
684
685 if prior_type == &quot;mono_stereo&quot;:
686 from .diffusion_prior import MonoToStereoDiffusionPrior
</pre>
</div>
</div>
</div>
<div id="issue-95">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/dit.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/dit.py</a><br>
<b>Line number: </b>252<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
248 mask=None,
249 return_info=False,
250 **kwargs):
251
252 assert causal == False, &quot;Causal mode is not supported for DiffusionTransformer&quot;
253
254 if cross_attn_cond_mask is not None:
255 cross_attn_cond_mask = cross_attn_cond_mask.bool()
</pre>
</div>
</div>
</div>
<div id="issue-96">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/factory.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/factory.py</a><br>
<b>Line number: </b>6<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
2
3 def create_model_from_config(model_config):
4 model_type = model_config.get(&#x27;model_type&#x27;, None)
5
6 assert model_type is not None, &#x27;model_type must be specified in model config&#x27;
7
8 if model_type == &#x27;autoencoder&#x27;:
9 from .autoencoders import create_autoencoder_from_config
</pre>
</div>
</div>
</div>
<div id="issue-97">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/factory.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/factory.py</a><br>
<b>Line number: </b>35<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
31
32 def create_pretransform_from_config(pretransform_config, sample_rate):
33 pretransform_type = pretransform_config.get(&#x27;type&#x27;, None)
34
35 assert pretransform_type is not None, &#x27;type must be specified in pretransform config&#x27;
36
37 if pretransform_type == &#x27;autoencoder&#x27;:
38 from .autoencoders import create_autoencoder_from_config
</pre>
</div>
</div>
</div>
<div id="issue-98">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/factory.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/factory.py</a><br>
<b>Line number: </b>87<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
83
84 def create_bottleneck_from_config(bottleneck_config):
85 bottleneck_type = bottleneck_config.get(&#x27;type&#x27;, None)
86
87 assert bottleneck_type is not None, &#x27;type must be specified in bottleneck config&#x27;
88
89 if bottleneck_type == &#x27;tanh&#x27;:
90 from .bottleneck import TanhBottleneck
</pre>
</div>
</div>
</div>
<div id="issue-99">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>70<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
66 ):
67
68 batch, num_quantizers, seq_len = sequence.shape
69
70 assert num_quantizers == self.num_quantizers, &quot;Number of quantizers in sequence must match number of quantizers in model&quot;
71
72 backbone_input = sum([self.embeds[i](sequence[:, i]) for i in range(num_quantizers)]) # [batch, seq_len, embed_dim]
73
</pre>
</div>
</div>
</div>
<div id="issue-100">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>154<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
150 global_cond_ids: tp.List[str] = []
151 ):
152 super().__init__()
153
154 assert pretransform.is_discrete, &quot;Pretransform must be discrete&quot;
155 self.pretransform = pretransform
156
157 self.pretransform.requires_grad_(False)
</pre>
</div>
</div>
</div>
<div id="issue-101">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>374<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
370 possible_batch_sizes.append(conditioning_tensors[list(conditioning_tensors.keys())[0]][0].shape[0])
371 else:
372 possible_batch_sizes.append(1)
373
374 assert [x == possible_batch_sizes[0] for x in possible_batch_sizes], &quot;Batch size must be consistent across inputs&quot;
375
376 batch_size = possible_batch_sizes[0]
377
</pre>
</div>
</div>
</div>
<div id="issue-102">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>380<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
376 batch_size = possible_batch_sizes[0]
377
378 if init_data is None:
379 # Initialize with zeros
380 assert batch_size &gt; 0
381 init_data = torch.zeros((batch_size, self.num_quantizers, 0), device=device, dtype=torch.long)
382
383 batch_size, num_quantizers, seq_len = init_data.shape
</pre>
</div>
</div>
</div>
<div id="issue-103">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>386<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
382
383 batch_size, num_quantizers, seq_len = init_data.shape
384
385 start_offset = seq_len
386 assert start_offset &lt; max_gen_len, &quot;init data longer than max gen length&quot;
387
388 pattern = self.lm.pattern_provider.get_pattern(max_gen_len)
389
</pre>
</div>
</div>
</div>
<div id="issue-104">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>399<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
395
396 gen_sequence, _, mask = pattern.build_pattern_sequence(gen_codes, self.lm.masked_token_id) # [batch, num_quantizers, gen_sequence_len]
397
398 start_offset_sequence = pattern.get_first_step_with_timesteps(start_offset)
399 assert start_offset_sequence is not None
400
401 # Generation
402 prev_offset = 0
</pre>
</div>
</div>
</div>
<div id="issue-105">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>443<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
439 # Callback to report progress
440 # Pass in the offset relative to the start of the sequence, and the length of the current sequence
441 callback(1 + offset - start_offset_sequence, gen_sequence_len - start_offset_sequence)
442
443 assert not (gen_sequence == unknown_token).any(), &quot;Unknown tokens in generated sequence&quot;
444
445 out_codes, _, out_mask = pattern.revert_pattern_sequence(gen_sequence, special_token=unknown_token)
446
</pre>
</div>
</div>
</div>
<div id="issue-106">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>448<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
444
445 out_codes, _, out_mask = pattern.revert_pattern_sequence(gen_sequence, special_token=unknown_token)
446
447 # sanity checks over the returned codes and corresponding masks
448 assert (out_codes[..., :max_gen_len] != unknown_token).all()
449 assert (out_mask[..., :max_gen_len] == 1).all()
450
451 #out_codes = out_codes[..., 0:max_gen_len]
</pre>
</div>
</div>
</div>
<div id="issue-107">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>449<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
445 out_codes, _, out_mask = pattern.revert_pattern_sequence(gen_sequence, special_token=unknown_token)
446
447 # sanity checks over the returned codes and corresponding masks
448 assert (out_codes[..., :max_gen_len] != unknown_token).all()
449 assert (out_mask[..., :max_gen_len] == 1).all()
450
451 #out_codes = out_codes[..., 0:max_gen_len]
452
</pre>
</div>
</div>
</div>
<div id="issue-108">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>473<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
469
470
471 def create_audio_lm_from_config(config):
472 model_config = config.get(&#x27;model&#x27;, None)
473 assert model_config is not None, &#x27;model config must be specified in config&#x27;
474
475 sample_rate = config.get(&#x27;sample_rate&#x27;, None)
476 assert sample_rate is not None, &quot;Must specify sample_rate in config&quot;
</pre>
</div>
</div>
</div>
<div id="issue-109">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>476<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
472 model_config = config.get(&#x27;model&#x27;, None)
473 assert model_config is not None, &#x27;model config must be specified in config&#x27;
474
475 sample_rate = config.get(&#x27;sample_rate&#x27;, None)
476 assert sample_rate is not None, &quot;Must specify sample_rate in config&quot;
477
478 lm_config = model_config.get(&#x27;lm&#x27;, None)
479 assert lm_config is not None, &#x27;lm config must be specified in model config&#x27;
</pre>
</div>
</div>
</div>
<div id="issue-110">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>479<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
475 sample_rate = config.get(&#x27;sample_rate&#x27;, None)
476 assert sample_rate is not None, &quot;Must specify sample_rate in config&quot;
477
478 lm_config = model_config.get(&#x27;lm&#x27;, None)
479 assert lm_config is not None, &#x27;lm config must be specified in model config&#x27;
480
481 codebook_pattern = lm_config.get(&quot;codebook_pattern&quot;, &quot;delay&quot;)
482
</pre>
</div>
</div>
</div>
<div id="issue-111">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>494<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
490 pretransform_config = model_config.get(&quot;pretransform&quot;, None)
491
492 pretransform = create_pretransform_from_config(pretransform_config, sample_rate)
493
494 assert pretransform.is_discrete, &quot;Pretransform must be discrete&quot;
495
496 min_input_length = pretransform.downsampling_ratio
497
</pre>
</div>
</div>
</div>
<div id="issue-112">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>513<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
509
510 lm_type = lm_config.get(&quot;type&quot;, None)
511 lm_model_config = lm_config.get(&quot;config&quot;, None)
512
513 assert lm_type is not None, &quot;Must specify lm type in lm config&quot;
514 assert lm_model_config is not None, &quot;Must specify lm model config in lm config&quot;
515
516 if lm_type == &quot;x-transformers&quot;:
</pre>
</div>
</div>
</div>
<div id="issue-113">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/lm.py</a><br>
<b>Line number: </b>514<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
510 lm_type = lm_config.get(&quot;type&quot;, None)
511 lm_model_config = lm_config.get(&quot;config&quot;, None)
512
513 assert lm_type is not None, &quot;Must specify lm type in lm config&quot;
514 assert lm_model_config is not None, &quot;Must specify lm model config in lm config&quot;
515
516 if lm_type == &quot;x-transformers&quot;:
517 backbone = XTransformersAudioLMBackbone(**lm_model_config)
</pre>
</div>
</div>
</div>
<div id="issue-114">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pqmf.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pqmf.py</a><br>
<b>Line number: </b>24<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
20 super(PQMF, self).__init__()
21
22 # Ensure num_bands is a power of 2
23 is_power_of_2 = (math.log2(num_bands) == int(math.log2(num_bands)))
24 assert is_power_of_2, &quot;&#x27;num_bands&#x27; must be a power of 2.&quot;
25
26 # Create the prototype filter
27 prototype_filter = design_prototype_filter(attenuation, num_bands)
</pre>
</div>
</div>
</div>
<div id="issue-115">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pretransforms.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pretransforms.py</a><br>
<b>Line number: </b>78<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
74
75 return decoded
76
77 def tokenize(self, x, **kwargs):
78 assert self.model.is_discrete, &quot;Cannot tokenize with a continuous model&quot;
79
80 _, info = self.model.encode(x, return_info = True, **kwargs)
81
</pre>
</div>
</div>
</div>
<div id="issue-116">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pretransforms.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pretransforms.py</a><br>
<b>Line number: </b>85<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
81
82 return info[self.model.bottleneck.tokens_id]
83
84 def decode_tokens(self, tokens, **kwargs):
85 assert self.model.is_discrete, &quot;Cannot decode tokens with a continuous model&quot;
86
87 return self.model.decode_tokens(tokens, **kwargs)
88
</pre>
</div>
</div>
</div>
<div id="issue-117">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pretransforms.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pretransforms.py</a><br>
<b>Line number: </b>225<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
221 self.model.to(torch.float16).eval().requires_grad_(False)
222
223 def encode(self, x):
224
225 assert False, &quot;Audiocraft compression models do not support continuous encoding&quot;
226
227 # latents = self.model.encoder(x)
228
</pre>
</div>
</div>
</div>
<div id="issue-118">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pretransforms.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/pretransforms.py</a><br>
<b>Line number: </b>242<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
238 # return output
239
240 def decode(self, z):
241
242 assert False, &quot;Audiocraft compression models do not support continuous decoding&quot;
243
244 # if self.scale != 1.0:
245 # z = z * self.scale
</pre>
</div>
</div>
</div>
<div id="issue-119">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py</a><br>
<b>Line number: </b>53<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
49 self.emb = nn.Embedding(max_seq_len, dim)
50
51 def forward(self, x, pos = None, seq_start_pos = None):
52 seq_len, device = x.shape[1], x.device
53 assert seq_len &lt;= self.max_seq_len, f&#x27;you are passing in a sequence length of {seq_len} but your absolute positional embedding has a max sequence length of {self.max_seq_len}&#x27;
54
55 if pos is None:
56 pos = torch.arange(seq_len, device = device)
</pre>
</div>
</div>
</div>
<div id="issue-120">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py</a><br>
<b>Line number: </b>68<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
64
65 class ScaledSinusoidalEmbedding(nn.Module):
66 def __init__(self, dim, theta = 10000):
67 super().__init__()
68 assert (dim % 2) == 0, &#x27;dimension must be divisible by 2&#x27;
69 self.scale = nn.Parameter(torch.ones(1) * dim ** -0.5)
70
71 half_dim = dim // 2
</pre>
</div>
</div>
</div>
<div id="issue-121">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py</a><br>
<b>Line number: </b>108<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
104
105 inv_freq = 1. / (base ** (torch.arange(0, dim, 2).float() / dim))
106 self.register_buffer(&#x27;inv_freq&#x27;, inv_freq)
107
108 assert interpolation_factor &gt;= 1.
109 self.interpolation_factor = interpolation_factor
110
111 if not use_xpos:
</pre>
</div>
</div>
</div>
<div id="issue-122">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py</a><br>
<b>Line number: </b>350<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
346 if q_len == 1 and causal:
347 causal = False
348
349 if mask is not None:
350 assert mask.ndim == 4
351 mask = mask.expand(batch, heads, q_len, k_len)
352
353 # handle kv cache - this should be bypassable in updated flash attention 2
</pre>
</div>
</div>
</div>
<div id="issue-123">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py</a><br>
<b>Line number: </b>482<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
478 out = natten.functional.natten1dav(attn, v, kernel_size = self.natten_kernel_size, dilation=1).to(dtype_in)
479
480 # Prioritize Flash Attention 2
481 elif self.use_fa_flash:
482 assert final_attn_mask is None, &#x27;masking not yet supported for Flash Attention 2&#x27;
483 # Flash Attention 2 requires FP16 inputs
484 fa_dtype_in = q.dtype
485 q, k, v = map(lambda t: rearrange(t, &#x27;b h n d -&gt; b n h d&#x27;).to(torch.float16), (q, k, v))
</pre>
</div>
</div>
</div>
<div id="issue-124">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/transformer.py</a><br>
<b>Line number: </b>772<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
768
769 if prepend_embeds is not None:
770 prepend_length, prepend_dim = prepend_embeds.shape[1:]
771
772 assert prepend_dim == x.shape[-1], &#x27;prepend dimension must match sequence dimension&#x27;
773
774 x = torch.cat((prepend_embeds, x), dim = -2)
775
</pre>
</div>
</div>
</div>
<div id="issue-125">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/wavelets.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/wavelets.py</a><br>
<b>Line number: </b>27<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
23 self.wavelet = wavelet
24 self.channels = channels
25 self.levels = levels
26 filt = get_filter_bank(wavelet)
27 assert filt.shape[-1] % 2 == 1
28 kernel = filt[:2, None]
29 kernel = torch.flip(kernel, dims=(-1,))
30 index_i = torch.repeat_interleave(torch.arange(2), channels)
</pre>
</div>
</div>
</div>
<div id="issue-126">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/wavelets.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/models/wavelets.py</a><br>
<b>Line number: </b>59<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
55 self.wavelet = wavelet
56 self.channels = channels
57 self.levels = levels
58 filt = get_filter_bank(wavelet)
59 assert filt.shape[-1] % 2 == 1
60 kernel = filt[2:, None]
61 index_i = torch.repeat_interleave(torch.arange(2), channels)
62 index_j = torch.tile(torch.arange(channels), (2,))
</pre>
</div>
</div>
</div>
<div id="issue-127">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py</a><br>
<b>Line number: </b>276<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
272 self.losses = MultiLoss(self.loss_modules)
273
274 self.log_loss_info = log_loss_info
275
276 assert lr is not None or optimizer_configs is not None, &quot;Must specify either lr or optimizer_configs in training config&quot;
277
278 if optimizer_configs is None:
279 optimizer_configs = {
</pre>
</div>
</div>
</div>
<div id="issue-128">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py</a><br>
<b>Line number: </b>332<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
328 with torch.cuda.amp.autocast():
329 conditioning = self.diffusion.conditioner(metadata, self.device)
330
331 # If mask_padding is on, randomly drop the padding masks to allow for learning silence padding
332 use_padding_mask = self.mask_padding and random.random() &gt; self.mask_padding_dropout
333
334 # Create batch tensor of attention masks from the &quot;mask&quot; field of the metadata array
335 if use_padding_mask:
</pre>
</div>
</div>
</div>
<div id="issue-129">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py</a><br>
<b>Line number: </b>622<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
618 self.losses = MultiLoss(self.loss_modules)
619
620 self.log_loss_info = log_loss_info
621
622 assert lr is not None or optimizer_configs is not None, &quot;Must specify either lr or optimizer_configs in training config&quot;
623
624 if optimizer_configs is None:
625 optimizer_configs = {
</pre>
</div>
</div>
</div>
<div id="issue-130">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py</a><br>
<b>Line number: </b>664<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
660 # Create a mask tensor for each batch element
661 masks = []
662
663 for i in range(b):
664 mask_type = random.randint(0, 2)
665
666 if mask_type == 0: # Random mask with multiple segments
667 num_segments = random.randint(1, self.max_mask_segments)
</pre>
</div>
</div>
</div>
<div id="issue-131">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py</a><br>
<b>Line number: </b>667<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
663 for i in range(b):
664 mask_type = random.randint(0, 2)
665
666 if mask_type == 0: # Random mask with multiple segments
667 num_segments = random.randint(1, self.max_mask_segments)
668 max_segment_length = max_mask_length // num_segments
669
670 segment_lengths = random.sample(range(1, max_segment_length + 1), num_segments)
</pre>
</div>
</div>
</div>
<div id="issue-132">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py</a><br>
<b>Line number: </b>674<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
670 segment_lengths = random.sample(range(1, max_segment_length + 1), num_segments)
671
672 mask = torch.ones((1, 1, sequence_length))
673 for length in segment_lengths:
674 mask_start = random.randint(0, sequence_length - length)
675 mask[:, :, mask_start:mask_start + length] = 0
676
677 elif mask_type == 1: # Full mask
</pre>
</div>
</div>
</div>
<div id="issue-133">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py</a><br>
<b>Line number: </b>682<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
678 mask = torch.zeros((1, 1, sequence_length))
679
680 elif mask_type == 2: # Causal mask
681 mask = torch.ones((1, 1, sequence_length))
682 mask_length = random.randint(1, max_mask_length)
683 mask[:, :, -mask_length:] = 0
684
685 mask = mask.to(sequence.device)
</pre>
</div>
</div>
</div>
<div id="issue-134">
<div class="issue-block issue-sev-low">
<b>blacklist: </b> Standard pseudo-random generators are not suitable for security/cryptographic purposes.<br>
<b>Test ID:</b> B311<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/330.html" target="_blank">CWE-330</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/diffusion.py</a><br>
<b>Line number: </b>1193<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b311-random</a><br>
<div class="code">
<pre>
1189 for j in js:
1190 if i == j or (i != j and sources_added &lt; num_sources):
1191 # Randomly offset the mixed element between 0 and the length of the source
1192 seq_len = reals.shape[2]
1193 offset = random.randint(0, seq_len-1)
1194 source[i, :, offset:] += reals[j, :, :-offset]
1195 if i == j:
1196 # If this is the real one, shift the reals as well to ensure alignment
</pre>
</div>
</div>
</div>
<div id="issue-135">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/factory.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/factory.py</a><br>
<b>Line number: </b>7<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
3 from ..models.factory import create_model_from_config
4
5 def create_training_wrapper_from_config(model_config, model):
6 model_type = model_config.get(&#x27;model_type&#x27;, None)
7 assert model_type is not None, &#x27;model_type must be specified in model config&#x27;
8
9 training_config = model_config.get(&#x27;training&#x27;, None)
10 assert training_config is not None, &#x27;training config must be specified in model config&#x27;
</pre>
</div>
</div>
</div>
<div id="issue-136">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/factory.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/factory.py</a><br>
<b>Line number: </b>10<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
6 model_type = model_config.get(&#x27;model_type&#x27;, None)
7 assert model_type is not None, &#x27;model_type must be specified in model config&#x27;
8
9 training_config = model_config.get(&#x27;training&#x27;, None)
10 assert training_config is not None, &#x27;training config must be specified in model config&#x27;
11
12 if model_type == &#x27;autoencoder&#x27;:
13 from .autoencoders import AutoencoderTrainingWrapper
</pre>
</div>
</div>
</div>
<div id="issue-137">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/factory.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/factory.py</a><br>
<b>Line number: </b>161<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
157 raise NotImplementedError(f&#x27;Unknown model type: {model_type}&#x27;)
158
159 def create_demo_callback_from_config(model_config, **kwargs):
160 model_type = model_config.get(&#x27;model_type&#x27;, None)
161 assert model_type is not None, &#x27;model_type must be specified in model config&#x27;
162
163 training_config = model_config.get(&#x27;training&#x27;, None)
164 assert training_config is not None, &#x27;training config must be specified in model config&#x27;
</pre>
</div>
</div>
</div>
<div id="issue-138">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/factory.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/factory.py</a><br>
<b>Line number: </b>164<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
160 model_type = model_config.get(&#x27;model_type&#x27;, None)
161 assert model_type is not None, &#x27;model_type must be specified in model config&#x27;
162
163 training_config = model_config.get(&#x27;training&#x27;, None)
164 assert training_config is not None, &#x27;training config must be specified in model config&#x27;
165
166 demo_config = training_config.get(&quot;demo&quot;, {})
167
</pre>
</div>
</div>
</div>
<div id="issue-139">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/lm.py</a><br>
<b>Line number: </b>40<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
36 self.model_ema = None
37 if use_ema:
38 self.model_ema = EMA(self.model, ema_model=ema_copy, beta=0.99, update_every=10)
39
40 assert lr is not None or optimizer_configs is not None, &quot;Must specify either lr or optimizer_configs in training config&quot;
41
42 if optimizer_configs is None:
43 optimizer_configs = {
</pre>
</div>
</div>
</div>
<div id="issue-140">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/lm.py</a><br>
<b>Line number: </b>97<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
93 ce (torch.Tensor): Cross entropy averaged over the codebooks
94 ce_per_codebook (list of torch.Tensor): Cross entropy per codebook (detached).
95 &quot;&quot;&quot;
96 B, K, T = targets.shape
97 assert logits.shape[:-1] == targets.shape
98 assert mask.shape == targets.shape
99 ce = torch.zeros([], device=targets.device)
100 ce_per_codebook: tp.List[torch.Tensor] = []
</pre>
</div>
</div>
</div>
<div id="issue-141">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/lm.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/lm.py</a><br>
<b>Line number: </b>98<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
94 ce_per_codebook (list of torch.Tensor): Cross entropy per codebook (detached).
95 &quot;&quot;&quot;
96 B, K, T = targets.shape
97 assert logits.shape[:-1] == targets.shape
98 assert mask.shape == targets.shape
99 ce = torch.zeros([], device=targets.device)
100 ce_per_codebook: tp.List[torch.Tensor] = []
101 for k in range(K):
</pre>
</div>
</div>
</div>
<div id="issue-142">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py</a><br>
<b>Line number: </b>331<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
327 print(e)
328 print(&quot;Try `pip install auraloss[all]`.&quot;)
329
330 if self.scale == &quot;mel&quot;:
331 assert sample_rate != None # Must set sample rate to use mel scale
332 assert n_bins &lt;= fft_size # Must be more FFT bins than Mel bins
333 fb = librosa.filters.mel(sr=sample_rate, n_fft=fft_size, n_mels=n_bins)
334 fb = torch.tensor(fb).unsqueeze(0)
</pre>
</div>
</div>
</div>
<div id="issue-143">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py</a><br>
<b>Line number: </b>332<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
328 print(&quot;Try `pip install auraloss[all]`.&quot;)
329
330 if self.scale == &quot;mel&quot;:
331 assert sample_rate != None # Must set sample rate to use mel scale
332 assert n_bins &lt;= fft_size # Must be more FFT bins than Mel bins
333 fb = librosa.filters.mel(sr=sample_rate, n_fft=fft_size, n_mels=n_bins)
334 fb = torch.tensor(fb).unsqueeze(0)
335
</pre>
</div>
</div>
</div>
<div id="issue-144">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py</a><br>
<b>Line number: </b>337<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
333 fb = librosa.filters.mel(sr=sample_rate, n_fft=fft_size, n_mels=n_bins)
334 fb = torch.tensor(fb).unsqueeze(0)
335
336 elif self.scale == &quot;chroma&quot;:
337 assert sample_rate != None # Must set sample rate to use chroma scale
338 assert n_bins &lt;= fft_size # Must be more FFT bins than chroma bins
339 fb = librosa.filters.chroma(
340 sr=sample_rate, n_fft=fft_size, n_chroma=n_bins
</pre>
</div>
</div>
</div>
<div id="issue-145">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py</a><br>
<b>Line number: </b>338<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
334 fb = torch.tensor(fb).unsqueeze(0)
335
336 elif self.scale == &quot;chroma&quot;:
337 assert sample_rate != None # Must set sample rate to use chroma scale
338 assert n_bins &lt;= fft_size # Must be more FFT bins than chroma bins
339 fb = librosa.filters.chroma(
340 sr=sample_rate, n_fft=fft_size, n_chroma=n_bins
341 )
</pre>
</div>
</div>
</div>
<div id="issue-146">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py</a><br>
<b>Line number: </b>485<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
481 scale_invariance: bool = False,
482 **kwargs,
483 ):
484 super().__init__()
485 assert len(fft_sizes) == len(hop_sizes) == len(win_lengths) # must define all
486 self.fft_sizes = fft_sizes
487 self.hop_sizes = hop_sizes
488 self.win_lengths = win_lengths
</pre>
</div>
</div>
</div>
<div id="issue-147">
<div class="issue-block issue-sev-low">
<b>assert_used: </b> Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.<br>
<b>Test ID:</b> B101<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/703.html" target="_blank">CWE-703</a><br>
<b>File: </b><a href="/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py" target="_blank">/custom_nodes/hanzo-studio-sound-lab/nodes/stable_audio_tools/training/losses/auraloss.py</a><br>
<b>Line number: </b>592<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b101_assert_used.html</a><br>
<div class="code">
<pre>
588 loss (torch.Tensor): Aggreate loss term. Only returned if output=&#x27;loss&#x27;.
589 loss (torch.Tensor), sum_loss (torch.Tensor), diff_loss (torch.Tensor):
590 Aggregate and intermediate loss terms. Only returned if output=&#x27;full&#x27;.
591 &quot;&quot;&quot;
592 assert input.shape == target.shape # must have same shape
593 bs, chs, seq_len = input.size()
594
595 # compute sum and difference signals for both
</pre>
</div>
</div>
</div>
</div>
</body>
</html>