Files
custom-nodes-security-scan/docs/ComfyUI-Easy-Use-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

4358 lines
190 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">32960</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>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-Easy-Use/py/api.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/api.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 @PromptServer.instance.routes.get(&quot;/easyuse/reboot&quot;)
44 def reboot(request):
45 try:
46 sys.stdout.close_log()
47 except Exception as e:
48 pass
49
50 return os.execv(sys.executable, [sys.executable] + sys.argv)
51
</pre>
</div>
</div>
</div>
<div id="issue-1">
<div class="issue-block issue-sev-low">
<b>start_process_with_no_shell: </b> Starting a process without a shell.<br>
<b>Test ID:</b> B606<br>
<b>Severity: </b>LOW<br>
<b>Confidence: </b>MEDIUM<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-Easy-Use/py/api.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/api.py</a><br>
<b>Line number: </b>50<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b606_start_process_with_no_shell.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b606_start_process_with_no_shell.html</a><br>
<div class="code">
<pre>
46 sys.stdout.close_log()
47 except Exception as e:
48 pass
49
50 return os.execv(sys.executable, [sys.executable] + sys.argv)
51
52 # parse csv
53 @PromptServer.instance.routes.post(&quot;/easyuse/upload/csv&quot;)
</pre>
</div>
</div>
</div>
<div id="issue-2">
<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-Easy-Use/py/brushnet/powerpaint_utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/brushnet/powerpaint_utils.py</a><br>
<b>Line number: </b>56<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>
52 Args:
53 tokens (Union[str, List[str]]): The tokens to be added.
54 &quot;&quot;&quot;
55 num_added_tokens = self.wrapped.add_tokens(tokens, *args, **kwargs)
56 assert num_added_tokens != 0, (
57 f&quot;The tokenizer already contains the token {tokens}. Please pass &quot;
58 &quot;a different `placeholder_token` that is not already in the &quot;
59 &quot;tokenizer.&quot;
60 )
61
62 def get_token_info(self, token: str) -&gt; dict:
63 &quot;&quot;&quot;Get the information of a token, including its start and end index in
</pre>
</div>
</div>
</div>
<div id="issue-3">
<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-Easy-Use/py/brushnet/powerpaint_utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/brushnet/powerpaint_utils.py</a><br>
<b>Line number: </b>259<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>
255 Args:
256 embeddings (List[dict]): A list of embedding to be check.
257 &quot;&quot;&quot;
258 names = [emb[&quot;name&quot;] for emb in embeddings]
259 assert len(names) == len(set(names)), (
260 &quot;Found duplicated names in &#x27;external_embeddings&#x27;. Name list: &quot; f&quot;&#x27;{names}&#x27;&quot;
261 )
262
263 def check_ids_overlap(self, embeddings):
264 &quot;&quot;&quot;Check whether overlap exist in token ids of &#x27;external_embeddings&#x27;.
</pre>
</div>
</div>
</div>
<div id="issue-4">
<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-Easy-Use/py/brushnet/powerpaint_utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/brushnet/powerpaint_utils.py</a><br>
<b>Line number: </b>274<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>
270 ids_range.sort() # sort by &#x27;start&#x27;
271 # check if &#x27;end&#x27; has overlapping
272 for idx in range(len(ids_range) - 1):
273 name1, name2 = ids_range[idx][-1], ids_range[idx + 1][-1]
274 assert ids_range[idx][1] &lt;= ids_range[idx + 1][0], (
275 f&quot;Found ids overlapping between embeddings &#x27;{name1}&#x27; &quot; f&quot;and &#x27;{name2}&#x27;.&quot;
276 )
277
278 def add_embeddings(self, embeddings: Optional[Union[dict, List[dict]]]):
279 &quot;&quot;&quot;Add external embeddings to this layer.
</pre>
</div>
</div>
</div>
<div id="issue-5">
<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-Easy-Use/py/brushnet/powerpaint_utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/brushnet/powerpaint_utils.py</a><br>
<b>Line number: </b>365<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>
361 new_embedding.append(embedding[s_idx:e_idx])
362
363 # check if the next embedding need to replace is valid
364 actually_ids_to_replace = [int(i) for i in input_ids[e_idx: e_idx + end - start]]
365 assert actually_ids_to_replace == target_ids_to_replace, (
366 f&quot;Invalid &#x27;input_ids&#x27; in position: {s_idx} to {e_idx}. &quot;
367 f&quot;Expect &#x27;{target_ids_to_replace}&#x27; for embedding &quot;
368 f&quot;&#x27;{name}&#x27; but found &#x27;{actually_ids_to_replace}&#x27;.&quot;
369 )
370
371 new_embedding.append(ext_emb)
372
</pre>
</div>
</div>
</div>
<div id="issue-6">
<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-Easy-Use/py/brushnet/powerpaint_utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/brushnet/powerpaint_utils.py</a><br>
<b>Line number: </b>395<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>
391 will be used. Defaults to None.
392
393 input_ids: shape like [bz, LENGTH] or [LENGTH].
394 &quot;&quot;&quot;
395 assert input_ids.ndim in [1, 2]
396 if input_ids.ndim == 1:
397 input_ids = input_ids.unsqueeze(0)
398
</pre>
</div>
</div>
</div>
<div id="issue-7">
<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-Easy-Use/py/brushnet/powerpaint_utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/brushnet/powerpaint_utils.py</a><br>
<b>Line number: </b>431<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>
427
428 # TODO: support add tokens as dict, then we can load pretrained tokens.
429 &quot;&quot;&quot;
430 if initialize_tokens is not None:
431 assert len(initialize_tokens) == len(
432 placeholder_tokens
433 ), &quot;placeholder_token should be the same length as initialize_token&quot;
434 for ii in range(len(placeholder_tokens)):
435 tokenizer.add_placeholder_token(placeholder_tokens[ii], num_vec_per_token=num_vectors_per_token)
436
</pre>
</div>
</div>
</div>
<div id="issue-8">
<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-Easy-Use/py/brushnet/powerpaint_utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/brushnet/powerpaint_utils.py</a><br>
<b>Line number: </b>442<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>
438 embedding_layer = text_encoder.text_model.embeddings.token_embedding
439 text_encoder.text_model.embeddings.token_embedding = EmbeddingLayerWithFixes(embedding_layer)
440 embedding_layer = text_encoder.text_model.embeddings.token_embedding
441
442 assert embedding_layer is not None, (
443 &quot;Do not support get embedding layer for current text encoder. &quot; &quot;Please check your configuration.&quot;
444 )
445 initialize_embedding = []
446 if initialize_tokens is not None:
447 for ii in range(len(placeholder_tokens)):
</pre>
</div>
</div>
</div>
<div id="issue-9">
<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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.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 AssertionError: If the frequency tensor doesn&#x27;t match the expected shape.
34 AssertionError: If the target tensor &#x27;x&#x27; doesn&#x27;t have the expected number of dimensions.
35 &quot;&quot;&quot;
36 ndim = x.ndim
37 assert 0 &lt;= 1 &lt; ndim
38
39 if isinstance(freqs_cis, tuple):
40 # freqs_cis: (cos, sin) in real space
</pre>
</div>
</div>
</div>
<div id="issue-10">
<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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>42<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>
38
39 if isinstance(freqs_cis, tuple):
40 # freqs_cis: (cos, sin) in real space
41 if head_first:
42 assert freqs_cis[0].shape == (x.shape[-2], x.shape[-1]), f&#x27;freqs_cis shape {freqs_cis[0].shape} does not match x shape {x.shape}&#x27;
43 shape = [d if i == ndim - 2 or i == ndim - 1 else 1 for i, d in enumerate(x.shape)]
44 else:
45 assert freqs_cis[0].shape == (x.shape[1], x.shape[-1]), f&#x27;freqs_cis shape {freqs_cis[0].shape} does not match x shape {x.shape}&#x27;
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>45<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>
41 if head_first:
42 assert freqs_cis[0].shape == (x.shape[-2], x.shape[-1]), f&#x27;freqs_cis shape {freqs_cis[0].shape} does not match x shape {x.shape}&#x27;
43 shape = [d if i == ndim - 2 or i == ndim - 1 else 1 for i, d in enumerate(x.shape)]
44 else:
45 assert freqs_cis[0].shape == (x.shape[1], x.shape[-1]), f&#x27;freqs_cis shape {freqs_cis[0].shape} does not match x shape {x.shape}&#x27;
46 shape = [d if i == 1 or i == ndim - 1 else 1 for i, d in enumerate(x.shape)]
47 return freqs_cis[0].view(*shape), freqs_cis[1].view(*shape)
48 else:
</pre>
</div>
</div>
</div>
<div id="issue-12">
<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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>51<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>
47 return freqs_cis[0].view(*shape), freqs_cis[1].view(*shape)
48 else:
49 # freqs_cis: values in complex space
50 if head_first:
51 assert freqs_cis.shape == (x.shape[-2], x.shape[-1]), f&#x27;freqs_cis shape {freqs_cis.shape} does not match x shape {x.shape}&#x27;
52 shape = [d if i == ndim - 2 or i == ndim - 1 else 1 for i, d in enumerate(x.shape)]
53 else:
54 assert freqs_cis.shape == (x.shape[1], x.shape[-1]), f&#x27;freqs_cis shape {freqs_cis.shape} does not match x shape {x.shape}&#x27;
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>54<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>
50 if head_first:
51 assert freqs_cis.shape == (x.shape[-2], x.shape[-1]), f&#x27;freqs_cis shape {freqs_cis.shape} does not match x shape {x.shape}&#x27;
52 shape = [d if i == ndim - 2 or i == ndim - 1 else 1 for i, d in enumerate(x.shape)]
53 else:
54 assert freqs_cis.shape == (x.shape[1], x.shape[-1]), f&#x27;freqs_cis shape {freqs_cis.shape} does not match x shape {x.shape}&#x27;
55 shape = [d if i == 1 or i == ndim - 1 else 1 for i, d in enumerate(x.shape)]
56 return freqs_cis.view(*shape)
57
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>125<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>
121 factory_kwargs = {&#x27;device&#x27;: device, &#x27;dtype&#x27;: dtype}
122 super().__init__()
123 self.dim = dim
124 self.num_heads = num_heads
125 assert self.dim % num_heads == 0, &quot;self.kdim must be divisible by num_heads&quot;
126 self.head_dim = self.dim // num_heads
127 assert self.head_dim % 8 == 0 and self.head_dim &lt;= 128, &quot;Only support head_dim &lt;= 128 and divisible by 8&quot;
128
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>127<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>
123 self.dim = dim
124 self.num_heads = num_heads
125 assert self.dim % num_heads == 0, &quot;self.kdim must be divisible by num_heads&quot;
126 self.head_dim = self.dim // num_heads
127 assert self.head_dim % 8 == 0 and self.head_dim &lt;= 128, &quot;Only support head_dim &lt;= 128 and divisible by 8&quot;
128
129 self.Wqkv = nn.Linear(dim, 3 * dim, bias=qkv_bias, **factory_kwargs)
130 # TODO: eps should be 1 / 65530 if using fp16
</pre>
</div>
</div>
</div>
<div id="issue-16">
<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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.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 # Apply RoPE if needed
155 if freqs_cis_img is not None:
156 qq, kk = apply_rotary_emb(q, k, freqs_cis_img)
157 assert qq.shape == q.shape and kk.shape == k.shape, f&#x27;qq: {qq.shape}, q: {q.shape}, kk: {kk.shape}, k: {k.shape}&#x27;
158 q, k = qq, kk
159
160 qkv = torch.stack([q, k, v], dim=2) # [b, s, 3, h, d]
</pre>
</div>
</div>
</div>
<div id="issue-17">
<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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>191<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>
187 super().__init__()
188 self.qdim = qdim
189 self.kdim = kdim
190 self.num_heads = num_heads
191 assert self.qdim % num_heads == 0, &quot;self.qdim must be divisible by num_heads&quot;
192 self.head_dim = self.qdim // num_heads
193 assert self.head_dim % 8 == 0 and self.head_dim &lt;= 128, &quot;Only support head_dim &lt;= 128 and divisible by 8&quot;
194
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>193<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>
189 self.kdim = kdim
190 self.num_heads = num_heads
191 assert self.qdim % num_heads == 0, &quot;self.qdim must be divisible by num_heads&quot;
192 self.head_dim = self.qdim // num_heads
193 assert self.head_dim % 8 == 0 and self.head_dim &lt;= 128, &quot;Only support head_dim &lt;= 128 and divisible by 8&quot;
194
195 self.scale = self.head_dim ** -0.5
196
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>231<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>
227
228 # Apply RoPE if needed
229 if freqs_cis_img is not None:
230 qq, _ = apply_rotary_emb(q, None, freqs_cis_img)
231 assert qq.shape == q.shape, f&#x27;qq: {qq.shape}, q: {q.shape}&#x27;
232 q = qq # [b, s1, h, d]
233 kv = torch.stack([k, v], dim=2) # [b, s1, 2, h, d]
234 context = self.inner_attn(q, kv) # [b, s1, h, d]
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>266<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>
262 super().__init__()
263 self.qdim = qdim
264 self.kdim = kdim
265 self.num_heads = num_heads
266 assert self.qdim % num_heads == 0, &quot;self.qdim must be divisible by num_heads&quot;
267 self.head_dim = self.qdim // num_heads
268 assert self.head_dim % 8 == 0 and self.head_dim &lt;= 128, &quot;Only support head_dim &lt;= 128 and divisible by 8&quot;
269 self.scale = self.head_dim ** -0.5
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>268<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>
264 self.kdim = kdim
265 self.num_heads = num_heads
266 assert self.qdim % num_heads == 0, &quot;self.qdim must be divisible by num_heads&quot;
267 self.head_dim = self.qdim // num_heads
268 assert self.head_dim % 8 == 0 and self.head_dim &lt;= 128, &quot;Only support head_dim &lt;= 128 and divisible by 8&quot;
269 self.scale = self.head_dim ** -0.5
270
271 self.q_proj = nn.Linear(qdim, qdim, bias=qkv_bias, **factory_kwargs)
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>304<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>
300
301 # Apply RoPE if needed
302 if freqs_cis_img is not None:
303 qq, _ = apply_rotary_emb(q, None, freqs_cis_img)
304 assert qq.shape == q.shape, f&#x27;qq: {qq.shape}, q: {q.shape}&#x27;
305 q = qq
306
307 q = q * self.scale
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>336<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>
332 ):
333 super().__init__()
334 self.dim = dim
335 self.num_heads = num_heads
336 assert self.dim % num_heads == 0, &#x27;dim should be divisible by num_heads&#x27;
337 self.head_dim = self.dim // num_heads
338 # This assertion is aligned with flash attention
339 assert self.head_dim % 8 == 0 and self.head_dim &lt;= 128, &quot;Only support head_dim &lt;= 128 and divisible by 8&quot;
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>339<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>
335 self.num_heads = num_heads
336 assert self.dim % num_heads == 0, &#x27;dim should be divisible by num_heads&#x27;
337 self.head_dim = self.dim // num_heads
338 # This assertion is aligned with flash attention
339 assert self.head_dim % 8 == 0 and self.head_dim &lt;= 128, &quot;Only support head_dim &lt;= 128 and divisible by 8&quot;
340 self.scale = self.head_dim ** -0.5
341
342 # qkv --&gt; Wqkv
</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-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/attn_layers.py</a><br>
<b>Line number: </b>361<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>
357
358 # Apply RoPE if needed
359 if freqs_cis_img is not None:
360 qq, kk = apply_rotary_emb(q, k, freqs_cis_img, head_first=True)
361 assert qq.shape == q.shape and kk.shape == k.shape, \
362 f&#x27;qq: {qq.shape}, q: {q.shape}, kk: {kk.shape}, k: {k.shape}&#x27;
363 q, k = qq, kk
364
365 q = q * self.scale
</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-Easy-Use/py/dit/hunyuanDiT/models/models.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/models.py</a><br>
<b>Line number: </b>423<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>
419 &quot;&quot;&quot;
420 c = self.unpatchify_channels
421 p = self.x_embedder.patch_size[0]
422 # h = w = int(x.shape[1] ** 0.5)
423 assert h * w == x.shape[1]
424
425 x = x.reshape(shape=(x.shape[0], h, w, p, p, c))
426 x = torch.einsum(&#x27;nhwpqc-&gt;nchpwq&#x27;, x)
</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-Easy-Use/py/dit/hunyuanDiT/models/posemb_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/posemb_layers.py</a><br>
<b>Line number: </b>84<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>
80 return pos_embed
81
82
83 def get_2d_sincos_pos_embed_from_grid(embed_dim, grid):
84 assert embed_dim % 2 == 0
85
86 # use half of dimensions to encode grid_h
87 emb_h = get_1d_sincos_pos_embed_from_grid(embed_dim // 2, grid[0]) # (H*W, D/2)
</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-Easy-Use/py/dit/hunyuanDiT/models/posemb_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/posemb_layers.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 embed_dim: output dimension for each position
97 pos: a list of positions to be encoded: size (W,H)
98 out: (M, D)
99 &quot;&quot;&quot;
100 assert embed_dim % 2 == 0
101 omega = np.arange(embed_dim // 2, dtype=np.float64)
102 omega /= embed_dim / 2.
103 omega = 1. / 10000**omega # (D/2,)
</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-Easy-Use/py/dit/hunyuanDiT/models/posemb_layers.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/hunyuanDiT/models/posemb_layers.py</a><br>
<b>Line number: </b>146<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>
142 return pos_embed
143
144
145 def get_2d_rotary_pos_embed_from_grid(embed_dim, grid, use_real=False):
146 assert embed_dim % 4 == 0
147
148 # use half of dimensions to encode grid_h
149 emb_h = get_1d_rotary_pos_embed(embed_dim // 2, grid[0].reshape(-1), use_real=use_real) # (H*W, D/4)
</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-Easy-Use/py/dit/pixArt/models/PixArt.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArt.py</a><br>
<b>Line number: </b>193<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>
189 &quot;&quot;&quot;
190 c = self.out_channels
191 p = self.x_embedder.patch_size[0]
192 h = w = int(x.shape[1] ** 0.5)
193 assert h * w == x.shape[1]
194
195 x = x.reshape(shape=(x.shape[0], h, w, p, p, c))
196 x = torch.einsum(&#x27;nhwpqc-&gt;nchpwq&#x27;, x)
</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-Easy-Use/py/dit/pixArt/models/PixArt.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArt.py</a><br>
<b>Line number: </b>222<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>
218 return pos_embed.astype(np.float32)
219
220
221 def get_2d_sincos_pos_embed_from_grid(embed_dim, grid):
222 assert embed_dim % 2 == 0
223
224 # use half of dimensions to encode grid_h
225 emb_h = get_1d_sincos_pos_embed_from_grid(embed_dim // 2, grid[0]) # (H*W, D/2)
</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-Easy-Use/py/dit/pixArt/models/PixArt.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArt.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 embed_dim: output dimension for each position
235 pos: a list of positions to be encoded: size (M,)
236 out: (M, D)
237 &quot;&quot;&quot;
238 assert embed_dim % 2 == 0
239 omega = np.arange(embed_dim // 2, dtype=np.float64)
240 omega /= embed_dim / 2.
241 omega = 1. / 10000 ** omega # (D/2,)
</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-Easy-Use/py/dit/pixArt/models/PixArtMS.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArtMS.py</a><br>
<b>Line number: </b>268<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>
264 imgs: (N, H, W, C)
265 &quot;&quot;&quot;
266 c = self.out_channels
267 p = self.x_embedder.patch_size[0]
268 assert self.h * self.w == x.shape[1]
269
270 x = x.reshape(shape=(x.shape[0], self.h, self.w, p, p, c))
271 x = torch.einsum(&#x27;nhwpqc-&gt;nchpwq&#x27;, x)
</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-Easy-Use/py/dit/pixArt/models/PixArt_blocks.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArt_blocks.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
37 class MultiHeadCrossAttention(nn.Module):
38 def __init__(self, d_model, num_heads, attn_drop=0., proj_drop=0., **block_kwargs):
39 super(MultiHeadCrossAttention, self).__init__()
40 assert d_model % num_heads == 0, &quot;d_model must be divisible by num_heads&quot;
41
42 self.d_model = d_model
43 self.num_heads = num_heads
</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-Easy-Use/py/dit/pixArt/models/PixArt_blocks.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArt_blocks.py</a><br>
<b>Line number: </b>371<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>
367
368 def forward(self, s, bs):
369 if s.ndim == 1:
370 s = s[:, None]
371 assert s.ndim == 2
372 if s.shape[0] != bs:
373 s = s.repeat(bs//s.shape[0], 1)
374 assert s.shape[0] == bs
</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-Easy-Use/py/dit/pixArt/models/PixArt_blocks.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArt_blocks.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 s = s[:, None]
371 assert s.ndim == 2
372 if s.shape[0] != bs:
373 s = s.repeat(bs//s.shape[0], 1)
374 assert s.shape[0] == bs
375 b, dims = s.shape[0], s.shape[1]
376 s = rearrange(s, &quot;b d -&gt; (b d)&quot;)
377 s_freq = self.timestep_embedding(s, self.frequency_embedding_size)
</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-Easy-Use/py/dit/pixArt/models/PixArt_blocks.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArt_blocks.py</a><br>
<b>Line number: </b>438<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>
434 return caption
435
436 def forward(self, caption, train, force_drop_ids=None):
437 if train:
438 assert caption.shape[2:] == self.y_embedding.shape
439 use_dropout = self.uncond_prob &gt; 0
440 if (train and use_dropout) or (force_drop_ids is not None):
441 caption = self.token_drop(caption, force_drop_ids)
</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-Easy-Use/py/dit/pixArt/models/PixArt_blocks.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/PixArt_blocks.py</a><br>
<b>Line number: </b>471<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>
467 caption = torch.where(drop_ids[:, None, None, None], self.y_embedding, caption)
468 return global_caption, caption
469
470 def forward(self, caption, train, force_drop_ids=None):
471 assert caption.shape[2: ] == self.y_embedding.shape
472 global_caption = caption.mean(dim=2).squeeze()
473 use_dropout = self.uncond_prob &gt; 0
474 if (train and use_dropout) or (force_drop_ids is not None):
</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-Easy-Use/py/dit/pixArt/models/pixart_controlnet.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/pixart_controlnet.py</a><br>
<b>Line number: </b>191<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>
187 imgs: (N, H, W, C)
188 &quot;&quot;&quot;
189 c = self.out_channels
190 p = self.x_embedder.patch_size[0]
191 assert self.h * self.w == x.shape[1]
192
193 x = x.reshape(shape=(x.shape[0], self.h, self.w, p, p, c))
194 x = torch.einsum(&#x27;nhwpqc-&gt;nchpwq&#x27;, x)
</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-Easy-Use/py/dit/pixArt/models/utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/pixArt/models/utils.py</a><br>
<b>Line number: </b>19<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>
15 to_1tuple = _ntuple(1)
16 to_2tuple = _ntuple(2)
17
18 def set_grad_checkpoint(model, use_fp32_attention=False, gc_step=1):
19 assert isinstance(model, nn.Module)
20
21 def set_attr(module):
22 module.grad_checkpointing = True
</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-Easy-Use/py/dit/utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dit/utils.py</a><br>
<b>Line number: </b>33<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>
29 return torch.float8_e4m3fn
30 else:
31 raise NotImplementedError(f&quot;Unknown 8bit dtype &#x27;{s}&#x27;&quot;)
32 elif &quot;bnb&quot; in s:
33 assert s in [&quot;bnb8bit&quot;, &quot;bnb4bit&quot;], f&quot;Unknown bnb mode &#x27;{s}&#x27;&quot;
34 return s
35 elif s is None:
36 return None
</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-Easy-Use/py/dynamiCrafter/lvdm/distributions.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/distributions.py</a><br>
<b>Line number: </b>80<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>
76 for obj in (mean1, logvar1, mean2, logvar2):
77 if isinstance(obj, torch.Tensor):
78 tensor = obj
79 break
80 assert tensor is not None, &quot;at least one argument must be a Tensor&quot;
81
82 # Force variances to be Tensors. Broadcasting helps convert scalars to
83 # Tensors, but it does not work for torch.exp().
</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-Easy-Use/py/dynamiCrafter/lvdm/ema.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/ema.py</a><br>
<b>Line number: </b>44<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>
40 sname = self.m_name2s_name[key]
41 shadow_params[sname] = shadow_params[sname].type_as(m_param[key])
42 shadow_params[sname].sub_(one_minus_decay * (shadow_params[sname] - m_param[key]))
43 else:
44 assert not key in self.m_name2s_name
45
46 def copy_to(self, model):
47 m_param = dict(model.named_parameters())
</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-Easy-Use/py/dynamiCrafter/lvdm/ema.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/ema.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 for key in m_param:
50 if m_param[key].requires_grad:
51 m_param[key].data.copy_(shadow_params[self.m_name2s_name[key]].data)
52 else:
53 assert not key in self.m_name2s_name
54
55 def store(self, parameters):
56 &quot;&quot;&quot;
</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-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py</a><br>
<b>Line number: </b>33<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>
29 self.image_key = image_key
30 self.encoder = Encoder(**ddconfig)
31 self.decoder = Decoder(**ddconfig)
32 self.loss = instantiate_from_config(lossconfig)
33 assert ddconfig[&quot;double_z&quot;]
34 self.quant_conv = torch.nn.Conv2d(2*ddconfig[&quot;z_channels&quot;], 2*embed_dim, 1)
35 self.post_quant_conv = torch.nn.Conv2d(embed_dim, ddconfig[&quot;z_channels&quot;], 1)
36 self.embed_dim = embed_dim
</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-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py</a><br>
<b>Line number: </b>42<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>
38 self.test = test
39 self.test_args = test_args
40 self.logdir = logdir
41 if colorize_nlabels is not None:
42 assert type(colorize_nlabels)==int
43 self.register_buffer(&quot;colorize&quot;, torch.randn(3, colorize_nlabels, 1, 1))
44 if monitor is not None:
45 self.monitor = monitor
</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-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py</a><br>
<b>Line number: </b>73<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>
69 if self.test_args.save_reconstruction:
70 os.makedirs(self.root_dec, exist_ok=True)
71 if self.test_args.save_input:
72 os.makedirs(self.root_inputs, exist_ok=True)
73 assert(self.test_args is not None)
74 self.test_maximum = getattr(self.test_args, &#x27;test_maximum&#x27;, None)
75 self.count = 0
76 self.eval_metrics = {}
</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-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py</a><br>
<b>Line number: </b>186<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>
182 if not only_inputs:
183 xrec, posterior = self(x)
184 if x.shape[1] &gt; 3:
185 # colorize with random projection
186 assert xrec.shape[1] &gt; 3
187 x = self.to_rgb(x)
188 xrec = self.to_rgb(xrec)
189 log[&quot;samples&quot;] = self.decode(torch.randn_like(posterior.sample()))
</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-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/autoencoder.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 log[&quot;inputs&quot;] = x
192 return log
193
194 def to_rgb(self, x):
195 assert self.image_key == &quot;segmentation&quot;
196 if not hasattr(self, &quot;colorize&quot;):
197 self.register_buffer(&quot;colorize&quot;, torch.randn(3, x.shape[1], 1, 1).to(x))
198 x = F.conv2d(x, weight=self.colorize)
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>69<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>
65 logvar_init=0.,
66 rescale_betas_zero_snr=False,
67 ):
68 super().__init__()
69 assert parameterization in [&quot;eps&quot;, &quot;x0&quot;, &quot;v&quot;], &#x27;currently only supporting &quot;eps&quot; and &quot;x0&quot; and &quot;v&quot;&#x27;
70 self.parameterization = parameterization
71 mainlogger.info(f&quot;{self.__class__.__name__}: Running in {self.parameterization}-prediction mode&quot;)
72 self.cond_stage_model = None
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>131<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>
127 timesteps, = betas.shape
128 self.num_timesteps = int(timesteps)
129 self.linear_start = linear_start
130 self.linear_end = linear_end
131 assert alphas_cumprod.shape[0] == self.num_timesteps, &#x27;alphas have to be defined for each timestep&#x27;
132
133 to_torch = partial(torch.tensor, dtype=torch.float32)
134
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>176<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>
172 raise NotImplementedError(&quot;mu not supported&quot;)
173 # TODO how to choose this term
174 lvlb_weights[0] = lvlb_weights[1]
175 self.register_buffer(&#x27;lvlb_weights&#x27;, lvlb_weights, persistent=False)
176 assert not torch.isnan(self.lvlb_weights).all()
177
178 @contextmanager
179 def ema_scope(self, context=None):
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>383<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>
379 perframe_ae=False,
380 *args, **kwargs):
381 self.num_timesteps_cond = default(num_timesteps_cond, 1)
382 self.scale_by_std = scale_by_std
383 assert self.num_timesteps_cond &lt;= kwargs[&#x27;timesteps&#x27;]
384 # for backwards compatibility after implementation of DiffusionWrapper
385 ckpt_path = kwargs.pop(&quot;ckpt_path&quot;, None)
386 ignore_keys = kwargs.pop(&quot;ignore_keys&quot;, [])
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>421<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>
417 self.clip_denoised = False
418
419 self.cond_stage_forward = cond_stage_forward
420 self.encoder_type = encoder_type
421 assert(encoder_type in [&quot;2d&quot;, &quot;3d&quot;])
422 self.uncond_prob = uncond_prob
423 self.classifier_free_guidance = True if uncond_prob &gt; 0 else False
424 assert(uncond_type in [&quot;zero_embed&quot;, &quot;empty_seq&quot;])
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>424<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>
420 self.encoder_type = encoder_type
421 assert(encoder_type in [&quot;2d&quot;, &quot;3d&quot;])
422 self.uncond_prob = uncond_prob
423 self.classifier_free_guidance = True if uncond_prob &gt; 0 else False
424 assert(uncond_type in [&quot;zero_embed&quot;, &quot;empty_seq&quot;])
425 self.uncond_type = uncond_type
426
427 self.restarted_from_ckpt = False
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>465<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>
461 c = c.mode()
462 else:
463 c = self.cond_stage_model(c)
464 else:
465 assert hasattr(self.cond_stage_model, self.cond_stage_forward)
466 c = getattr(self.cond_stage_model, self.cond_stage_forward)(c)
467 return c
468
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>588<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>
584 t_in = t
585 model_out = self.apply_model(x, t_in, c, **kwargs)
586
587 if score_corrector is not None:
588 assert self.parameterization == &quot;eps&quot;
589 model_out = score_corrector.modify_score(self, model_out, x, t, c, **corrector_kwargs)
590
591 if self.parameterization == &quot;eps&quot;:
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.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
650 iterator = tqdm(reversed(range(0, timesteps)), desc=&#x27;Sampling t&#x27;, total=timesteps) if verbose else reversed(range(0, timesteps))
651
652 if mask is not None:
653 assert x0 is not None
654 assert x0.shape[2:3] == mask.shape[2:3] # spatial size has to match
655
656 for i in iterator:
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>654<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>
650 iterator = tqdm(reversed(range(0, timesteps)), desc=&#x27;Sampling t&#x27;, total=timesteps) if verbose else reversed(range(0, timesteps))
651
652 if mask is not None:
653 assert x0 is not None
654 assert x0.shape[2:3] == mask.shape[2:3] # spatial size has to match
655
656 for i in iterator:
657 ts = torch.full((b,), i, device=device, dtype=torch.long)
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.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
656 for i in iterator:
657 ts = torch.full((b,), i, device=device, dtype=torch.long)
658 if self.shorten_cond_schedule:
659 assert self.model.conditioning_key != &#x27;hybrid&#x27;
660 tc = self.cond_ids[ts].to(cond.device)
661 cond = self.q_sample(x_start=cond, t=tc, noise=torch.randn_like(cond))
662
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>722<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>
718 elif self.conditioning_key == &#x27;adm&#x27;:
719 cc = c_crossattn[0]
720 out = self.diffusion_model(x, t, y=cc)
721 elif self.conditioning_key == &#x27;hybrid-adm&#x27;:
722 assert c_adm is not None
723 xc = torch.cat([x] + c_concat, dim=1)
724 cc = torch.cat(c_crossattn, 1)
725 out = self.diffusion_model(xc, t, context=cc, y=c_adm, **kwargs)
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>727<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>
723 xc = torch.cat([x] + c_concat, dim=1)
724 cc = torch.cat(c_crossattn, 1)
725 out = self.diffusion_model(xc, t, context=cc, y=c_adm, **kwargs)
726 elif self.conditioning_key == &#x27;hybrid-time&#x27;:
727 assert s is not None
728 xc = torch.cat([x] + c_concat, dim=1)
729 cc = torch.cat(c_crossattn, 1)
730 out = self.diffusion_model(xc, t, context=cc, s=s)
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py</a><br>
<b>Line number: </b>751<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>
747 xc = x
748 out = self.diffusion_model(xc, t, context=cc, y=s, mask=mask)
749 elif self.conditioning_key == &#x27;hybrid-time-adm&#x27;: # adm means y, e.g., class index
750 # assert s is not None
751 assert c_adm is not None
752 xc = torch.cat([x] + c_concat, dim=1)
753 cc = torch.cat(c_crossattn, 1)
754 out = self.diffusion_model(xc, t, context=cc, s=s, y=c_adm)
</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-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/ddpm3d.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 xc = torch.cat([x] + c_concat, dim=1)
753 cc = torch.cat(c_crossattn, 1)
754 out = self.diffusion_model(xc, t, context=cc, s=s, y=c_adm)
755 elif self.conditioning_key == &#x27;crossattn-adm&#x27;:
756 assert c_adm is not None
757 cc = torch.cat(c_crossattn, 1)
758 out = self.diffusion_model(x, t, context=cc, y=c_adm)
759 else:
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim.py</a><br>
<b>Line number: </b>28<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>
24 def make_schedule(self, ddim_num_steps, ddim_discretize=&quot;uniform&quot;, ddim_eta=0., verbose=True):
25 self.ddim_timesteps = make_ddim_timesteps(ddim_discr_method=ddim_discretize, num_ddim_timesteps=ddim_num_steps,
26 num_ddpm_timesteps=self.ddpm_num_timesteps,verbose=verbose)
27 alphas_cumprod = self.model.alphas_cumprod
28 assert alphas_cumprod.shape[0] == self.ddpm_num_timesteps, &#x27;alphas have to be defined for each timestep&#x27;
29 to_torch = lambda x: x.clone().detach().to(torch.float32).to(self.model.device)
30
31 if self.model.use_dynamic_rescale:
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim.py</a><br>
<b>Line number: </b>175<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>
171 ts = torch.full((b,), step, device=device, dtype=torch.long)
172
173 ## use mask to blend noised original latent (img_orig) &amp; new sampled latent (img)
174 if mask is not None:
175 assert x0 is not None
176 if clean_cond:
177 img_orig = x0
178 else:
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim.py</a><br>
<b>Line number: </b>237<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>
233 else:
234 e_t = model_output
235
236 if score_corrector is not None:
237 assert self.model.parameterization == &quot;eps&quot;, &#x27;not implemented&#x27;
238 e_t = score_corrector.modify_score(self.model, e_t, x, t, c, **corrector_kwargs)
239
240 alphas = self.model.alphas_cumprod if use_original_steps else self.ddim_alphas
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim_multiplecond.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim_multiplecond.py</a><br>
<b>Line number: </b>28<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>
24 def make_schedule(self, ddim_num_steps, ddim_discretize=&quot;uniform&quot;, ddim_eta=0., verbose=True):
25 self.ddim_timesteps = make_ddim_timesteps(ddim_discr_method=ddim_discretize, num_ddim_timesteps=ddim_num_steps,
26 num_ddpm_timesteps=self.ddpm_num_timesteps,verbose=verbose)
27 alphas_cumprod = self.model.alphas_cumprod
28 assert alphas_cumprod.shape[0] == self.ddpm_num_timesteps, &#x27;alphas have to be defined for each timestep&#x27;
29 to_torch = lambda x: x.clone().detach().to(torch.float32).to(self.model.device)
30
31 if self.model.use_dynamic_rescale:
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim_multiplecond.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim_multiplecond.py</a><br>
<b>Line number: </b>179<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>
175 ts = torch.full((b,), step, device=device, dtype=torch.long)
176
177 ## use mask to blend noised original latent (img_orig) &amp; new sampled latent (img)
178 if mask is not None:
179 assert x0 is not None
180 if clean_cond:
181 img_orig = x0
182 else:
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim_multiplecond.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/ddim_multiplecond.py</a><br>
<b>Line number: </b>244<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>
240 else:
241 e_t = model_output
242
243 if score_corrector is not None:
244 assert self.model.parameterization == &quot;eps&quot;, &#x27;not implemented&#x27;
245 e_t = score_corrector.modify_score(self.model, e_t, x, t, c, **corrector_kwargs)
246
247 alphas = self.model.alphas_cumprod if use_original_steps else self.ddim_alphas
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.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 schedule == &#x27;discrete&#x27;:
100 if betas is not None:
101 log_alphas = 0.5 * torch.log(1 - betas).cumsum(dim=0)
102 else:
103 assert alphas_cumprod is not None
104 log_alphas = 0.5 * torch.log(alphas_cumprod)
105 self.total_N = len(log_alphas)
106 self.T = 1.
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.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 t_continuous = t_continuous.expand((x.shape[0]))
327 if guidance_type == &quot;uncond&quot;:
328 return noise_pred_fn(x, t_continuous)
329 elif guidance_type == &quot;classifier&quot;:
330 assert classifier_fn is not None
331 t_input = get_model_input_time(t_continuous)
332 cond_grad = cond_grad_fn(x, t_input)
333 sigma_t = noise_schedule.marginal_std(t_continuous)
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py</a><br>
<b>Line number: </b>346<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>
342 c_in = torch.cat([unconditional_condition, condition])
343 noise_uncond, noise = noise_pred_fn(x_in, t_in, cond=c_in).chunk(2)
344 return noise_uncond + guidance_scale * (noise - noise_uncond)
345
346 assert model_type in [&quot;noise&quot;, &quot;x_start&quot;, &quot;v&quot;]
347 assert guidance_type in [&quot;uncond&quot;, &quot;classifier&quot;, &quot;classifier-free&quot;]
348 return model_fn
349
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py</a><br>
<b>Line number: </b>347<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>
343 noise_uncond, noise = noise_pred_fn(x_in, t_in, cond=c_in).chunk(2)
344 return noise_uncond + guidance_scale * (noise - noise_uncond)
345
346 assert model_type in [&quot;noise&quot;, &quot;x_start&quot;, &quot;v&quot;]
347 assert guidance_type in [&quot;uncond&quot;, &quot;classifier&quot;, &quot;classifier-free&quot;]
348 return model_fn
349
350
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.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 t = t.view(-1)
470 if &#x27;bh&#x27; in self.variant:
471 return self.multistep_uni_pc_bh_update(x, model_prev_list, t_prev_list, t, order, **kwargs)
472 else:
473 assert self.variant == &#x27;vary_coeff&#x27;
474 return self.multistep_uni_pc_vary_update(x, model_prev_list, t_prev_list, t, order, **kwargs)
475
476 def multistep_uni_pc_vary_update(self, x, model_prev_list, t_prev_list, t, order, use_corrector=True):
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.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
476 def multistep_uni_pc_vary_update(self, x, model_prev_list, t_prev_list, t, order, use_corrector=True):
477 print(f&#x27;using unified predictor-corrector with order {order} (solver type: vary coeff)&#x27;)
478 ns = self.noise_schedule
479 assert order &lt;= len(model_prev_list)
480
481 # first compute rks
482 t_prev_0 = t_prev_list[-1]
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py</a><br>
<b>Line number: </b>582<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>
578
579 def multistep_uni_pc_bh_update(self, x, model_prev_list, t_prev_list, t, order, x_t=None, use_corrector=True):
580 print(f&#x27;using unified predictor-corrector with order {order} (solver type: B(h))&#x27;)
581 ns = self.noise_schedule
582 assert order &lt;= len(model_prev_list)
583 dims = x.dim()
584
585 # first compute rks
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py</a><br>
<b>Line number: </b>708<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>
704 t_0 = 1. / self.noise_schedule.total_N if t_end is None else t_end
705 t_T = self.noise_schedule.T if t_start is None else t_start
706 device = x.device
707 if method == &#x27;multistep&#x27;:
708 assert steps &gt;= order
709 timesteps = self.get_time_steps(skip_type=skip_type, t_T=t_T, t_0=t_0, N=steps, device=device)
710 assert timesteps.shape[0] - 1 == steps
711 with torch.no_grad():
</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-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/models/samplers/unipc/uni_pc.py</a><br>
<b>Line number: </b>710<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>
706 device = x.device
707 if method == &#x27;multistep&#x27;:
708 assert steps &gt;= order
709 timesteps = self.get_time_steps(skip_type=skip_type, t_T=t_T, t_0=t_0, N=steps, device=device)
710 assert timesteps.shape[0] - 1 == steps
711 with torch.no_grad():
712 vec_t = timesteps[0].expand((x.shape[0]))
713 model_prev_list = [self.model_fn(x, vec_t)]
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/attention.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/attention.py</a><br>
<b>Line number: </b>84<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>
80 )
81
82 self.relative_position = relative_position
83 if self.relative_position:
84 assert(temporal_length is not None)
85 self.relative_position_k = RelativePosition(num_units=dim_head, max_relative_position=temporal_length)
86 self.relative_position_v = RelativePosition(num_units=dim_head, max_relative_position=temporal_length)
87 else:
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/attention.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/attention.py</a><br>
<b>Line number: </b>608<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>
604 else:
605 self.proj_in = operations.Linear(in_channels, inner_dim, device=device, dtype=dtype)
606
607 if relative_position:
608 assert(temporal_length is not None)
609 attention_cls = partial(CrossAttention, relative_position=True, temporal_length=temporal_length, device=device, dtype=dtype)
610 else:
611 attention_cls = partial(CrossAttention, temporal_length=temporal_length, device=device, dtype=dtype)
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/attention.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/attention.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 attention_cls = partial(CrossAttention, relative_position=True, temporal_length=temporal_length, device=device, dtype=dtype)
610 else:
611 attention_cls = partial(CrossAttention, temporal_length=temporal_length, device=device, dtype=dtype)
612 if self.causal_attention:
613 assert(temporal_length is not None)
614 self.mask = torch.tril(torch.ones([1, temporal_length, temporal_length]))
615
616 if self.only_self_att:
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/encoders/condition.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/encoders/condition.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
97 def __init__(self, version=&quot;openai/clip-vit-large-patch14&quot;, device=&quot;cuda&quot;, max_length=77,
98 freeze=True, layer=&quot;last&quot;, layer_idx=None): # clip-vit-base-patch32
99 super().__init__()
100 assert layer in self.LAYERS
101 self.tokenizer = CLIPTokenizer.from_pretrained(version)
102 self.transformer = CLIPTextModel.from_pretrained(version)
103 self.device = device
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/encoders/condition.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/encoders/condition.py</a><br>
<b>Line number: </b>110<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>
106 self.freeze()
107 self.layer = layer
108 self.layer_idx = layer_idx
109 if layer == &quot;hidden&quot;:
110 assert layer_idx is not None
111 assert 0 &lt;= abs(layer_idx) &lt;= 12
112
113 def freeze(self):
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/encoders/condition.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/encoders/condition.py</a><br>
<b>Line number: </b>111<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>
107 self.layer = layer
108 self.layer_idx = layer_idx
109 if layer == &quot;hidden&quot;:
110 assert layer_idx is not None
111 assert 0 &lt;= abs(layer_idx) &lt;= 12
112
113 def freeze(self):
114 self.transformer = self.transformer.eval()
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/encoders/condition.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/encoders/condition.py</a><br>
<b>Line number: </b>187<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>
183
184 def __init__(self, arch=&quot;ViT-H-14&quot;, version=&quot;laion2b_s32b_b79k&quot;, device=&quot;cuda&quot;, max_length=77,
185 freeze=True, layer=&quot;last&quot;):
186 super().__init__()
187 assert layer in self.LAYERS
188 model, _, _ = open_clip.create_model_and_transforms(arch, device=torch.device(&#x27;cpu&#x27;), pretrained=version)
189 del model.visual
190 self.model = model
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py</a><br>
<b>Line number: </b>105<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>
101
102 return x+h_
103
104 def make_attn(in_channels, attn_type=&quot;vanilla&quot;, device=None, dtype=None):
105 assert attn_type in [&quot;vanilla&quot;, &quot;linear&quot;, &quot;none&quot;], f&#x27;attn_type {attn_type} unknown&#x27;
106 #print(f&quot;making attention of type &#x27;{attn_type}&#x27; with {in_channels} in_channels&quot;)
107 if attn_type == &quot;vanilla&quot;:
108 return AttnBlock(in_channels, device=device, dtype=dtype)
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py</a><br>
<b>Line number: </b>169<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>
165 Build sinusoidal embeddings.
166 This matches the implementation in tensor2tensor, but differs slightly
167 from the description in Section 3.5 of &quot;Attention Is All You Need&quot;.
168 &quot;&quot;&quot;
169 assert len(timesteps.shape) == 1
170
171 half_dim = embedding_dim // 2
172 emb = math.log(10000) / (half_dim - 1)
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py</a><br>
<b>Line number: </b>419<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>
415 # assume aligned context, cat along channel axis
416 x = torch.cat((x, context), dim=1)
417 if self.use_timestep:
418 # timestep embedding
419 assert t is not None
420 temb = get_timestep_embedding(t, self.ch)
421 temb = self.temb.dense[0](temb)
422 temb = nonlinearity(temb)
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py</a><br>
<b>Line number: </b>918<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>
914
915 class Upsampler(nn.Module):
916 def __init__(self, in_size, out_size, in_channels, out_channels, ch_mult=2, device=None, dtype=None, operations=ops):
917 super().__init__()
918 assert out_size &gt;= in_size
919 num_blocks = int(np.log2(out_size//in_size))+1
920 factor_up = 1.+ (out_size % in_size)
921 print(f&quot;Building {self.__class__.__name__} with in_size: {in_size} --&gt; out_size {out_size} and factor {factor_up}&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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py</a><br>
<b>Line number: </b>942<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>
938 self.mode = mode
939 if self.with_conv:
940 print(f&quot;Note: {self.__class__.__name} uses learned downsampling and will ignore the fixed {mode} mode&quot;)
941 raise NotImplementedError()
942 assert in_channels is not None
943 # no asymmetric padding in torch conv, must do it ourselves
944 self.conv = operations.Conv2d(in_channels,
945 in_channels,
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py</a><br>
<b>Line number: </b>972<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>
968 dtype=None,
969 operations=ops):
970 super().__init__()
971 if pretrained_config is None:
972 assert pretrained_model is not None, &#x27;Either &quot;pretrained_model&quot; or &quot;pretrained_config&quot; must not be None&#x27;
973 self.pretrained_model = pretrained_model
974 else:
975 assert pretrained_config is not None, &#x27;Either &quot;pretrained_model&quot; or &quot;pretrained_config&quot; must not be None&#x27;
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/ae_modules.py</a><br>
<b>Line number: </b>975<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>
971 if pretrained_config is None:
972 assert pretrained_model is not None, &#x27;Either &quot;pretrained_model&quot; or &quot;pretrained_config&quot; must not be None&#x27;
973 self.pretrained_model = pretrained_model
974 else:
975 assert pretrained_config is not None, &#x27;Either &quot;pretrained_model&quot; or &quot;pretrained_config&quot; must not be None&#x27;
976 self.instantiate_pretrained(pretrained_config)
977
978 self.do_reshape = reshape
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py</a><br>
<b>Line number: </b>81<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>
77 self.op = operations.conv_nd(
78 dims, self.channels, self.out_channels, 3, stride=stride, padding=padding
79 )
80 else:
81 assert self.channels == self.out_channels
82 self.op = avg_pool_nd(dims, kernel_size=stride, stride=stride)
83
84 def forward(self, x):
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.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 assert self.channels == self.out_channels
82 self.op = avg_pool_nd(dims, kernel_size=stride, stride=stride)
83
84 def forward(self, x):
85 assert x.shape[1] == self.channels
86 return self.op(x)
87
88 class Upsample(nn.Module):
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py</a><br>
<b>Line number: </b>107<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>
103 if use_conv:
104 self.conv = operations.conv_nd(dims, self.channels, self.out_channels, 3, padding=padding, dtype=dtype, device=device)
105
106 def forward(self, x):
107 assert x.shape[1] == self.channels
108 if self.dims == 3:
109 x = F.interpolate(x, (x.shape[2], x.shape[3] * 2, x.shape[4] * 2), mode=&#x27;nearest&#x27;)
110 else:
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.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 operations=ops
424 ):
425 super(UNetModel, self).__init__()
426 if num_heads == -1:
427 assert num_head_channels != -1, &#x27;Either num_heads or num_head_channels has to be set&#x27;
428 if num_head_channels == -1:
429 assert num_heads != -1, &#x27;Either num_heads or num_head_channels has to be set&#x27;
430
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.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 super(UNetModel, self).__init__()
426 if num_heads == -1:
427 assert num_head_channels != -1, &#x27;Either num_heads or num_head_channels has to be set&#x27;
428 if num_head_channels == -1:
429 assert num_heads != -1, &#x27;Either num_heads or num_head_channels has to be set&#x27;
430
431 self.in_channels = in_channels
432 self.model_channels = model_channels
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/networks/openaimodel3d.py</a><br>
<b>Line number: </b>785<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>
781 patch = transformer_patches[&quot;input_block_patch_after_skip&quot;]
782 for p in patch:
783 h = p(h, transformer_options)
784 if features_adapter is not None:
785 assert len(features_adapter)==adapter_idx, &#x27;Wrong features_adapter&#x27;
786 transformer_options[&quot;block&quot;] = (&quot;middle&quot;, 0)
787 h = forward_timestep_embed(
788 self.middle_block,
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py</a><br>
<b>Line number: </b>408<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>
404 self.has_pos_emb = position_infused_attn
405 self.pia_pos_emb = FixedPositionalEmbedding(dim) if position_infused_attn else None
406 self.rotary_pos_emb = always(None)
407
408 assert rel_pos_num_buckets &lt;= rel_pos_max_distance, &#x27;number of relative position buckets must be less than the relative position max distance&#x27;
409 self.rel_pos = None
410
411 self.pre_norm = pre_norm
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py</a><br>
<b>Line number: </b>437<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>
433 if exists(custom_layers):
434 layer_types = custom_layers
435 elif exists(par_ratio):
436 par_depth = depth * len(default_block)
437 assert 1 &lt; par_ratio &lt;= par_depth, &#x27;par ratio out of range&#x27;
438 default_block = tuple(filter(not_equals(&#x27;f&#x27;), default_block))
439 par_attn = par_depth // par_ratio
440 depth_cut = par_depth * 2 // 3 # 2 / 3 attention layer cutoff suggested by PAR paper
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py</a><br>
<b>Line number: </b>442<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>
438 default_block = tuple(filter(not_equals(&#x27;f&#x27;), default_block))
439 par_attn = par_depth // par_ratio
440 depth_cut = par_depth * 2 // 3 # 2 / 3 attention layer cutoff suggested by PAR paper
441 par_width = (depth_cut + depth_cut // par_attn) // par_attn
442 assert len(default_block) &lt;= par_width, &#x27;default block is too large for par_ratio&#x27;
443 par_block = default_block + (&#x27;f&#x27;,) * (par_width - len(default_block))
444 par_head = par_block * par_attn
445 layer_types = par_head + (&#x27;f&#x27;,) * (par_depth - len(par_head))
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py</a><br>
<b>Line number: </b>447<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>
443 par_block = default_block + (&#x27;f&#x27;,) * (par_width - len(default_block))
444 par_head = par_block * par_attn
445 layer_types = par_head + (&#x27;f&#x27;,) * (par_depth - len(par_head))
446 elif exists(sandwich_coef):
447 assert sandwich_coef &gt; 0 and sandwich_coef &lt;= depth, &#x27;sandwich coefficient should be less than the depth&#x27;
448 layer_types = (&#x27;a&#x27;,) * sandwich_coef + default_block * (depth - sandwich_coef) + (&#x27;f&#x27;,) * sandwich_coef
449 else:
450 layer_types = default_block * depth
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py</a><br>
<b>Line number: </b>542<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>
538
539
540 class Encoder(AttentionLayers):
541 def __init__(self, **kwargs):
542 assert &#x27;causal&#x27; not in kwargs, &#x27;cannot set causality on encoder&#x27;
543 super().__init__(causal=False, **kwargs)
544
545
</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-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/dynamiCrafter/lvdm/modules/x_transformer.py</a><br>
<b>Line number: </b>562<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>
558 tie_embedding=False,
559 use_pos_emb=True
560 ):
561 super().__init__()
562 assert isinstance(attn_layers, AttentionLayers), &#x27;attention layers must be one of Encoder or Decoder&#x27;
563
564 dim = attn_layers.dim
565 emb_dim = default(emb_dim, dim)
</pre>
</div>
</div>
</div>
<div id="issue-106">
<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-Easy-Use/py/easyNodes.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/easyNodes.py</a><br>
<b>Line number: </b>8<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>
4 import numpy as np
5 import comfy.utils, comfy.sample, comfy.samplers, comfy.controlnet, comfy.model_base, comfy.model_management
6 try:
7 import comfy.sampler_helpers, comfy.supported_models
8 except:
9 pass
10 from comfy.sd import CLIP, VAE
11 from comfy.model_patcher import ModelPatcher
12 from comfy_extras.chainner_models import model_loading
</pre>
</div>
</div>
</div>
<div id="issue-107">
<div class="issue-block issue-sev-medium">
<b>blacklist: </b> Audit url open for permitted schemes. Allowing use of file:/ or custom schemes is often unexpected.<br>
<b>Test ID:</b> B310<br>
<b>Severity: </b>MEDIUM<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/22.html" target="_blank">CWE-22</a><br>
<b>File: </b><a href="/custom_nodes/Hanzo Studio-Easy-Use/py/easyNodes.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/easyNodes.py</a><br>
<b>Line number: </b>396<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b310-urllib-urlopen" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b310-urllib-urlopen</a><br>
<div class="code">
<pre>
392 def INPUT_TYPES(s):
393 max_float_value = 1.95
394 prompt_path = os.path.join(RESOURCES_DIR, &#x27;portrait_prompt.json&#x27;)
395 if not os.path.exists(prompt_path):
396 response = urlopen(&#x27;https://raw.githubusercontent.com/yolain/Hanzo Studio-Easy-Use/main/resources/portrait_prompt.json&#x27;)
397 temp_prompt = json.loads(response.read())
398 prompt_serialized = json.dumps(temp_prompt, indent=4)
399 with open(prompt_path, &quot;w&quot;) as f:
</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-Easy-Use/py/easyNodes.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/easyNodes.py</a><br>
<b>Line number: </b>1970<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>
1966
1967 def hydit_pipeloader(self, ckpt_name, model_name, vae_name, clip_name, mt5_name, device, dtype, resolution, empty_latent_width, empty_latent_height, positive, negative, batch_size, prompt=None, my_unique_id=None):
1968 dtype = string_to_dtype(dtype, &quot;text_encoder&quot;)
1969 if device == &quot;cpu&quot;:
1970 assert dtype in [None, torch.float32,
1971 torch.bfloat16], f&quot;Can&#x27;t use dtype &#x27;{dtype}&#x27; with CPU! Set dtype to &#x27;default&#x27; or &#x27;bf16&#x27;.&quot;
1972
1973 # Clean models from loaded_objects
1974 easyCache.update_loaded_objects(prompt)
</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-Easy-Use/py/human_parsing/transforms.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/human_parsing/transforms.py</a><br>
<b>Line number: </b>31<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>
27 def flip_back(output_flipped, matched_parts):
28 &#x27;&#x27;&#x27;
29 ouput_flipped: numpy.ndarray(batch_size, num_joints, height, width)
30 &#x27;&#x27;&#x27;
31 assert output_flipped.ndim == 4,\
32 &#x27;output_flipped should be [batch_size, num_joints, height, width]&#x27;
33
34 output_flipped = output_flipped[:, :, :, ::-1]
35
</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-Easy-Use/py/ic_light/__init__.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/ic_light/__init__.py</a><br>
<b>Line number: </b>26<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>
22
23
24 class VAEEncodeArgMax(VAEEncode):
25 def encode(self, vae, pixels):
26 assert isinstance(
27 vae.first_stage_model, AutoencoderKL
28 ), &quot;ArgMax only supported for AutoencoderKL&quot;
29 original_sample_mode = vae.first_stage_model.regularization.sample
30 vae.first_stage_model.regularization.sample = False
31 ret = super().encode(vae, pixels)
</pre>
</div>
</div>
</div>
<div id="issue-111">
<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-Easy-Use/py/ic_light/__init__.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/ic_light/__init__.py</a><br>
<b>Line number: </b>147<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>
143
144 def apply(self, ic_model_path, model: ModelPatcher, c_concat: dict, ic_model=None) -&gt; Tuple[ModelPatcher]:
145 try:
146 ModelPatcher.calculate_weight = calculate_weight_adjust_channel(ModelPatcher.calculate_weight)
147 except:
148 pass
149
150 device = comfy.model_management.get_torch_device()
151 dtype = comfy.model_management.unet_dtype()
</pre>
</div>
</div>
</div>
<div id="issue-112">
<div class="issue-block issue-sev-medium">
<b>blacklist: </b> Use of possibly insecure function - consider using safer ast.literal_eval.<br>
<b>Test ID:</b> B307<br>
<b>Severity: </b>MEDIUM<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-Easy-Use/py/kolors/chatglm/modeling_chatglm.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/modeling_chatglm.py</a><br>
<b>Line number: </b>1019<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b307-eval" target="_blank">https://bandit.readthedocs.io/en/1.7.9/blacklists/blacklist_calls.html#b307-eval</a><br>
<div class="code">
<pre>
1015 if history[0][&quot;role&quot;] == &quot;system&quot; and &quot;tools&quot; in history[0]:
1016 content = &quot;\n&quot;.join(content.split(&quot;\n&quot;)[1:-1])
1017 def tool_call(**kwargs):
1018 return kwargs
1019 parameters = eval(content)
1020 content = {&quot;name&quot;: metadata.strip(), &quot;parameters&quot;: parameters}
1021 else:
1022 content = {&quot;name&quot;: metadata.strip(), &quot;content&quot;: content}
</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-Easy-Use/py/kolors/chatglm/quantization.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/quantization.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 def compress_int4_weight(weight: torch.Tensor): # (n, m)
68 with torch.cuda.device(weight.device):
69 n, m = weight.size(0), weight.size(1)
70 assert m % 2 == 0
71 m = m // 2
72 out = torch.empty(n, m, dtype=torch.int8, device=&quot;cuda&quot;)
73 stream = torch.cuda.current_stream()
</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-Easy-Use/py/kolors/chatglm/quantization.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/quantization.py</a><br>
<b>Line number: </b>89<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>
85 return out
86
87
88 def extract_weight_to_half(weight: torch.Tensor, scale_list: torch.Tensor, source_bit_width: int):
89 assert scale_list.dtype in [torch.half, torch.bfloat16]
90 assert weight.dtype in [torch.int8]
91 if source_bit_width == 8:
92 return weight.to(scale_list.dtype) * scale_list[:, None]
</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-Easy-Use/py/kolors/chatglm/quantization.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/quantization.py</a><br>
<b>Line number: </b>90<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>
86
87
88 def extract_weight_to_half(weight: torch.Tensor, scale_list: torch.Tensor, source_bit_width: int):
89 assert scale_list.dtype in [torch.half, torch.bfloat16]
90 assert weight.dtype in [torch.int8]
91 if source_bit_width == 8:
92 return weight.to(scale_list.dtype) * scale_list[:, None]
93 elif source_bit_width == 4:
</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-Easy-Use/py/kolors/chatglm/quantization.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/quantization.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 func = (
95 kernels.int4WeightExtractionHalf if scale_list.dtype == torch.half else kernels.int4WeightExtractionBFloat16
96 )
97 else:
98 assert False, &quot;Unsupported bit-width&quot;
99
100 with torch.cuda.device(weight.device):
101 n, m = weight.size(0), weight.size(1)
</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-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py</a><br>
<b>Line number: </b>13<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>
9
10 class SPTokenizer:
11 def __init__(self, model_path: str):
12 # reload tokenizer
13 assert os.path.isfile(model_path), model_path
14 self.sp_model = SentencePieceProcessor(model_file=model_path)
15
16 # BOS / EOS token IDs
</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-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py</a><br>
<b>Line number: </b>21<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>
17 self.n_words: int = self.sp_model.vocab_size()
18 self.bos_id: int = self.sp_model.bos_id()
19 self.eos_id: int = self.sp_model.eos_id()
20 self.pad_id: int = self.sp_model.unk_id()
21 assert self.sp_model.vocab_size() == self.sp_model.get_piece_size()
22
23 role_special_tokens = [&quot;&lt;|system|&gt;&quot;, &quot;&lt;|user|&gt;&quot;, &quot;&lt;|assistant|&gt;&quot;, &quot;&lt;|observation|&gt;&quot;]
24 special_tokens = [&quot;[MASK]&quot;, &quot;[gMASK]&quot;, &quot;[sMASK]&quot;, &quot;sop&quot;, &quot;eop&quot;] + role_special_tokens
</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-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py</a><br>
<b>Line number: </b>49<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>
45 else:
46 return self.sp_model.EncodeAsPieces(s)
47
48 def encode(self, s: str, bos: bool = False, eos: bool = False) -&gt; List[int]:
49 assert type(s) is str
50 t = self.sp_model.encode(s)
51 if bos:
52 t = [self.bos_id] + t
</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-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py</a><br>
<b>Line number: </b>114<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>
110
111 def get_command(self, token):
112 if token in self.special_tokens:
113 return self.special_tokens[token]
114 assert token in self.tokenizer.special_tokens, f&quot;{token} is not a special token for {self.name}&quot;
115 return self.tokenizer.special_tokens[token]
116
117 @property
</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-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py</a><br>
<b>Line number: </b>194<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>
190 prefix_tokens = [self.get_command(&quot;[gMASK]&quot;), self.get_command(&quot;sop&quot;)]
191 return prefix_tokens
192
193 def build_single_message(self, role, metadata, message):
194 assert role in [&quot;system&quot;, &quot;user&quot;, &quot;assistant&quot;, &quot;observation&quot;], role
195 role_tokens = [self.get_command(f&quot;&lt;|{role}|&gt;&quot;)] + self.tokenizer.encode(f&quot;{metadata}\n&quot;)
196 message_tokens = self.tokenizer.encode(message)
197 tokens = role_tokens + message_tokens
</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-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/chatglm/tokenization_chatglm.py</a><br>
<b>Line number: </b>270<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>
266 return_attention_mask:
267 (optional) Set to False to avoid returning attention mask (default: set to model specifics)
268 &quot;&quot;&quot;
269 # Load from model defaults
270 assert self.padding_side == &quot;left&quot;
271
272 required_input = encoded_inputs[self.model_input_names[0]]
273 seq_length = len(required_input)
</pre>
</div>
</div>
</div>
<div id="issue-123">
<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-Easy-Use/py/kolors/loader.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/loader.py</a><br>
<b>Line number: </b>106<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>
102 try:
103 from accelerate import init_empty_weights
104 from accelerate.utils import set_module_tensor_to_device
105 is_accelerate_available = True
106 except:
107 pass
108
109 from contextlib import nullcontext
110 with (init_empty_weights() if is_accelerate_available else nullcontext()):
</pre>
</div>
</div>
</div>
<div id="issue-124">
<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-Easy-Use/py/kolors/text_encode.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/kolors/text_encode.py</a><br>
<b>Line number: </b>16<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>
12 # Function to randomly select an option from the brackets
13
14 def choose_random_option(match):
15 options = match.group(1).split(&#x27;|&#x27;)
16 return random.choice(options)
17
18 prompt = re.sub(r&#x27;\{([^{}]*)\}&#x27;, choose_random_option, prompt)
19
</pre>
</div>
</div>
</div>
<div id="issue-125">
<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-Easy-Use/py/layer_diffuse/__init__.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/layer_diffuse/__init__.py</a><br>
<b>Line number: </b>63<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>
59 image = image.movedim(-1, 1)
60
61 try:
62 ModelPatcher.calculate_weight = calculate_weight_adjust_channel(ModelPatcher.calculate_weight)
63 except:
64 pass
65
66 if method in [LayerMethod.FG_ONLY_CONV, LayerMethod.FG_ONLY_ATTN] and sd_version == &#x27;sd1&#x27;:
67 self.frames = 1
</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-Easy-Use/py/layer_diffuse/model.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/layer_diffuse/model.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 def decode_pixel(
306 self, pixel: torch.TensorType, latent: torch.TensorType
307 ) -&gt; torch.TensorType:
308 # pixel.shape = [B, C=3, H, W]
309 assert pixel.shape[1] == 3
310 pixel_device = pixel.device
311 pixel_dtype = pixel.dtype
312
</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-Easy-Use/py/layer_diffuse/model.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/layer_diffuse/model.py</a><br>
<b>Line number: </b>318<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>
314 latent = latent.to(device=self.load_device, dtype=self.dtype)
315 # y.shape = [B, C=4, H, W]
316 y = self.estimate_augmented(pixel, latent)
317 y = y.clip(0, 1)
318 assert y.shape[1] == 4
319 # Restore image to original device of input image.
320 return y.to(pixel_device, dtype=pixel_dtype)
321
</pre>
</div>
</div>
</div>
<div id="issue-128">
<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-Easy-Use/py/libs/add_resources.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/add_resources.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 show_index: bool = False,
69 follow_symlinks: bool = False,
70 append_version: bool = False,
71 ) -&gt; web.AbstractResource:
72 assert prefix.startswith(&quot;/&quot;)
73 if prefix.endswith(&quot;/&quot;):
74 prefix = prefix[:-1]
75 resource = LimitResource(
</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-Easy-Use/py/libs/colorfix.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/colorfix.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 eps (float): A small value added to the variance to avoid
45 divide-by-zero. Default: 1e-5.
46 &quot;&quot;&quot;
47 size = feat.size()
48 assert len(size) == 4, &#x27;The input feature should be 4D tensor.&#x27;
49 b, c = size[:2]
50 feat_var = feat.view(b, c, -1).var(dim=2) + eps
51 feat_std = feat_var.sqrt().view(b, c, 1, 1)
</pre>
</div>
</div>
</div>
<div id="issue-130">
<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-Easy-Use/py/libs/dynthres_core.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/dynthres_core.py</a><br>
<b>Line number: </b>63<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>
59 mimic_scale = self.interpret_scale(self.mimic_scale, self.mimic_mode, self.mimic_scale_min)
60 cfg_scale = self.interpret_scale(cfg_scale, self.cfg_mode, self.cfg_scale_min)
61 # uncond shape is (batch, 4, height, width)
62 conds_per_batch = cond.shape[0] / uncond.shape[0]
63 assert conds_per_batch == int(conds_per_batch), &quot;Expected # of conds per batch to be constant across batches&quot;
64 cond_stacked = cond.reshape((-1, int(conds_per_batch)) + uncond.shape[1:])
65
66 ### Normal first part of the CFG Scale logic, basically
</pre>
</div>
</div>
</div>
<div id="issue-131">
<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-Easy-Use/py/libs/image.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/image.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 elif self == ResizeMode.INNER_FIT:
118 return 1
119 elif self == ResizeMode.OUTER_FIT:
120 return 2
121 assert False, &quot;NOTREACHED&quot;
122
123
124
</pre>
</div>
</div>
</div>
<div id="issue-132">
<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-Easy-Use/py/libs/loader.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/loader.py</a><br>
<b>Line number: </b>585<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>
581 from comfy.sd3_clip import SD3Tokenizer, SD3ClipModel
582 import copy
583
584 clip = sd3_clip.clone()
585 assert clip.cond_stage_model.t5xxl is not None, &quot;CLIP must have T5 loaded!&quot;
586
587 # remove transformer
588 transformer = clip.cond_stage_model.t5xxl.transformer
</pre>
</div>
</div>
</div>
<div id="issue-133">
<div class="issue-block issue-sev-medium">
<b>yaml_load: </b> Use of unsafe yaml load. Allows instantiation of arbitrary objects. Consider yaml.safe_load().<br>
<b>Test ID:</b> B506<br>
<b>Severity: </b>MEDIUM<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/20.html" target="_blank">CWE-20</a><br>
<b>File: </b><a href="/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py</a><br>
<b>Line number: </b>37<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b506_yaml_load.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b506_yaml_load.html</a><br>
<div class="code">
<pre>
33
34 def getAPIKeys(self):
35 if os.path.isfile(config_path):
36 with open(config_path, &#x27;r&#x27;) as f:
37 data = yaml.load(f, Loader=yaml.FullLoader)
38 if not data:
39 data = {&#x27;STABILITY_API_KEY&#x27;: default_key, &#x27;STABILITY_API_DEFAULT&#x27;:0}
40 with open(config_path, &#x27;w&#x27;) as f:
</pre>
</div>
</div>
</div>
<div id="issue-134">
<div class="issue-block issue-sev-medium">
<b>yaml_load: </b> Use of unsafe yaml load. Allows instantiation of arbitrary objects. Consider yaml.safe_load().<br>
<b>Test ID:</b> B506<br>
<b>Severity: </b>MEDIUM<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/20.html" target="_blank">CWE-20</a><br>
<b>File: </b><a href="/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py</a><br>
<b>Line number: </b>64<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b506_yaml_load.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b506_yaml_load.html</a><br>
<div class="code">
<pre>
60 if len(api_keys) &gt; 0:
61 self.api_keys = api_keys
62 # load and save the yaml file
63 with open(config_path, &#x27;r&#x27;) as f:
64 data = yaml.load(f, Loader=yaml.FullLoader)
65 data[&#x27;STABILITY_API_KEY&#x27;] = api_keys
66 with open(config_path, &#x27;w&#x27;) as f:
67 yaml.dump(data, f)
</pre>
</div>
</div>
</div>
<div id="issue-135">
<div class="issue-block issue-sev-medium">
<b>yaml_load: </b> Use of unsafe yaml load. Allows instantiation of arbitrary objects. Consider yaml.safe_load().<br>
<b>Test ID:</b> B506<br>
<b>Severity: </b>MEDIUM<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/20.html" target="_blank">CWE-20</a><br>
<b>File: </b><a href="/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py</a><br>
<b>Line number: </b>75<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b506_yaml_load.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b506_yaml_load.html</a><br>
<div class="code">
<pre>
71 if current is not None:
72 self.api_current = current
73 # load and save the yaml file
74 with open(config_path, &#x27;r&#x27;) as f:
75 data = yaml.load(f, Loader=yaml.FullLoader)
76 data[&#x27;STABILITY_API_DEFAULT&#x27;] = current
77 with open(config_path, &#x27;w&#x27;) as f:
78 yaml.dump(data, f)
</pre>
</div>
</div>
</div>
<div id="issue-136">
<div class="issue-block issue-sev-medium">
<b>request_without_timeout: </b> Requests call without timeout<br>
<b>Test ID:</b> B113<br>
<b>Severity: </b>MEDIUM<br>
<b>Confidence: </b>LOW<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/400.html" target="_blank">CWE-400</a><br>
<b>File: </b><a href="/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py</a><br>
<b>Line number: </b>104<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b113_request_without_timeout.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b113_request_without_timeout.html</a><br>
<div class="code">
<pre>
100 image_byte = pil2byte(pil_image)
101 files = {&quot;image&quot;: (&quot;output.png&quot;, image_byte, &#x27;image/png&#x27;)}
102 data[&#x27;strength&#x27;] = strength
103
104 response = requests.post(url,
105 headers={&quot;authorization&quot;: f&quot;{api_key}&quot;, &quot;accept&quot;: &quot;application/json&quot;},
106 files=files,
107 data=data,
108 )
109 if response.status_code == 200:
110 PromptServer.instance.send_sync(&#x27;stable-diffusion-api-generate-succeed&#x27;,{&quot;model&quot;:model})
111 json_data = response.json()
</pre>
</div>
</div>
</div>
<div id="issue-137">
<div class="issue-block issue-sev-medium">
<b>request_without_timeout: </b> Requests call without timeout<br>
<b>Test ID:</b> B113<br>
<b>Severity: </b>MEDIUM<br>
<b>Confidence: </b>LOW<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/400.html" target="_blank">CWE-400</a><br>
<b>File: </b><a href="/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py</a><br>
<b>Line number: </b>133<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b113_request_without_timeout.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b113_request_without_timeout.html</a><br>
<div class="code">
<pre>
129 name = self.api_keys[self.api_current][&#x27;name&#x27;]
130 if cache and name in self.user_info:
131 return self.user_info[name]
132 else:
133 response = requests.get(url, headers={&quot;Authorization&quot;: f&quot;Bearer {api_key}&quot;})
134 if response.status_code == 200:
135 user_info = response.json()
136 self.user_info[name] = user_info
</pre>
</div>
</div>
</div>
<div id="issue-138">
<div class="issue-block issue-sev-medium">
<b>request_without_timeout: </b> Requests call without timeout<br>
<b>Test ID:</b> B113<br>
<b>Severity: </b>MEDIUM<br>
<b>Confidence: </b>LOW<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/400.html" target="_blank">CWE-400</a><br>
<b>File: </b><a href="/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/stability.py</a><br>
<b>Line number: </b>146<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b113_request_without_timeout.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b113_request_without_timeout.html</a><br>
<div class="code">
<pre>
142 # get user balance
143 async def getUserBalance(self):
144 url = f&quot;{self.api_url}/v1/user/balance&quot;
145 api_key = self.api_keys[self.api_current][&#x27;key&#x27;]
146 response = requests.get(url, headers={
147 &quot;Authorization&quot;: f&quot;Bearer {api_key}&quot;
148 })
149 if response.status_code == 200:
150 return response.json()
151 else:
</pre>
</div>
</div>
</div>
<div id="issue-139">
<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-Easy-Use/py/libs/utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/utils.py</a><br>
<b>Line number: </b>52<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>
48 except:
49 run_install = False
50
51 if run_install:
52 import subprocess
53 package_command = package + &#x27;==&#x27; + v if v is not None else package
54 PromptServer.instance.send_sync(&quot;easyuse-toast&quot;, {&#x27;content&#x27;: f&quot;Installing {package_command}...&quot;, &#x27;duration&#x27;: 5000})
55 result = subprocess.run([sys.executable, &#x27;-s&#x27;, &#x27;-m&#x27;, &#x27;pip&#x27;, &#x27;install&#x27;, package_command], capture_output=True, text=True)
</pre>
</div>
</div>
</div>
<div id="issue-140">
<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-Easy-Use/py/libs/utils.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/utils.py</a><br>
<b>Line number: </b>55<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>
51 if run_install:
52 import subprocess
53 package_command = package + &#x27;==&#x27; + v if v is not None else package
54 PromptServer.instance.send_sync(&quot;easyuse-toast&quot;, {&#x27;content&#x27;: f&quot;Installing {package_command}...&quot;, &#x27;duration&#x27;: 5000})
55 result = subprocess.run([sys.executable, &#x27;-s&#x27;, &#x27;-m&#x27;, &#x27;pip&#x27;, &#x27;install&#x27;, package_command], capture_output=True, text=True)
56 if result.returncode == 0:
57 PromptServer.instance.send_sync(&quot;easyuse-toast&quot;, {&#x27;content&#x27;: f&quot;{package} installed successfully&quot;, &#x27;type&#x27;: &#x27;success&#x27;, &#x27;duration&#x27;: 5000})
58 print(f&quot;Package {package} installed successfully&quot;)
</pre>
</div>
</div>
</div>
<div id="issue-141">
<div class="issue-block issue-sev-medium">
<b>yaml_load: </b> Use of unsafe yaml load. Allows instantiation of arbitrary objects. Consider yaml.safe_load().<br>
<b>Test ID:</b> B506<br>
<b>Severity: </b>MEDIUM<br>
<b>Confidence: </b>HIGH<br>
<b>CWE: </b><a href="https://cwe.mitre.org/data/definitions/20.html" target="_blank">CWE-20</a><br>
<b>File: </b><a href="/custom_nodes/Hanzo Studio-Easy-Use/py/libs/wildcards.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/wildcards.py</a><br>
<b>Line number: </b>47<br>
<b>More info: </b><a href="https://bandit.readthedocs.io/en/1.7.9/plugins/b506_yaml_load.html" target="_blank">https://bandit.readthedocs.io/en/1.7.9/plugins/b506_yaml_load.html</a><br>
<div class="code">
<pre>
43 easy_wildcard_dict[key] = lines
44 elif file.endswith(&#x27;.yaml&#x27;):
45 file_path = os.path.join(root, file)
46 with open(file_path, &#x27;r&#x27;) as f:
47 yaml_data = yaml.load(f, Loader=yaml.FullLoader)
48
49 for k, v in yaml_data.items():
50 read_wildcard(k, v)
</pre>
</div>
</div>
</div>
<div id="issue-142">
<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-Easy-Use/py/libs/wildcards.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/wildcards.py</a><br>
<b>Line number: </b>129<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>
125
126 if select_range is None:
127 select_count = 1
128 else:
129 select_count = random.randint(select_range[0], select_range[1])
130
131 if select_count &gt; len(options):
132 selected_items = options
</pre>
</div>
</div>
</div>
<div id="issue-143">
<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-Easy-Use/py/libs/wildcards.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/wildcards.py</a><br>
<b>Line number: </b>134<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>
130
131 if select_count &gt; len(options):
132 selected_items = options
133 else:
134 selected_items = random.choices(options, weights=normalized_probabilities, k=select_count)
135 selected_items = set(selected_items)
136
137 try_count = 0
</pre>
</div>
</div>
</div>
<div id="issue-144">
<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-Easy-Use/py/libs/wildcards.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/wildcards.py</a><br>
<b>Line number: </b>140<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>
136
137 try_count = 0
138 while len(selected_items) &lt; select_count and try_count &lt; 10:
139 remaining_count = select_count - len(selected_items)
140 additional_items = random.choices(options, weights=normalized_probabilities, k=remaining_count)
141 selected_items |= set(additional_items)
142 try_count += 1
143
</pre>
</div>
</div>
</div>
<div id="issue-145">
<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-Easy-Use/py/libs/wildcards.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/wildcards.py</a><br>
<b>Line number: </b>167<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>
163 for match in matches:
164 keyword = match.lower()
165 keyword = wildcard_normalize(keyword)
166 if keyword in easy_wildcard_dict:
167 replacement = random.choice(easy_wildcard_dict[keyword])
168 replacements_found = True
169 string = string.replace(f&quot;__{match}__&quot;, replacement, 1)
170 elif &#x27;*&#x27; in keyword:
</pre>
</div>
</div>
</div>
<div id="issue-146">
<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-Easy-Use/py/libs/wildcards.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/libs/wildcards.py</a><br>
<b>Line number: </b>180<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>
176 total_patterns += v
177 found = True
178
179 if found:
180 replacement = random.choice(total_patterns)
181 replacements_found = True
182 string = string.replace(f&quot;__{match}__&quot;, replacement, 1)
183 elif &#x27;/&#x27; not in keyword:
</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-Easy-Use/py/logic.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/logic.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 return True, None, None
34 def error_if_mismatched_list_args(args: Dict[str, List[Any]]) -&gt; None:
35 is_valid, failed_key1, failed_key2 = validate_list_args(args)
36 if not is_valid:
37 assert failed_key1 is not None
38 assert failed_key2 is not None
39 raise ValueError(
40 f&quot;Mismatched list inputs received. {failed_key1}({len(args[failed_key1])}) !== {failed_key2}({len(args[failed_key2])})&quot;
</pre>
</div>
</div>
</div>
<div id="issue-148">
<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-Easy-Use/py/logic.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/logic.py</a><br>
<b>Line number: </b>38<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>
34 def error_if_mismatched_list_args(args: Dict[str, List[Any]]) -&gt; None:
35 is_valid, failed_key1, failed_key2 = validate_list_args(args)
36 if not is_valid:
37 assert failed_key1 is not None
38 assert failed_key2 is not None
39 raise ValueError(
40 f&quot;Mismatched list inputs received. {failed_key1}({len(args[failed_key1])}) !== {failed_key2}({len(args[failed_key2])})&quot;
41 )
</pre>
</div>
</div>
</div>
<div id="issue-149">
<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-Easy-Use/py/server.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/server.py</a><br>
<b>Line number: </b>37<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>
33 self.base_value -= 1
34 if self.base_value &lt; 0:
35 self.base_value = 1125899906842624
36 elif self.action == SGmode.RAND:
37 self.base_value = random.randint(0, 1125899906842624)
38
39 return seed
40
</pre>
</div>
</div>
</div>
<div id="issue-150">
<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-Easy-Use/py/server.py" target="_blank">/custom_nodes/Hanzo Studio-Easy-Use/py/server.py</a><br>
<b>Line number: </b>55<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>
51 value = value - 1
52 if value &lt; 0:
53 value = 1125899906842624
54 elif action == &#x27;randomize&#x27; or action == &#x27;randomize for each node&#x27;:
55 value = random.randint(0, 1125899906842624)
56 if seed_is_global:
57 v[&#x27;inputs&#x27;][&#x27;value&#x27;] = value
58
</pre>
</div>
</div>
</div>
</div>
</body>
</html>