The ultimate training toolkit for finetuning diffusion models
This report presents the forensic synthetic code analysis of ostris/ai-toolkit, a Python project with 11,846 GitHub stars. SynthScan v2.0 examined 190,743 lines of code across 635 source files, recording 2134 pattern matches distributed across 21 syntactic categories. The overall adjusted score of 16.1 places this repository in the Moderate AI signal band.
The scanner applied 160+ deterministic lexical heuristics, multi-line block detectors, abstract syntax tree depth profilers, and a cross-file Jaccard similarity matrix to construct a statistically normalised synthetic code estimate. All matches are individually weighted by severity coefficient and contextual multiplier before summation, and the resulting headline score is temporally discounted to account for the repository's development history relative to the commercial emergence of large language model coding tooling (November 2022 onward).
This chart maps the temporal evolution of the adjusted synthetic code score across successive scan runs. An upward trajectory indicates ongoing incorporation of AI-generated code or expanding LLM-assisted scaffolding; a stable or declining trajectory may reflect active human refactoring, code removal, or the adoption of stricter authorship policies. The dashed secondary line (right axis) independently tracks total raw pattern hit count, which can diverge from the normalised score when codebase size changes significantly between scans.
Classifies detected patterns by their diagnostic confidence and structural impact. CRITICAL patterns (coefficient 10) represent definitive synthetic signatures — hallucinated imports, explicit LLM attribution metadata — virtually never produced by human authors. HIGH (5) indicates strong structural tells such as cross-file repetition or cross-linguistic idioms. MEDIUM (2) covers recognisable conversational padding and AI-specific vocabulary. LOW (1) captures subtle indicators like tautological comments and generic boilerplate that require density to carry independent signal.
This horizontal bar chart decomposes the repository's raw synthetic code score by top-level directory, allowing you to pinpoint precisely which modules or components carry the highest AI authorship density. Directories with disproportionately high scores relative to their size warrant targeted manual review: concentrated AI signatures often trace back to mass-generated configuration layers, auto-ported test suites, LLM-scaffolded boilerplate classes, or entire subsystems authored under heavy copilot assistance. Use this view to prioritise your human code-review effort.
The scanner identified 2134 distinct pattern matches across 21 syntactic categories. Each entry below represents a discrete location in the source code where the engine recorded a statistically significant AI authorship indicator. Expand any category row to inspect the individual file paths, line numbers, code snippets, and the lexical context (CODE, COMMENT, or STRING) in which each match was detected.
Reading the findings table: The Severity column indicates the diagnostic confidence level (CRITICAL / HIGH / MEDIUM / LOW). The Context column identifies whether the match occurred inside executable code, an inline comment, or a string literal — comment-context matches receive a ×1.5 weight because LLMs systematically over-annotate. The ⚡ bolt icon marks clustered matches: three or more patterns within a 10-line window, each receiving an additional ×1.5 density multiplier as dense clusters constitute far stronger evidence of synthetic authorship than isolated hits.
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | extensions_built_in/audio_models/base_audio_model.py | 1 | CODE | |
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 6 | CODE | |
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 6 | CODE | |
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 14 | CODE | |
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 17 | CODE | |
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 17 | CODE | |
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 24 | CODE | |
| LOW | extensions_built_in/audio_models/ace_step/__init__.py | 1 | CODE | |
| LOW | extensions_built_in/audio_models/ace_step/__init__.py | 1 | CODE | |
| LOW | …ensions_built_in/audio_models/ace_step/src/pipeline.py | 4 | CODE | |
| LOW | …ensions_built_in/audio_models/ace_step/src/pipeline.py | 5 | CODE | |
| LOW | extensions_built_in/concept_replacer/ConceptReplacer.py | 3 | CODE | |
| LOW | extensions_built_in/concept_replacer/ConceptReplacer.py | 4 | CODE | |
| LOW | extensions_built_in/concept_replacer/ConceptReplacer.py | 4 | CODE | |
| LOW | extensions_built_in/concept_replacer/ConceptReplacer.py | 5 | CODE | |
| LOW | …nsions_built_in/concept_slider/ConceptSliderTrainer.py | 7 | CODE | |
| LOW | extensions_built_in/diffusion_models/krea2/krea2.py | 60 | CODE | |
| LOW | extensions_built_in/diffusion_models/krea2/__init__.py | 1 | CODE | |
| LOW | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 14 | CODE | |
| LOW | extensions_built_in/diffusion_models/chroma/__init__.py | 1 | CODE | |
| LOW | extensions_built_in/diffusion_models/chroma/__init__.py | 2 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 6 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 17 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 17 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 19 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 20 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 21 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 28 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 6 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 9 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 17 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 17 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 19 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 19 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 20 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 21 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 29 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/src/radiance.py | 7 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/src/radiance.py | 7 | CODE | |
| LOW | …nsions_built_in/diffusion_models/ideogram4/__init__.py | 1 | CODE | |
| LOW | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 5 | CODE | |
| LOW | …built_in/diffusion_models/ideogram4/src/latent_norm.py | 1 | CODE | |
| LOW | …built_in/diffusion_models/ideogram4/src/transformer.py | 9 | CODE | |
| LOW | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 8 | CODE | |
| LOW | …ensions_built_in/diffusion_models/ideogram4/src/vae.py | 3 | CODE | |
| LOW | extensions_built_in/diffusion_models/ltx2/__init__.py | 1 | CODE | |
| LOW | extensions_built_in/diffusion_models/ltx2/__init__.py | 1 | CODE | |
| LOW | extensions_built_in/diffusion_models/ltx2/__init__.py | 1 | CODE | |
| LOW | extensions_built_in/diffusion_models/ltx2/ltx2.py | 11 | CODE | |
| LOW | …ns_built_in/diffusion_models/prx_pixel_t2i/__init__.py | 1 | CODE | |
| LOW | …uilt_in/diffusion_models/prx_pixel_t2i/src/__init__.py | 3 | CODE | |
| LOW | …uilt_in/diffusion_models/prx_pixel_t2i/src/__init__.py | 4 | CODE | |
| LOW | …/diffusion_models/nucleus_image/nucleus_image_model.py | 20 | CODE | |
| LOW | …ns_built_in/diffusion_models/nucleus_image/__init__.py | 1 | CODE | |
| LOW | extensions_built_in/diffusion_models/flux2/__init__.py | 1 | CODE | |
| LOW | extensions_built_in/diffusion_models/flux2/__init__.py | 2 | CODE | |
| LOW | extensions_built_in/diffusion_models/flux2/__init__.py | 2 | CODE | |
| LOW | …ensions_built_in/diffusion_models/flux2/flux2_model.py | 30 | CODE | |
| LOW | …ons_built_in/diffusion_models/flux_kontext/__init__.py | 1 | CODE | |
| LOW | …built_in/diffusion_models/flux_kontext/flux_kontext.py | 2 | CODE | |
| 612 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| MEDIUM | run_mac.zsh | 11 | # ── Banner ───────────────────────────────────────────────────────── | COMMENT |
| MEDIUM | run_mac.zsh | 25 | # ── 1. Ensure uv (prebuilt static binary, kept inside the repo) ───── | COMMENT |
| MEDIUM | run_mac.zsh | 54 | # ── 3. Sync the environment and start the UI ──────────────────────── | COMMENT |
| MEDIUM | run_linux.sh | 12 | # ── Banner ───────────────────────────────────────────────────────── | COMMENT |
| MEDIUM | run_linux.sh | 26 | # ── 1. Ensure uv (prebuilt static binary, kept inside the repo) ───── | COMMENT |
| MEDIUM | run_linux.sh | 55 | # ── 3. Sync the environment and start the UI ──────────────────────── | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 29 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 31 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 76 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 78 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 434 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 436 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 524 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 526 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 959 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 961 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 1150 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 1152 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 1177 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 1179 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 1343 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 1345 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 1467 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/audio_models/ace_step/src/model.py | 1469 | # ═══════════════════════════════════════════════════════════════════════════════ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 230 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 232 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 479 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 481 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 542 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 544 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 627 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 629 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 774 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 776 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 845 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | extensions_built_in/diffusion_models/krea2/krea2.py | 847 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 27 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 29 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 61 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 63 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 240 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 242 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 270 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 272 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 211 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 213 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 438 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 440 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 472 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 474 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 553 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 555 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 596 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 598 | # ------------------------------------------------------------------ | COMMENT |
| MEDIUM | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 68 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 77 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 102 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 104 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 161 | # --------------------------------------------------------------------------- | COMMENT |
| MEDIUM | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 163 | # --------------------------------------------------------------------------- | COMMENT |
| 149 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | extensions_built_in/audio_models/base_audio_model.py | 72 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW | extensions_built_in/audio_models/base_audio_model.py | 80 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 237 | def get_transformer_block_names(self) -> Optional[List[str]]: | CODE |
| LOW | extensions_built_in/audio_models/ace_step/src/model.py | 929 | def enable_gradient_checkpointing(self): | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/krea2/krea2.py | 867 | def get_transformer_block_names(self) -> Optional[List[str]]: | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/krea2/krea2.py | 870 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/krea2/krea2.py | 876 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW | extensions_built_in/diffusion_models/krea2/krea2.py | 103 | def patch_qwen_vl_patch_embed(model): | CODE |
| LOW | extensions_built_in/diffusion_models/krea2/krea2.py | 383 | def get_quantization_exclude_modules(self): | CODE |
| LOW | extensions_built_in/diffusion_models/krea2/krea2.py | 608 | def _batch_ref_latents_from_batch( | CODE |
| LOW | extensions_built_in/diffusion_models/krea2/src/mmdit.py | 478 | def enable_gradient_checkpointing(self): | CODE |
| LOW | extensions_built_in/diffusion_models/krea2/src/mmdit.py | 481 | def disable_gradient_checkpointing(self): | CODE |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 446 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 454 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 428 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 436 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW | …tensions_built_in/diffusion_models/chroma/src/model.py | 56 | def modify_mask_to_attend_padding(mask, max_seq_length, num_extra_padding=8): | CODE |
| LOW | …tensions_built_in/diffusion_models/chroma/src/model.py | 178 | def enable_gradient_checkpointing(self, enable: bool = True): | CODE |
| LOW | …sions_built_in/diffusion_models/chroma/src/radiance.py | 70 | def modify_mask_to_attend_padding(mask, max_seq_length, num_extra_padding=8): | CODE |
| LOW | …sions_built_in/diffusion_models/chroma/src/radiance.py | 226 | def enable_gradient_checkpointing(self, enable: bool = True): | CODE |
| LOW | …ensions_built_in/diffusion_models/chroma/src/layers.py | 463 | def _modulation_shift_scale_fn(x, scale, shift): | CODE |
| LOW | …ensions_built_in/diffusion_models/chroma/src/layers.py | 521 | def modulation_shift_scale_fn(self, x, scale, shift): | CODE |
| LOW | …ensions_built_in/diffusion_models/chroma/src/layers.py | 649 | def modulation_shift_scale_fn(self, x, scale, shift): | CODE |
| LOW | …ensions_built_in/diffusion_models/chroma/src/layers.py | 704 | def modulation_shift_scale_fn(self, x, scale, shift): | CODE |
| LOW | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 113 | def _load_component_state_dict(base: str, subfolder: str, basename: str) -> dict: | CODE |
| LOW | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 199 | def text_embedding_space_version(self): | CODE |
| LOW⚡ | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 620 | def get_transformer_block_names(self) -> Optional[List[str]]: | CODE |
| LOW⚡ | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 623 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW⚡ | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 630 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW⚡ | …built_in/diffusion_models/ideogram4/src/transformer.py | 412 | def enable_gradient_checkpointing(self): | CODE |
| LOW⚡ | …built_in/diffusion_models/ideogram4/src/transformer.py | 415 | def disable_gradient_checkpointing(self): | CODE |
| LOW | …ensions_built_in/diffusion_models/ideogram4/src/vae.py | 188 | def enable_gradient_checkpointing(self): | CODE |
| LOW | …ensions_built_in/diffusion_models/ideogram4/src/vae.py | 289 | def enable_gradient_checkpointing(self): | CODE |
| LOW | …ensions_built_in/diffusion_models/ideogram4/src/vae.py | 381 | def enable_gradient_checkpointing(self): | CODE |
| LOW | …ensions_built_in/diffusion_models/ideogram4/src/vae.py | 398 | def convert_diffusers_state_dict(src: dict[str, Tensor]) -> dict[str, Tensor]: | CODE |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 129 | def update_state_dict_inplace( | CODE |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 202 | def split_transformer_and_connector_state_dict( | CODE |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 915 | def get_model_state_dict_from_combined_ckpt( | CODE |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 1064 | def convert_lora_original_to_diffusers( | CODE |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 1098 | def convert_lora_diffusers_to_original( | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/ltx2/ltx2.py | 1191 | def get_transformer_block_names(self) -> Optional[List[str]]: | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/ltx2/ltx2.py | 1194 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/ltx2/ltx2.py | 1202 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW | …ilt_in/diffusion_models/prx_pixel_t2i/prx_pixel_t2i.py | 294 | def get_latent_noise_from_latents(self, latents: torch.Tensor, noise_offset=0.0): | CODE |
| LOW⚡ | …ilt_in/diffusion_models/prx_pixel_t2i/prx_pixel_t2i.py | 332 | def get_transformer_block_names(self) -> Optional[List[str]]: | CODE |
| LOW⚡ | …ilt_in/diffusion_models/prx_pixel_t2i/prx_pixel_t2i.py | 335 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW⚡ | …ilt_in/diffusion_models/prx_pixel_t2i/prx_pixel_t2i.py | 341 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW | …/diffusion_models/prx_pixel_t2i/src/transformer_prx.py | 49 | def maybe_adjust_dtype_for_device( | CODE |
| LOW | …/diffusion_models/prx_pixel_t2i/src/transformer_prx.py | 784 | def _compute_timestep_embedding( | STRING |
| LOW⚡ | …/diffusion_models/nucleus_image/nucleus_image_model.py | 405 | def get_transformer_block_names(self) -> Optional[List[str]]: | CODE |
| LOW⚡ | …/diffusion_models/nucleus_image/nucleus_image_model.py | 408 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW⚡ | …/diffusion_models/nucleus_image/nucleus_image_model.py | 415 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW⚡ | …ensions_built_in/diffusion_models/flux2/flux2_model.py | 505 | def get_transformer_block_names(self) -> Optional[List[str]]: | CODE |
| LOW⚡ | …ensions_built_in/diffusion_models/flux2/flux2_model.py | 508 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW⚡ | …ensions_built_in/diffusion_models/flux2/flux2_model.py | 515 | def convert_lora_weights_before_load(self, state_dict): | CODE |
| LOW | extensions_built_in/diffusion_models/flux2/src/model.py | 141 | def enable_gradient_checkpointing(self): | CODE |
| LOW | …ons_built_in/diffusion_models/flux2/src/autoencoder.py | 194 | def enable_gradient_checkpointing(self): | CODE |
| LOW | …ons_built_in/diffusion_models/flux2/src/autoencoder.py | 295 | def enable_gradient_checkpointing(self): | CODE |
| LOW | …ons_built_in/diffusion_models/flux2/src/autoencoder.py | 398 | def enable_gradient_checkpointing(self): | CODE |
| LOW | …nsions_built_in/diffusion_models/flux2/src/pipeline.py | 93 | def _get_mistral_prompt_embeds( | CODE |
| 317 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| HIGH | extensions_built_in/diffusion_models/krea2/krea2.py | 0 | qwen-vl's vision patch_embed is a conv3d whose kernel == stride, i.e. a plain linear projection of each flattened patch. | STRING |
| HIGH | …built_in/diffusion_models/mageflow/src/text_encoder.py | 0 | qwen-vl's vision patch_embed is a conv3d whose kernel == stride, i.e. a plain linear projection of each flattened patch. | STRING |
| HIGH | extensions_built_in/captioner/Qwen3VLCaptioner.py | 0 | qwen-vl's vision patch_embed is a conv3d whose kernel == stride, i.e. a plain linear projection of each flattened patch. | STRING |
| HIGH | …nsions_built_in/diffusion_models/krea2/src/pipeline.py | 0 | lightweight flow-matching sampler used by ai-toolkit's preview generation. | STRING |
| HIGH | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 0 | lightweight flow-matching sampler used by ai-toolkit's preview generation. | STRING |
| HIGH | …ons_built_in/diffusion_models/mageflow/src/pipeline.py | 0 | lightweight flow-matching sampler used by ai-toolkit's preview generation. | STRING |
| HIGH | …_built_in/diffusion_models/boogu_image/src/pipeline.py | 0 | lightweight flow-matching sampler used by ai-toolkit's preview generation. | STRING |
| HIGH | …ensions_built_in/diffusion_models/chroma/src/layers.py | 0 | create sinusoidal timestep embeddings. :param t: a 1-d tensor of n indices, one per batch element. these may be fraction | STRING |
| HIGH | extensions_built_in/diffusion_models/flux2/src/model.py | 0 | create sinusoidal timestep embeddings. :param t: a 1-d tensor of n indices, one per batch element. these may be fraction | STRING |
| HIGH | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 0 | create sinusoidal timestep embeddings. :param t: a 1-d tensor of n indices, one per batch element. these may be fraction | STRING |
| HIGH | …igen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py | 0 | omnigen2 attention processor module copyright 2025 baai, the omnigen2 team and the huggingface team. all rights reserved | STRING |
| HIGH | …s/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py | 0 | omnigen2 attention processor module copyright 2025 baai, the omnigen2 team and the huggingface team. all rights reserved | STRING |
| HIGH | …sion_models/omnigen2/src/models/attention_processor.py | 0 | omnigen2 attention processor module copyright 2025 baai, the omnigen2 team and the huggingface team. all rights reserved | STRING |
| HIGH | …igen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py | 0 | calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. handles custom t | STRING |
| HIGH | …s/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py | 0 | calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. handles custom t | STRING |
| HIGH | …elines/hidream_image/pipeline_hidream_image_editing.py | 0 | calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. handles custom t | STRING |
| HIGH | …/src/pipelines/hidream_image/pipeline_hidream_image.py | 0 | calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. handles custom t | STRING |
| HIGH | …src/schedulers/scheduling_flow_match_euler_discrete.py | 0 | output class for the scheduler's `step` function output. args: prev_sample (`torch.floattensor` of shape `(batch_size, n | STRING |
| HIGH | …sion_models/hidream/src/schedulers/flash_flow_match.py | 0 | output class for the scheduler's `step` function output. args: prev_sample (`torch.floattensor` of shape `(batch_size, n | STRING |
| HIGH | toolkit/samplers/mean_flow_scheduler.py | 0 | output class for the scheduler's `step` function output. args: prev_sample (`torch.floattensor` of shape `(batch_size, n | STRING |
| HIGH | …src/schedulers/scheduling_flow_match_euler_discrete.py | 0 | the index counter for current timestep. it will increase 1 after each scheduler step. | STRING |
| HIGH | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 0 | the index counter for current timestep. it will increase 1 after each scheduler step. | STRING |
| HIGH | …sion_models/hidream/src/schedulers/flash_flow_match.py | 0 | the index counter for current timestep. it will increase 1 after each scheduler step. | STRING |
| HIGH | …sion_models/hidream/src/schedulers/fm_solvers_unipc.py | 0 | the index counter for current timestep. it will increase 1 after each scheduler step. | STRING |
| HIGH | …src/schedulers/scheduling_flow_match_euler_discrete.py | 0 | the index for the first timestep. it should be set from pipeline with `set_begin_index` method. | STRING |
| HIGH | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 0 | the index for the first timestep. it should be set from pipeline with `set_begin_index` method. | STRING |
| HIGH | …sion_models/hidream/src/schedulers/flash_flow_match.py | 0 | the index for the first timestep. it should be set from pipeline with `set_begin_index` method. | STRING |
| HIGH | …sion_models/hidream/src/schedulers/fm_solvers_unipc.py | 0 | the index for the first timestep. it should be set from pipeline with `set_begin_index` method. | STRING |
| HIGH | …src/schedulers/scheduling_flow_match_euler_discrete.py | 0 | sets the begin index for the scheduler. this function should be run from pipeline before the inference. args: begin_inde | STRING |
| HIGH | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 0 | sets the begin index for the scheduler. this function should be run from pipeline before the inference. args: begin_inde | STRING |
| HIGH | …sion_models/hidream/src/schedulers/flash_flow_match.py | 0 | sets the begin index for the scheduler. this function should be run from pipeline before the inference. args: begin_inde | STRING |
| HIGH | …sion_models/hidream/src/schedulers/fm_solvers_unipc.py | 0 | sets the begin index for the scheduler. this function should be run from pipeline before the inference. args: begin_inde | STRING |
| HIGH | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 0 | "dynamic thresholding: at each sampling step we set s to a certain percentile absolute pixel value in xt0 (the predictio | STRING |
| HIGH | …sion_models/hidream/src/schedulers/fm_solvers_unipc.py | 0 | "dynamic thresholding: at each sampling step we set s to a certain percentile absolute pixel value in xt0 (the predictio | STRING |
| HIGH | toolkit/samplers/custom_lcm_scheduler.py | 0 | "dynamic thresholding: at each sampling step we set s to a certain percentile absolute pixel value in xt0 (the predictio | STRING |
| HIGH | …usion_models/hidream/src/models/attention_processor.py | 0 | attention processor used typically in processing the sd3-like self-attention projections. | STRING |
| HIGH | toolkit/ip_adapter.py | 0 | attention processor used typically in processing the sd3-like self-attention projections. | STRING |
| HIGH | toolkit/models/flux_sage_attn.py | 0 | attention processor used typically in processing the sd3-like self-attention projections. | STRING |
| HIGH | toolkit/models/vd_adapter.py | 0 | attention processor used typically in processing the sd3-like self-attention projections. | STRING |
| HIGH | extensions_built_in/sd_trainer/DiffusionTrainer.py | 0 | start a daemon thread that periodically checks should_stop() and terminates the process immediately when triggered. | STRING |
| HIGH | extensions_built_in/sd_trainer/UITrainer.py | 0 | start a daemon thread that periodically checks should_stop() and terminates the process immediately when triggered. | STRING |
| HIGH | extensions_built_in/captioner/BaseCaptioner.py | 0 | start a daemon thread that periodically checks should_stop() and terminates the process immediately when triggered. | STRING |
| HIGH | extensions_built_in/sd_trainer/DiffusionTrainer.py | 0 | helper method to run an async coroutine and track the task. | STRING |
| HIGH | extensions_built_in/sd_trainer/UITrainer.py | 0 | helper method to run an async coroutine and track the task. | STRING |
| HIGH | extensions_built_in/captioner/BaseCaptioner.py | 0 | helper method to run an async coroutine and track the task. | STRING |
| HIGH | extensions_built_in/sd_trainer/DiffusionTrainer.py | 0 | create a new connection for each operation to avoid locking. | STRING |
| HIGH | extensions_built_in/sd_trainer/UITrainer.py | 0 | create a new connection for each operation to avoid locking. | STRING |
| HIGH | extensions_built_in/captioner/BaseCaptioner.py | 0 | create a new connection for each operation to avoid locking. | STRING |
| HIGH | toolkit/reference_adapter.py | 0 | attention processor for custom te for pytorch 2.0. args: hidden_size (`int`): the hidden size of the attention layer. cr | STRING |
| HIGH | toolkit/util/ip_adapter_utils.py | 0 | attention processor for custom te for pytorch 2.0. args: hidden_size (`int`): the hidden size of the attention layer. cr | STRING |
| HIGH | toolkit/models/te_adapter.py | 0 | attention processor for custom te for pytorch 2.0. args: hidden_size (`int`): the hidden size of the attention layer. cr | STRING |
| HIGH | toolkit/util/ip_adapter_utils.py | 0 | processor for implementing scaled dot-product attention (enabled by default if you're using pytorch 2.0). | STRING |
| HIGH | toolkit/models/single_value_adapter.py | 0 | processor for implementing scaled dot-product attention (enabled by default if you're using pytorch 2.0). | STRING |
| HIGH | toolkit/models/vd_adapter.py | 0 | processor for implementing scaled dot-product attention (enabled by default if you're using pytorch 2.0). | STRING |
| HIGH | toolkit/optimizers/automagic.py | 0 | performs a single optimization step. arguments: closure (callable, optional): a closure that reevaluates the model and r | STRING |
| HIGH | toolkit/optimizers/adafactor.py | 0 | performs a single optimization step. arguments: closure (callable, optional): a closure that reevaluates the model and r | STRING |
| HIGH | toolkit/optimizers/prodigy_8bit.py | 0 | performs a single optimization step. arguments: closure (callable, optional): a closure that reevaluates the model and r | STRING |
| HIGH | toolkit/optimizers/adam8bit.py | 0 | performs a single optimization step. arguments: closure (callable, optional): a closure that reevaluates the model and r | STRING |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | flux_train_ui.py | 25 | CODE | |
| LOW | ui_scripts/upsample_ideogram4_caption.py | 132 | CODE | |
| LOW | ui_scripts/merge_loras.py | 36 | CODE | |
| LOW | extensions_built_in/diffusion_models/chroma/pipeline.py | 148 | CODE | |
| LOW | …sions_built_in/diffusion_models/chroma/chroma_model.py | 98 | CODE | |
| LOW | …lt_in/diffusion_models/chroma/chroma_radiance_model.py | 99 | CODE | |
| LOW | …ensions_built_in/diffusion_models/chroma/src/layers.py | 92 | CODE | |
| LOW | …ns_built_in/diffusion_models/ideogram4/src/pipeline.py | 312 | CODE | |
| LOW | …ensions_built_in/diffusion_models/ideogram4/src/vae.py | 191 | CODE | |
| LOW | …ensions_built_in/diffusion_models/ideogram4/src/vae.py | 292 | CODE | |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 915 | CODE | |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 970 | CODE | |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 1016 | CODE | |
| LOW | extensions_built_in/diffusion_models/ltx2/ltx2.py | 861 | CODE | |
| LOW | …ensions_built_in/diffusion_models/flux2/flux2_model.py | 336 | CODE | |
| LOW | …ons_built_in/diffusion_models/flux2/src/autoencoder.py | 197 | CODE | |
| LOW | …ons_built_in/diffusion_models/flux2/src/autoencoder.py | 298 | CODE | |
| LOW | extensions_built_in/diffusion_models/anima/anima.py | 203 | CODE | |
| LOW | …in/diffusion_models/qwen_image/qwen_image_edit_plus.py | 231 | CODE | |
| LOW | …in/diffusion_models/qwen_image/qwen_image_pipelines.py | 28 | CODE | |
| LOW | …usion_models/omnigen2/src/pipelines/image_processor.py | 135 | CODE | |
| LOW | …igen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py | 341 | CODE | |
| LOW | …igen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py | 709 | CODE | |
| LOW | …s/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py | 346 | CODE | |
| LOW | …s/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py | 607 | CODE | |
| LOW | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 205 | CODE | |
| LOW | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 493 | CODE | |
| LOW | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 595 | CODE | |
| LOW | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 664 | CODE | |
| LOW | …lt_in/diffusion_models/boogu_image/boogu_image_edit.py | 188 | CODE | |
| LOW | …iffusion_models/boogu_image/src/attention_processor.py | 264 | CODE | |
| LOW | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 955 | CODE | |
| LOW | …ons_built_in/diffusion_models/minimax_h3/minimax_h3.py | 670 | CODE | |
| LOW | …ons_built_in/diffusion_models/minimax_h3/minimax_h3.py | 904 | CODE | |
| LOW | …ilt_in/diffusion_models/minimax_h3/src/text_encoder.py | 117 | CODE | |
| LOW | …s_built_in/diffusion_models/minimax_h3/src/pipeline.py | 65 | CODE | |
| LOW | …nsions_built_in/diffusion_models/minimax_h3/src/vae.py | 696 | CODE | |
| LOW | …built_in/diffusion_models/z_image/z_image_l2p_model.py | 294 | CODE | |
| LOW | …ions_built_in/diffusion_models/wan22/wan22_5b_model.py | 250 | CODE | |
| LOW | …ions_built_in/diffusion_models/wan22/wan22_pipeline.py | 46 | CODE | |
| LOW | …ons_built_in/diffusion_models/wan22/wan22_14b_model.py | 495 | CODE | |
| LOW | …elines/hidream_image/pipeline_hidream_image_editing.py | 569 | CODE | |
| LOW | …elines/hidream_image/pipeline_hidream_image_editing.py | 773 | CODE | |
| LOW | …/src/pipelines/hidream_image/pipeline_hidream_image.py | 516 | CODE | |
| LOW | …in/diffusion_models/hidream/src/hidream_o1/pipeline.py | 32 | CODE | |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 1260 | CODE | |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 2345 | CODE | |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 2368 | CODE | |
| LOW | extensions_built_in/dataset_tools/SuperTagger.py | 71 | CODE | |
| LOW | extensions_built_in/dataset_tools/SyncFromCollection.py | 95 | CODE | |
| LOW | …nsions_built_in/advanced_generator/Img2ImgGenerator.py | 55 | CODE | |
| LOW | …nsions_built_in/advanced_generator/Img2ImgGenerator.py | 105 | CODE | |
| LOW | …ions_built_in/advanced_generator/ReferenceGenerator.py | 55 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 142 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 294 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 525 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 1097 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 1214 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 1382 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 2264 | CODE | |
| 207 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW⚡ | run.py | 123 | except Exception as e: | CODE |
| LOW⚡ | run.py | 128 | except Exception as e2: | CODE |
| LOW⚡ | run.py | 136 | except Exception as e2: | CODE |
| LOW | run_modal.py | 135 | except Exception as e: | CODE |
| MEDIUM | run_modal.py | 136 | print(f"Error running job: {e}") | CODE |
| LOW | extensions_built_in/diffusion_models/krea2/krea2.py | 321 | except Exception as e: | CODE |
| LOW | …sions_built_in/diffusion_models/ideogram4/ideogram4.py | 301 | except Exception as e: | CODE |
| LOW | …ensions_built_in/diffusion_models/omnigen2/__init__.py | 239 | except Exception as e: | CODE |
| LOW | …ons_built_in/diffusion_models/minimax_h3/minimax_h3.py | 348 | except Exception as e: | CODE |
| LOW | …ons_built_in/diffusion_models/minimax_h3/minimax_h3.py | 1528 | except Exception: | CODE |
| LOW | …_in/diffusion_models/minimax_h3/src/ref_video_cache.py | 271 | except Exception: | CODE |
| LOW | …ilt_in/diffusion_models/minimax_h3/src/text_encoder.py | 50 | except Exception: | CODE |
| MEDIUM | …ilt_in/diffusion_models/minimax_h3/src/text_encoder.py | 44 | def video_has_audio(path) -> bool: | CODE |
| LOW | extensions_built_in/diffusion_models/z_image/z_image.py | 103 | except Exception as e: | CODE |
| LOW | …_built_in/diffusion_models/hidream/hidream_o1_model.py | 164 | except Exception as e: | CODE |
| LOW⚡ | …elines/hidream_image/pipeline_hidream_image_editing.py | 1091 | except Exception as e: | STRING |
| LOW⚡ | …elines/hidream_image/pipeline_hidream_image_editing.py | 1101 | except Exception as e: | STRING |
| LOW | extensions_built_in/dataset_tools/SuperTagger.py | 184 | except Exception: | CODE |
| LOW | extensions_built_in/dataset_tools/SyncFromCollection.py | 83 | except Exception as e: | CODE |
| LOW | extensions_built_in/dataset_tools/SyncFromCollection.py | 110 | except Exception as e: | CODE |
| LOW | extensions_built_in/dataset_tools/tools/image_tools.py | 35 | except Exception as e: | CODE |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 139 | except Exception: | CODE |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 963 | except Exception as e: | CODE |
| LOW | extensions_built_in/sd_trainer/DiffusionTrainer.py | 84 | except Exception: | CODE |
| LOW | extensions_built_in/sd_trainer/DiffusionTrainer.py | 310 | except Exception as e: | CODE |
| LOW | extensions_built_in/sd_trainer/DiffusionTrainer.py | 334 | except Exception as db_err: | CODE |
| LOW | extensions_built_in/sd_trainer/UITrainer.py | 76 | except Exception: | CODE |
| LOW | extensions_built_in/sd_trainer/UITrainer.py | 212 | except Exception as e: | CODE |
| LOW | extensions_built_in/captioner/Qwen3OmniCaptioner.py | 710 | except Exception: | CODE |
| LOW | extensions_built_in/captioner/Qwen3OmniCaptioner.py | 858 | except Exception as e: | CODE |
| LOW | extensions_built_in/captioner/Qwen3OmniCaptioner.py | 891 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/Qwen3OmniCaptioner.py | 892 | print(f"Error preprocessing {path}: {e}") | CODE |
| LOW | extensions_built_in/captioner/Qwen3OmniCaptioner.py | 940 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/Qwen3OmniCaptioner.py | 941 | print(f"Error processing {file_path}: {e}") | CODE |
| LOW | extensions_built_in/captioner/Qwen3VLCaptioner.py | 177 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/Qwen3VLCaptioner.py | 178 | print(f"Error processing {file_path}: {e}") | CODE |
| LOW | extensions_built_in/captioner/AceStepCaptioner.py | 267 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/AceStepCaptioner.py | 268 | print(f"Error processing {file_path}: {e}") | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 132 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/BaseCaptioner.py | 133 | print(f"Error captioning file {file_path}: {e}") | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 232 | except Exception as e: | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 303 | except Exception: | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 470 | except Exception as db_err: | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 484 | except Exception as e: | CODE |
| LOW | extensions_built_in/captioner/Ideogram4Captioner.py | 181 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/Ideogram4Captioner.py | 182 | print(f"Error processing {file_path}: {e}") | CODE |
| LOW | testing/test_vae.py | 64 | except Exception as e: | CODE |
| MEDIUM | testing/test_vae.py | 65 | print(f"Error processing {img_path}: {e}") | CODE |
| LOW | testing/test_ltx_dataloader.py | 176 | except Exception: | CODE |
| LOW | testing/test_ltx_dataloader.py | 201 | except Exception as e: | CODE |
| LOW | testing/test_ltx_dataloader.py | 208 | except Exception: | CODE |
| LOW | testing/test_ltx_dataloader.py | 219 | except Exception: | CODE |
| LOW | testing/test_ltx_dataloader.py | 224 | except Exception: | CODE |
| LOW | testing/generate_weight_mappings.py | 299 | except Exception as e: | CODE |
| MEDIUM | testing/generate_weight_mappings.py | 300 | print(f"Error on key {ldm_key}") | CODE |
| LOW | manager/util.py | 220 | except Exception as e: # noqa: BLE001 - surface any network failure clearly | CODE |
| MEDIUM⚡ | scripts/convert_diffusers_to_comfy_transformer_only.py | 56 | print("Error: Cannot use both --do_8_bit and --do_8bit_scaled at the same time.") | CODE |
| MEDIUM⚡ | scripts/convert_diffusers_to_comfy_transformer_only.py | 63 | print(f"Error: Missing transformer folder: {diffusers_path}") | CODE |
| MEDIUM⚡ | scripts/convert_diffusers_to_comfy_transformer_only.py | 70 | print(f"Error: Missing transformer index json: {original_json_path}") | CODE |
| MEDIUM | scripts/convert_diffusers_to_comfy_transformer_only.py | 286 | print(f"Error: Missing transformer safetensors file: {file}") | CODE |
| 122 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | flux_train_ui.py | 358 | "Include sample prompts to test out your trained model. Don't forget to include your trigger word/senten | STRING |
| MEDIUM | config/examples/generate.example.yaml | 22 | # here ate the flags you can use for prompts. Always start with | COMMENT |
| MEDIUM | config/examples/train_lora_flex_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flex_24gb.yaml | 82 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flex_24gb.yaml | 99 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_flux_kontext_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flux_kontext_24gb.yaml | 74 | # and then you can use this model. | COMMENT |
| MEDIUM | config/examples/train_lora_flux_kontext_24gb.yaml | 86 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flux_kontext_24gb.yaml | 104 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_sd35_large_24gb.yaml | 15 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_sd35_large_24gb.yaml | 78 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_sd35_large_24gb.yaml | 95 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | …fig/examples/train_lora_qwen_image_edit_2509_32gb.yaml | 103 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_full_fine_tune_flex.yaml | 15 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_full_fine_tune_flex.yaml | 88 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_full_fine_tune_flex.yaml | 105 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_chroma_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_chroma_24gb.yaml | 85 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_chroma_24gb.yaml | 102 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_hidream_48.yaml | 22 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_hidream_48.yaml | 93 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_hidream_48.yaml | 110 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_wan21_14b_24gb.yaml | 18 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_wan21_14b_24gb.yaml | 91 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_wan21_14b_24gb.yaml | 99 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_omnigen2_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_omnigen2_24gb.yaml | 75 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_omnigen2_24gb.yaml | 92 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_lumina.yaml | 15 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_lumina.yaml | 77 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_lumina.yaml | 94 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_wan22_14b_24gb.yaml | 92 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_wan22_14b_24gb.yaml | 109 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_wan21_1b_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_wan21_1b_24gb.yaml | 80 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_wan21_1b_24gb.yaml | 88 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_flex_redux.yaml | 48 | # for normal redux, we are just recreating the same image, so you can use the same folder path above | COMMENT |
| MEDIUM | config/examples/train_flex_redux.yaml | 54 | # this is what I used for the 24GB card, but feel free to adjust | COMMENT |
| MEDIUM | config/examples/train_flex_redux.yaml | 110 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_flux_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flux_24gb.yaml | 77 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flux_24gb.yaml | 94 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_qwen_image_edit_32gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_qwen_image_edit_32gb.yaml | 100 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_flux_schnell_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flux_schnell_24gb.yaml | 79 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flux_schnell_24gb.yaml | 96 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_full_fine_tune_lumina.yaml | 15 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_full_fine_tune_lumina.yaml | 80 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_full_fine_tune_lumina.yaml | 97 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_flex2_24gb.yaml | 18 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flex2_24gb.yaml | 136 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_flex2_24gb.yaml | 139 | # you can use a single inpaint or single control image on your samples. | COMMENT |
| MEDIUM | config/examples/train_lora_flex2_24gb.yaml | 163 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/train_lora_qwen_image_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_qwen_image_24gb.yaml | 76 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/train_lora_qwen_image_24gb.yaml | 93 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| MEDIUM | config/examples/modal/modal_train_lora_flux_24gb.yaml | 14 | # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/modal/modal_train_lora_flux_24gb.yaml | 77 | # you can add [trigger] to the prompts here and it will be replaced with the trigger word | COMMENT |
| MEDIUM | config/examples/modal/modal_train_lora_flux_24gb.yaml | 94 | # you can add any additional meta info here. [name] is replaced with config name at top | COMMENT |
| 4 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 280 | CODE | |
| LOW | …ns_built_in/audio_models/ace_step/ace_step_15_model.py | 331 | CODE | |
| LOW | extensions_built_in/audio_models/ace_step/src/model.py | 1363 | CODE | |
| LOW | extensions_built_in/audio_models/ace_step/src/model.py | 861 | CODE | |
| LOW | …ensions_built_in/audio_models/ace_step/src/pipeline.py | 75 | CODE | |
| LOW | extensions_built_in/diffusion_models/krea2/src/mmdit.py | 262 | CODE | |
| LOW | extensions_built_in/diffusion_models/krea2/src/mmdit.py | 596 | CODE | |
| LOW | extensions_built_in/diffusion_models/chroma/pipeline.py | 148 | CODE | |
| LOW | …/diffusion_models/prx_pixel_t2i/src/transformer_prx.py | 709 | CODE | |
| LOW | …nsions_built_in/diffusion_models/flux2/src/pipeline.py | 282 | CODE | |
| LOW | …ensions_built_in/diffusion_models/f_light/src/model.py | 309 | CODE | |
| LOW | …ions_built_in/diffusion_models/f_light/src/pipeline.py | 160 | CODE | |
| LOW | …in/diffusion_models/qwen_image/qwen_image_pipelines.py | 28 | CODE | |
| LOW | …_built_in/diffusion_models/mageflow/src/transformer.py | 258 | CODE | |
| LOW | …tensions_built_in/diffusion_models/mageflow/src/vae.py | 435 | CODE | |
| LOW | …tensions_built_in/diffusion_models/mageflow/src/vae.py | 497 | CODE | |
| LOW | …usion_models/omnigen2/src/pipelines/image_processor.py | 49 | CODE | |
| LOW | …igen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py | 341 | CODE | |
| LOW | …igen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py | 494 | CODE | |
| LOW | …igen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py | 638 | CODE | |
| LOW | …s/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py | 346 | CODE | |
| LOW | …s/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py | 462 | CODE | |
| LOW | …nigen2/src/models/transformers/transformer_omnigen2.py | 215 | CODE | |
| LOW | …diffusion_models/omnigen2/src/ops/triton/layer_norm.py | 54 | CODE | |
| LOW | …diffusion_models/omnigen2/src/ops/triton/layer_norm.py | 114 | CODE | |
| LOW | …diffusion_models/omnigen2/src/ops/triton/layer_norm.py | 298 | CODE | |
| LOW | …diffusion_models/omnigen2/src/ops/triton/layer_norm.py | 611 | CODE | |
| LOW | …diffusion_models/omnigen2/src/ops/triton/layer_norm.py | 1012 | CODE | |
| LOW | …diffusion_models/omnigen2/src/ops/triton/layer_norm.py | 1052 | CODE | |
| LOW | …diffusion_models/omnigen2/src/ops/triton/layer_norm.py | 756 | CODE | |
| LOW | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 205 | CODE | |
| LOW | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 509 | CODE | |
| LOW | …_built_in/diffusion_models/boogu_image/src/pipeline.py | 130 | CODE | |
| LOW | …s_built_in/diffusion_models/minimax_h3/src/pipeline.py | 65 | CODE | |
| LOW | …nsions_built_in/diffusion_models/minimax_h3/src/vae.py | 438 | CODE | |
| LOW | …ions_built_in/diffusion_models/wan22/wan22_pipeline.py | 46 | CODE | |
| LOW | …n/diffusion_models/zeta_chroma/zeta_chroma_pipeline.py | 60 | CODE | |
| LOW | …elines/hidream_image/pipeline_hidream_image_editing.py | 341 | CODE | |
| LOW | …elines/hidream_image/pipeline_hidream_image_editing.py | 569 | CODE | |
| LOW | …elines/hidream_image/pipeline_hidream_image_editing.py | 773 | CODE | |
| LOW | …/src/pipelines/hidream_image/pipeline_hidream_image.py | 291 | CODE | |
| LOW | …/src/pipelines/hidream_image/pipeline_hidream_image.py | 516 | CODE | |
| LOW | …lt_in/diffusion_models/hidream/src/models/attention.py | 9 | CODE | |
| LOW | …m/src/models/transformers/transformer_hidream_image.py | 237 | CODE | |
| LOW | …in/diffusion_models/hidream/src/hidream_o1/pipeline.py | 290 | CODE | |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 950 | CODE | |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 1876 | CODE | |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 2158 | CODE | |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 2245 | CODE | |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 2381 | CODE | |
| LOW | …sion_models/hidream/src/schedulers/flash_flow_match.py | 67 | CODE | |
| LOW | …sion_models/hidream/src/schedulers/fm_solvers_unipc.py | 77 | CODE | |
| LOW | …s_built_in/diffusion_models/ernie_image/transformer.py | 152 | CODE | |
| LOW | …s_built_in/diffusion_models/ernie_image/transformer.py | 158 | CODE | |
| LOW | …s_built_in/diffusion_models/ernie_image/transformer.py | 295 | CODE | |
| LOW | extensions_built_in/dataset_tools/tools/fuyu_utils.py | 55 | CODE | |
| LOW | extensions_built_in/captioner/Qwen3OmniCaptioner.py | 131 | CODE | |
| LOW | extensions_built_in/flex2/pipeline.py | 28 | CODE | |
| LOW | manager/spec.py | 135 | CODE | |
| LOW | jobs/process/models/vgg19_critic.py | 72 | CODE | |
| 56 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | spark_requirements.txt | 1 | # NVIDIA RTX Spark — native Windows-on-ARM (win_arm64) requirement set. | COMMENT |
| LOW | ui/src/utils/terminalEmulator.ts | 1 | // Minimal terminal emulator for rendering job logs the way a real terminal | COMMENT |
| LOW | config/examples/generate.example.yaml | 21 | COMMENT | |
| LOW | config/examples/train_lora_flux_kontext_24gb.yaml | 21 | dtype: float16 # precision to save | COMMENT |
| LOW | config/examples/train_lora_hidream_48.yaml | 1 | # HiDream training is still highly experimental. The settings here will take ~35.2GB of vram to train. | COMMENT |
| LOW | config/examples/train_lora_wan21_14b_24gb.yaml | 1 | # IMPORTANT: The Wan2.1 14B model is huge. This config should work on 24GB GPUs. It cannot | COMMENT |
| LOW | config/examples/train_slider.example.yml | 1 | --- | COMMENT |
| LOW | config/examples/train_slider.example.yml | 41 | # for sliders we are adjusting representation of the concept (unet), | COMMENT |
| LOW | config/examples/train_slider.example.yml | 81 | save_every: 50 # save every this many steps | COMMENT |
| LOW | config/examples/train_slider.example.yml | 141 | # slider training config, best for last | COMMENT |
| LOW | config/examples/train_slider.example.yml | 161 | # for example, if we are adjusting the representation of a person, we would use "person" | COMMENT |
| LOW | config/examples/train_slider.example.yml | 181 | # max 76 tokens for now | COMMENT |
| LOW | config/examples/train_slider.example.yml | 201 | # regardless if they are smiling or not, however, the closer the concept is to the target_class | COMMENT |
| LOW | config/examples/train_slider.example.yml | 221 | # It is saved in the model so be aware of that. The software will include this | COMMENT |
| LOW | config/examples/train_lora_flex2_24gb.yaml | 41 | # this will automatically generate the controls for you before training. The current script is not | COMMENT |
| LOW | …ns_built_in/concept_replacer/config/train.example.yaml | 81 | # You can put any information you want here, and it will be saved in the model. | COMMENT |
| LOW | extensions_built_in/diffusion_models/krea2/krea2.py | 201 | # through the Qwen3-VL encoder alongside the prompt (edit-plus style, so | COMMENT |
| LOW | …sions_built_in/diffusion_models/chroma/src/__init__.py | 1 | # This is taken and slightly modified from https://github.com/lodestone-rock/flow/tree/master/src/models/chroma | COMMENT |
| LOW | …/diffusion_models/prx_pixel_t2i/src/transformer_prx.py | 1 | # PRXTransformer2DModel -- vendored, self-contained copy of the Photoroom PRX | COMMENT |
| LOW | …usion_models/omnigen2/src/pipelines/image_processor.py | 1 | # Copyright 2024 The HuggingFace Team. All rights reserved. | COMMENT |
| LOW | …in/diffusion_models/omnigen2/src/utils/import_utils.py | 1 | # Copyright 2024 The HuggingFace Team. All rights reserved. | COMMENT |
| LOW | …_in/diffusion_models/omnigen2/src/models/embeddings.py | 1 | # Copyright 2024 The HuggingFace Team. All rights reserved. | COMMENT |
| LOW | …dels/omnigen2/src/models/transformers/block_lumina2.py | 1 | COMMENT | |
| LOW | …src/schedulers/scheduling_flow_match_euler_discrete.py | 1 | # Copyright 2024 Stability AI, Katherine Crowson and The HuggingFace Team. All rights reserved. | COMMENT |
| LOW | …igen2/src/schedulers/scheduling_dpmsolver_multistep.py | 1 | # Copyright 2024 TSAIL Team and The HuggingFace Team. All rights reserved. | COMMENT |
| LOW | …ns_built_in/diffusion_models/hidream/src/models/moe.py | 81 | # scores_for_seq_aux = scores_for_aux.view(bsz, seq_len, -1) | COMMENT |
| LOW | …ns_built_in/diffusion_models/hidream/src/models/moe.py | 121 | x = x.view(-1, x.shape[-1]) | COMMENT |
| LOW | …models/hidream/src/hidream_o1/qwen3_vl_transformers.py | 1 | # Copyright 2025 The Qwen Team and The HuggingFace Inc. team. All rights reserved. | COMMENT |
| LOW | …sion_models/hidream/src/schedulers/flash_flow_match.py | 1 | # Copyright 2024 Stability AI, Katherine Crowson and The HuggingFace Team. All rights reserved. | COMMENT |
| LOW | …ilt_in/diffusion_models/example_model/example_model.py | 81 | # Class names of modules whose Linear layers get LoRA'd. Matched against | COMMENT |
| LOW | …ns_built_in/diffusion_models/example_model/__init__.py | 1 | # This is a documentation-only TEMPLATE model. Start with README.md in this | COMMENT |
| LOW | …s_built_in/diffusion_models/ernie_image/transformer.py | 1 | # Copyright 2025 Baidu ERNIE-Image Team and The HuggingFace Team. All rights reserved. | COMMENT |
| LOW | extensions_built_in/dataset_tools/tools/fuyu_utils.py | 61 | # # autoregressively generate text | COMMENT |
| LOW | extensions_built_in/dataset_tools/tools/sync_tools.py | 201 | # input_path = img_path | COMMENT |
| LOW | extensions_built_in/dataset_tools/tools/sync_tools.py | 221 | # img = cv2.imread(img_path) | COMMENT |
| LOW | extensions_built_in/dataset_tools/tools/sync_tools.py | 241 | # os.makedirs(os.path.dirname(img_path), exist_ok=True) | COMMENT |
| LOW | extensions_built_in/dataset_tools/tools/sync_tools.py | 261 | # # update eq captions | COMMENT |
| LOW | …_built_in/advanced_generator/config/train.example.yaml | 81 | # You can put any information you want here, and it will be saved in the model. | COMMENT |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 2241 | COMMENT | |
| LOW | …tensions_built_in/sd_trainer/config/train.example.yaml | 81 | # You can put any information you want here, and it will be saved in the model. | COMMENT |
| LOW | extensions_built_in/captioner/Qwen3OmniCaptioner.py | 321 | # gate on token count, not pair count: decode (1 token per sequence) | COMMENT |
| LOW | …ltimate_slider_trainer/UltimateSliderTrainerProcess.py | 361 | # with torch.set_grad_enabled(self.train_config.train_text_encoder): | COMMENT |
| LOW | extensions/example/ExampleMergeModels.py | 41 | # checking requirements. This is called before the run() function | COMMENT |
| LOW | testing/test_bucket_dataloader.py | 61 | dataset_config = DatasetConfig( | COMMENT |
| LOW | testing/test_bucket_dataloader.py | 101 | COMMENT | |
| LOW | scripts/convert_diffusers_to_comfy_transformer_only.py | 1 | ####################################################### | COMMENT |
| LOW | scripts/test_quantizations.py | 41 | import torch # noqa: E402 | COMMENT |
| LOW | scripts/convert_diffusers_to_comfy.py | 1 | ####################################################### | COMMENT |
| LOW | jobs/process/TrainVAEProcess.py | 861 | COMMENT | |
| LOW | jobs/process/BaseSDTrainProcess.py | 921 | meta = load_metadata_from_safetensors(latest_save_path) | COMMENT |
| LOW | jobs/process/BaseSDTrainProcess.py | 1841 | torch.backends.cuda.enable_math_sdp(True) | COMMENT |
| LOW | jobs/process/TrainSliderProcess.py | 461 | current_timestep, | COMMENT |
| LOW | jobs/process/TrainSliderProcess.py | 481 | # prompt_pair.positive_target, # negative prompt | COMMENT |
| LOW | toolkit/config_modules.py | 1061 | # it will select a random start frame and pull the frames at the given fps | COMMENT |
| LOW | toolkit/config_modules.py | 1381 | # --l Specify the CFG scale for the generated image. | COMMENT |
| LOW | toolkit/data_loader.py | 721 | # todo build scheduler that can get buckets from all datasets that match | COMMENT |
| LOW | toolkit/inversion_utils.py | 241 | return noisy_sample, zs, xts | COMMENT |
| LOW | toolkit/inversion_utils.py | 261 | # model.unet.sample_size, | COMMENT |
| LOW | toolkit/inversion_utils.py | 281 | # noisy_sample = xts[idx][None] | COMMENT |
| LOW | toolkit/inversion_utils.py | 301 | # pred_original_sample = (noisy_sample - (1 - alpha_bar[t]) ** 0.5 * noise_pred) / alpha_bar[t] ** 0.5 | COMMENT |
| 27 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| MEDIUM | …ensions_built_in/diffusion_models/chroma/src/layers.py | 229 | # Create a 2D meshgrid of coordinates. | COMMENT |
| MEDIUM | …ensions_built_in/diffusion_models/chroma/src/layers.py | 237 | # Create a 1D tensor of frequency values from 0 to max_freqs-1. | COMMENT |
| MEDIUM | extensions_built_in/sd_trainer/DiffusionTrainer.py | 35 | # Create a thread pool for database operations | COMMENT |
| MEDIUM | extensions_built_in/sd_trainer/DiffusionTrainer.py | 96 | # Create a task and track it | COMMENT |
| MEDIUM | extensions_built_in/sd_trainer/UITrainer.py | 29 | # Create a thread pool for database operations | COMMENT |
| MEDIUM | extensions_built_in/sd_trainer/UITrainer.py | 88 | # Create a task and track it | COMMENT |
| MEDIUM | extensions_built_in/captioner/BaseCaptioner.py | 81 | # Create a thread pool for database operations | COMMENT |
| MEDIUM | extensions_built_in/captioner/BaseCaptioner.py | 315 | # Create a task and track it | COMMENT |
| MEDIUM | testing/shrink_pixart_sm.py | 43 | # Create a new state dict for the reduced model | COMMENT |
| MEDIUM | testing/shrink_pixart_sm3.py | 35 | # Create a new state dict for the reduced model | COMMENT |
| MEDIUM | testing/shrink_pixart_sm2.py | 45 | # Create a new state dict for the reduced model | COMMENT |
| MEDIUM | scripts/convert_diffusers_to_comfy_transformer_only.py | 361 | # Define the float8 range | COMMENT |
| MEDIUM | scripts/convert_diffusers_to_comfy.py | 355 | # Define the float8 range | COMMENT |
| MEDIUM | toolkit/style.py | 34 | # Define the separate loss function | COMMENT |
| MEDIUM | toolkit/dataloader_mixins.py | 91 | # Define the normalization transform | COMMENT |
| MEDIUM | toolkit/dataloader_mixins.py | 1176 | # Create a background with the specified transparent color | COMMENT |
| MEDIUM | toolkit/util/mask.py | 69 | # Create a low-frequency noise field first (for smooth organic shapes) | COMMENT |
| MEDIUM | toolkit/util/shuffle.py | 38 | # Create a slice object to shuffle along the specified axis | COMMENT |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| HIGH | …igen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py | 286 | Get prompt embeddings from the Qwen2 text encoder. Args: prompt: The prompt or list of pro | STRING |
| HIGH | …s/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py | 274 | Get prompt embeddings from the Qwen2 text encoder. Args: prompt: The prompt or list of pro | STRING |
| HIGH | …elines/hidream_image/pipeline_hidream_image_editing.py | 809 | Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[st | STRING |
| HIGH | extensions_built_in/flex2/pipeline.py | 52 | Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[st | STRING |
| HIGH | toolkit/train_pipelines.py | 45 | Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[st | STRING |
| HIGH | toolkit/pipelines.py | 50 | Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[st | STRING |
| HIGH | toolkit/pipelines.py | 590 | Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[st | STRING |
| HIGH | toolkit/pipelines.py | 1198 | Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[st | STRING |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | config/examples/train_slider.example.yml | 18 | # just use one for now | COMMENT |
| MEDIUM | config/examples/train_slider.example.yml | 188 | # this will make the LoRA more robust. You probably want this on unless prompt order | COMMENT |
| MEDIUM | config/examples/train_lora_flex2_24gb.yaml | 121 | # if you are not training with it. Controls are a little more robust and can be left out, | COMMENT |
| LOW | extensions_built_in/diffusion_models/ltx2/ltx2.py | 89 | image, # will contain a dict that can be dumped ditectly into encode_video, just add output_path to it. | CODE |
| MEDIUM | …uilt_in/diffusion_models/prx_pixel_t2i/src/pipeline.py | 53 | # called by the sampler harness (inside a try/except, so optional) | COMMENT |
| LOW | …m/src/models/transformers/transformer_hidream_image.py | 322 | # the implementation on hidream during train was wrong, just use the inference one. | COMMENT |
| MEDIUM | …uilt_in/diffusion_models/example_model/src/pipeline.py | 79 | # called by the sampler harness (inside a try/except, so optional); | COMMENT |
| LOW | jobs/process/BaseExtractProcess.py | 60 | # we will just add process it to the end of the filename if there is more than one process | COMMENT |
| LOW | toolkit/stable_diffusion_model.py | 1711 | # todo, should we just use the Text encoder for the refiner? Fine tuned versions will differ | COMMENT |
| LOW | toolkit/stable_diffusion_model.py | 1931 | # if we only have 1 timestep, we can just use the same timestep for all | COMMENT |
| LOW | toolkit/stable_diffusion_model.py | 2026 | # just use the first second text encoder | COMMENT |
| LOW | toolkit/stable_diffusion_model.py | 2041 | # just use the first second text encoder | COMMENT |
| LOW | toolkit/control_generator.py | 122 | # just pass the PIL image straight through. | COMMENT |
| LOW | toolkit/models/base_model.py | 888 | # if we only have 1 timestep, we can just use the same timestep for all | COMMENT |
| LOW | toolkit/models/te_adapter.py | 399 | # just use aura pile | STRING |
| LOW | toolkit/optimizers/optimizer_utils.py | 197 | # If no dtype specified, just pass through to parent | COMMENT |
| MEDIUM | toolkit/optimizers/automagicEXPERIMENT.py | 394 | # unchanged keeps every property that made v3 robust. A window of H | COMMENT |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | run_modal.py | 80 | # Check if we have DEBUG_TOOLKIT in env | COMMENT |
| LOW | run_modal.py | 82 | # Set torch to trace mode | COMMENT |
| LOW | …ions_built_in/diffusion_models/f_light/src/pipeline.py | 224 | # Check if guidance is needed | COMMENT |
| LOW⚡ | …elines/hidream_image/pipeline_hidream_image_editing.py | 1096 | # Check if we need to switch from editing stage to refining stage | STRING |
| LOW | scripts/update_sponsors.py | 274 | # Check if required environment variables are set | COMMENT |
| LOW | toolkit/timer.py | 38 | # Check if this timer's buffer exceeds max_buffer and remove the oldest if it does | COMMENT |
| LOW⚡ | toolkit/extension.py | 42 | # Check if the value is a list | COMMENT |
| LOW⚡ | toolkit/extension.py | 44 | # Iterate over the list and add the classes to the main list | COMMENT |
| LOW | toolkit/dequantize.py | 65 | # Check if the parameter exists | COMMENT |
| LOW | toolkit/train_tools.py | 226 | if not chunk.eq(chunk[0, 0]).all(): # Check if all elements in the chunk are the same as the first element | CODE |
| LOW | toolkit/data_transfer_object/data_loader.py | 131 | # Check if video opened successfully | COMMENT |
| LOW | toolkit/util/get_model.py | 35 | # Check if the value is a list | COMMENT |
| LOW | toolkit/util/get_model.py | 37 | # Iterate over the list and add the classes to the main list | COMMENT |
| LOW | toolkit/optimizers/automagic.py | 185 | # Check if exp_avg_sq_row and exp_avg_sq_col exist for factored case | COMMENT |
| LOW | toolkit/optimizers/automagic.py | 193 | # Check if exp_avg_sq exists for non-factored case | COMMENT |
| LOW | toolkit/optimizers/automagic.py | 323 | # Check if lr_mask exists in the state before trying to access it | COMMENT |
| LOW | toolkit/optimizers/automagic.py | 337 | # Check if state_dict has the expected structure | COMMENT |
| LOW | toolkit/optimizers/automagic.py | 339 | # Check if at least one state entry has an lr_mask, which is specific to Automagic | COMMENT |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | ui/src/hooks/useJobLossLog.tsx | 43 | // Step 1: get key list (we can do this by calling endpoint once; it returns keys) | COMMENT |
| LOW | ui/src/hooks/useJobLossLog.tsx | 54 | // Step 2: fetch each loss key incrementally (since_step per key if polling) | COMMENT |
| LOW | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 345 | # Step 1: modulation for both streams. | COMMENT |
| LOW | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 362 | # Step 2: joint attention on [instruct + img]. | COMMENT |
| LOW⚡ | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 389 | # Step 3: image self-attention. | COMMENT |
| LOW⚡ | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 397 | # Step 4: residual updates. | COMMENT |
| LOW⚡ | toolkit/basic.py | 45 | # Step 1: Calculate mean and variance of content features | COMMENT |
| LOW⚡ | toolkit/basic.py | 49 | # Step 2: Calculate mean and variance of style features | COMMENT |
| LOW⚡ | toolkit/basic.py | 53 | # Step 3: Normalize content features | COMMENT |
| LOW⚡ | toolkit/basic.py | 57 | # Step 4: Scale and shift normalized content with style's statistics | COMMENT |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | ui/src/hooks/useJobLossLog.tsx | 43 | // Step 1: get key list (we can do this by calling endpoint once; it returns keys) | COMMENT |
| LOW | ui/src/hooks/useJobLossLog.tsx | 54 | // Step 2: fetch each loss key incrementally (since_step per key if polling) | COMMENT |
| LOW | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 345 | # Step 1: modulation for both streams. | COMMENT |
| LOW | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 362 | # Step 2: joint attention on [instruct + img]. | COMMENT |
| LOW⚡ | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 389 | # Step 3: image self-attention. | COMMENT |
| LOW⚡ | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 397 | # Step 4: residual updates. | COMMENT |
| LOW⚡ | toolkit/basic.py | 45 | # Step 1: Calculate mean and variance of content features | COMMENT |
| LOW⚡ | toolkit/basic.py | 49 | # Step 2: Calculate mean and variance of style features | COMMENT |
| LOW⚡ | toolkit/basic.py | 53 | # Step 3: Normalize content features | COMMENT |
| LOW⚡ | toolkit/basic.py | 57 | # Step 4: Scale and shift normalized content with style's statistics | COMMENT |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | extensions_built_in/diffusion_models/krea2/__init__.py | 3 | __all__ = ["Krea2Model"] | CODE |
| LOW⚡ | …built_in/diffusion_models/ideogram4/src/transformer.py | 418 | def set_attention_backend(self, backend: str) -> None: | CODE |
| LOW | …ns_built_in/diffusion_models/prx_pixel_t2i/__init__.py | 3 | __all__ = ["PRXPixelT2IModel"] | CODE |
| LOW | …ions_built_in/diffusion_models/f_light/src/__init__.py | 5 | __all__ = ["FLitePipeline", "FLitePipelineOutput", "APGConfig", "DiT"] | CODE |
| LOW | …ions_built_in/diffusion_models/f_light/src/pipeline.py | 19 | logger = logging.getLogger(__name__) | CODE |
| LOW | …ensions_built_in/diffusion_models/mageflow/src/attn.py | 73 | __all__ = ["flash_attn_varlen_func"] | CODE |
| LOW | …on_models/omnigen2/src/models/transformers/__init__.py | 3 | __all__ = ["OmniGen2Transformer2DModel"] | CODE |
| LOW | …ions_built_in/diffusion_models/boogu_image/__init__.py | 4 | __all__ = ["BooguImageModel", "BooguImageEditModel"] | CODE |
| LOW | …ilt_in/diffusion_models/boogu_image/src/transformer.py | 690 | def set_attention_backend(self, backend: str) -> None: | CODE |
| LOW | …elines/hidream_image/pipeline_hidream_image_editing.py | 34 | logger = logging.getLogger(__name__) | CODE |
| LOW | …ns_built_in/diffusion_models/example_model/__init__.py | 12 | __all__ = ["ExampleModel"] | CODE |
| LOW | toolkit/network_mixins.py | 182 | def _set_runtime_scale(self: Module, value) -> None: | CODE |
| LOW | toolkit/kohya_lora.py | 81 | def _set_runtime_scale(self, value) -> None: | CODE |
| LOW | toolkit/optimizers/automagic2.py | 103 | def _update_param(self, p: torch.Tensor, group: dict) -> None: | CODE |
| LOW | toolkit/optimizers/automagic3.py | 413 | def _update_param(self, p: torch.Tensor, group: dict) -> None: | CODE |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| HIGH | ui_scripts/upsample_ideogram4_caption.py | 295 | '({"index": i, "caption": {...}|null}) instead of a single final list.' | CODE |
| HIGH | manager/env.py | 581 | 'cmd /s /c "call "%s" >nul && set"' % vcvars, | CODE |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| CRITICAL | toolkit/ip_adapter.py | 1152 | yield from self.image_encoder.vision_model.embeddings.position_embedding.parameters(recurse) | CODE |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| HIGH | config/examples/train_slider.example.yml | 166 | # you can leave it blank to affect everything. In this example, we are adjusting | COMMENT |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| MEDIUM | toolkit/models/lokr.py | 351 | # org_forward here does raw CUDA stream/event orchestration and | COMMENT |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| MEDIUM | toolkit/unloader.py | 22 | CODE |