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,309 GitHub stars. SynthScan v2.0 examined 170,358 lines of code across 574 source files, recording 1923 pattern matches distributed across 20 syntactic categories. The overall adjusted score of 16.0 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).
Longitudinal tracking requires multiple scan runs. Once this repository is re-scanned after new commits land, this chart will visualise how the synthetic code signal evolves over time — enabling you to detect whether AI authorship is growing, stabilising, or being actively corrected by human engineers.
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 1923 distinct pattern matches across 20 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/ltx2.py | 10 | 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 | |
| LOW | …built_in/diffusion_models/flux_kontext/flux_kontext.py | 5 | CODE | |
| 594 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| MEDIUM | run_mac.zsh | 7 | # ── Banner ───────────────────────────────────────────────────────── | COMMENT |
| MEDIUM | run_mac.zsh | 43 | # ── 1. Download standalone Python if needed ───────────────────────── | COMMENT |
| MEDIUM | run_mac.zsh | 70 | # ── 2. Create venv if it doesn't exist ────────────────────────────── | COMMENT |
| MEDIUM | run_mac.zsh | 77 | # ── 3. Download / update portable Node.js ────────────────────────── | COMMENT |
| MEDIUM | run_mac.zsh | 119 | # ── 4. Install / update PyTorch packages ──────────────────────────── | COMMENT |
| MEDIUM | run_mac.zsh | 141 | # ── 5. Install / update requirements.txt ──────────────────────────── | COMMENT |
| MEDIUM | run_mac.zsh | 160 | # ── 6. Build 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 |
| 94 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 | 464 | def enable_gradient_checkpointing(self): | CODE |
| LOW | extensions_built_in/diffusion_models/krea2/src/mmdit.py | 467 | 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 | 878 | def get_model_state_dict_from_combined_ckpt( | CODE |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 1027 | def convert_lora_original_to_diffusers( | CODE |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 1061 | def convert_lora_diffusers_to_original( | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/ltx2/ltx2.py | 1141 | def get_transformer_block_names(self) -> Optional[List[str]]: | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/ltx2/ltx2.py | 1144 | def convert_lora_weights_before_save(self, state_dict): | CODE |
| LOW⚡ | extensions_built_in/diffusion_models/ltx2/ltx2.py | 1152 | 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 |
| 263 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| 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 | …_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 | 27 | 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 | 878 | CODE | |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 933 | CODE | |
| LOW | …_in/diffusion_models/ltx2/convert_ltx2_to_diffusers.py | 979 | CODE | |
| LOW | extensions_built_in/diffusion_models/ltx2/ltx2.py | 847 | 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 | …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 | …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 | 116 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 241 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 484 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 1003 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 1120 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 1288 | CODE | |
| LOW | extensions_built_in/sd_trainer/SDTrainer.py | 2115 | CODE | |
| LOW | extensions_built_in/sd_trainer/DiffusionTrainer.py | 59 | CODE | |
| LOW | extensions_built_in/sd_trainer/DiffusionTrainer.py | 118 | CODE | |
| LOW | extensions_built_in/sd_trainer/UITrainer.py | 51 | CODE | |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 157 | CODE | |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 233 | CODE | |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 292 | CODE | |
| 182 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW⚡ | run.py | 116 | except Exception as e: | CODE |
| LOW⚡ | run.py | 121 | except Exception as e2: | CODE |
| LOW⚡ | run.py | 129 | except Exception as e2: | CODE |
| LOW | run_modal.py | 136 | except Exception as e: | CODE |
| MEDIUM | run_modal.py | 137 | 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 | 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 | 871 | 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 | 276 | except Exception as e: | CODE |
| LOW | extensions_built_in/sd_trainer/DiffusionTrainer.py | 290 | 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/Qwen3VLCaptioner.py | 121 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/Qwen3VLCaptioner.py | 122 | print(f"Error processing {file_path}: {e}") | CODE |
| LOW | extensions_built_in/captioner/AceStepCaptioner.py | 259 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/AceStepCaptioner.py | 260 | print(f"Error processing {file_path}: {e}") | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 122 | except Exception as e: | CODE |
| MEDIUM | extensions_built_in/captioner/BaseCaptioner.py | 123 | print(f"Error captioning file {file_path}: {e}") | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 215 | except Exception as e: | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 258 | except Exception: | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 421 | except Exception as db_err: | CODE |
| LOW | extensions_built_in/captioner/BaseCaptioner.py | 435 | 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 |
| 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 |
| LOW | scripts/add_mask_dataset.py | 84 | except Exception: | CODE |
| LOW | scripts/add_mask_dataset.py | 102 | except Exception: | CODE |
| LOW | scripts/add_mask_dataset.py | 128 | except Exception: | CODE |
| MEDIUM | scripts/add_mask_dataset.py | 196 | print(f"Error: {img_dir} is not a directory") | CODE |
| MEDIUM | scripts/add_mask_dataset.py | 211 | print(f"Error: no images found in {img_dir}") | CODE |
| MEDIUM | scripts/update_sponsors.py | 91 | print(f"Error fetching Patreon data: {e}") | CODE |
| MEDIUM | scripts/update_sponsors.py | 109 | print("Error: Neither GITHUB_USERNAME nor GITHUB_ORG is set") | CODE |
| MEDIUM | scripts/update_sponsors.py | 211 | print(f"Error fetching GitHub sponsors data: {e}") | STRING |
| MEDIUM | scripts/update_sponsors.py | 287 | print(f"Error: Missing required environment variables: {', '.join(missing_vars)}") | CODE |
| LOW | scripts/repair_dataset_folder.py | 39 | except Exception as e: | CODE |
| MEDIUM | scripts/repair_dataset_folder.py | 40 | print(f"Error opening {img_path}: {e}") | CODE |
| LOW | scripts/repair_dataset_folder.py | 51 | except Exception as e: | CODE |
| MEDIUM | scripts/repair_dataset_folder.py | 52 | print(f"Error rotating {img_path}: {e}") | CODE |
| 94 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 | 81 | # 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 | 98 | # 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 | 85 | # 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 | 103 | # 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 | 77 | # 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 | 94 | # 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 | 102 | # 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 | 87 | # 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 | 104 | # 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 | 84 | # 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 | 101 | # 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 | 92 | # 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 | 109 | # 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 | 90 | # 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 | 98 | # 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 | 74 | # 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 | 91 | # 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 | 76 | # 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 | 93 | # 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 | 91 | # 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 | 108 | # 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 | 79 | # 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 | 87 | # 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 | 109 | # 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 | 76 | # 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 | 93 | # 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 | 99 | # 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 | 78 | # 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 | 95 | # 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 | 79 | # 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 | 96 | # 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 | 135 | # 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 | 138 | # you can use a single inpaint or single control image on your samples. | COMMENT |
| MEDIUM | config/examples/train_lora_flex2_24gb.yaml | 162 | # 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 | 75 | # 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 | 92 | # 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 | 76 | # 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 | 93 | # 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 | 248 | CODE | |
| LOW | extensions_built_in/diffusion_models/krea2/src/mmdit.py | 582 | 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 | …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 | …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/flex2/pipeline.py | 28 | CODE | |
| LOW | jobs/process/models/vgg19_critic.py | 72 | CODE | |
| LOW | jobs/process/models/critic.py | 106 | CODE | |
| LOW | toolkit/config_modules.py | 1071 | CODE | |
| LOW | toolkit/stable_diffusion_model.py | 1865 | CODE | |
| LOW | toolkit/stable_diffusion_model.py | 2323 | CODE | |
| LOW | toolkit/lorm.py | 54 | CODE | |
| LOW | toolkit/photomaker_pipeline.py | 219 | CODE | |
| LOW | toolkit/lora_special.py | 51 | CODE | |
| 48 more matches not shown… | ||||
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| 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 | # The below is an example, but you can put your grocery list in it if you want. | 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 | extensions_built_in/diffusion_models/krea2/src/mmdit.py | 521 | 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 | # The below is an example, but you can put your grocery list in it if you want. | COMMENT |
| LOW | …tensions_built_in/sd_trainer/config/train.example.yaml | 81 | # The below is an example, but you can put your grocery list in it if you want. | 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/convert_diffusers_to_comfy.py | 1 | ####################################################### | COMMENT |
| LOW | jobs/process/TrainVAEProcess.py | 861 | 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 | 1281 | # --h Specify the height of the generated image. | 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 |
| LOW | toolkit/dataloader_mixins.py | 1281 | # image must be square. If it is not, we will resize/squish it so it is, that way we don't crop out data | COMMENT |
| LOW | toolkit/ip_adapter.py | 201 | COMMENT | |
| LOW | toolkit/custom_adapter.py | 281 | else: | COMMENT |
| LOW | toolkit/lycoris_special.py | 121 | "Upsample2D", | COMMENT |
| LOW | toolkit/guidance.py | 121 | # [baseline_conditional_noisy_latents, baseline_unconditional_noisy_latents], | COMMENT |
| LOW | toolkit/pipelines.py | 441 | # denoising_end=denoising_end, | COMMENT |
| LOW | toolkit/util/convrot_quant.py | 1141 | COMMENT | |
| LOW | toolkit/util/convrot_quant.py | 1201 | # near-gaussian so amax (~4 sigma) leaves the optimal-MSE step (~1.2 / 0.6 | COMMENT |
| 14 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 | 71 | # Create a thread pool for database operations | COMMENT |
| MEDIUM | extensions_built_in/captioner/BaseCaptioner.py | 270 | # 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 | 88 | # Define the normalization transform | COMMENT |
| MEDIUM | toolkit/dataloader_mixins.py | 1033 | # 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 | 327 | Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[st | STRING |
| HIGH | toolkit/pipelines.py | 867 | Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[st | STRING |
| HIGH | toolkit/pipelines.py | 1475 | 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 | 87 | 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 | 1694 | # todo, should we just use the Text encoder for the refiner? Fine tuned versions will differ | COMMENT |
| LOW | toolkit/stable_diffusion_model.py | 1914 | # if we only have 1 timestep, we can just use the same timestep for all | COMMENT |
| LOW | toolkit/stable_diffusion_model.py | 2009 | # just use the first second text encoder | COMMENT |
| LOW | toolkit/stable_diffusion_model.py | 2024 | # 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 | 842 | # 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 |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| LOW | run_modal.py | 81 | # Check if we have DEBUG_TOOLKIT in env | COMMENT |
| LOW | run_modal.py | 83 | # 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 | 105 | # 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 | 42 | // Step 1: get key list (we can do this by calling endpoint once; it returns keys) | COMMENT |
| LOW | ui/src/hooks/useJobLossLog.tsx | 53 | // 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 | 42 | // Step 1: get key list (we can do this by calling endpoint once; it returns keys) | COMMENT |
| LOW | ui/src/hooks/useJobLossLog.tsx | 53 | // 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 | …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/optimizers/automagic2.py | 103 | def _update_param(self, p: torch.Tensor, group: dict) -> None: | CODE |
| LOW | toolkit/optimizers/automagic3.py | 396 | def _update_param(self, p: torch.Tensor, group: dict) -> None: | 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 |
|---|---|---|---|---|
| HIGH | ui_scripts/upsample_ideogram4_caption.py | 295 | '({"index": i, "caption": {...}|null}) instead of a single final list.' | CODE |
| Severity | File | Line | Snippet | Context |
|---|---|---|---|---|
| MEDIUM | toolkit/unloader.py | 24 | CODE |