LongLive 2.0: Infra - Long Video Gen
418 matches across 12 categories. Click a row to expand file-level details.
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | inference_sp.py | 99 | def _expected_inference_samples(config): |
| LOW | inference_sp.py | 166 | def materialize_quantized_generator(model, device, materialize_fn, stage_desc, is_main_process): |
| LOW | inference_sp.py | 176 | def configure_generator_torch_compile(pipeline, config, is_main_process): |
| LOW | inference.py | 140 | def _expected_inference_samples(config): |
| LOW | inference.py | 215 | def materialize_quantized_generator(model, device, materialize_fn, stage_desc): |
| LOW | inference.py | 229 | def configure_generator_torch_compile(pipeline, config): |
| LOW | pipeline/causal_diffusion_inference.py | 954 | def _set_all_modules_sink_size(self, sink_size_value: int): |
| LOW | pipeline/causal_diffusion_inference.py | 968 | def _set_all_modules_global_sink_size(self, value: int): |
| LOW | pipeline/causal_diffusion_inference.py | 861 | def _initialize_crossattn_cache(self, batch_size, dtype, device): |
| LOW | pipeline/causal_diffusion_inference.py | 895 | def _initialize_sample_scheduler(self, noise): |
| LOW | pipeline/causal_diffusion_inference.py | 918 | def _set_all_modules_max_attention_size(self, local_attn_size_value: int): |
| LOW | pipeline/causal_diffusion_inference.py | 1001 | def _update_sink_for_scene_cut(self, kv_cache, current_num_frames): |
| LOW | pipeline/self_forcing_training.py | 372 | def _set_all_modules_sink_size(self, sink_size_value: int): |
| LOW | pipeline/self_forcing_training.py | 384 | def _set_all_modules_global_sink_size(self, value: int): |
| LOW | pipeline/self_forcing_training.py | 82 | def _build_default_denoising_step_list(self, sampling_steps): |
| LOW | pipeline/self_forcing_training.py | 111 | def generate_chunk_with_cache( |
| LOW | pipeline/self_forcing_training.py | 288 | def inference_with_trajectory( |
| LOW | pipeline/self_forcing_training.py | 409 | def _inference_with_trajectory_inner( |
| LOW | pipeline/self_forcing_training.py | 698 | def _initialize_crossattn_cache(self, batch_size, dtype, device): |
| LOW | pipeline/self_forcing_training.py | 744 | def _set_all_modules_max_attention_size(self, local_attn_size_value: int): |
| LOW | fouroversix/setup.py | 32 | def get_cuda_bare_metal_version() -> Version | None: |
| LOW | fouroversix/tests/test_correctness.py | 52 | def test_backend_outputs_are_consistent( # noqa: C901, PLR0915 |
| LOW | fouroversix/scripts/resources.py | 52 | def has_untracked_or_unstaged_changes(self) -> bool: |
| LOW | fouroversix/scripts/resources.py | 150 | def install_fouroversix_non_editable() -> None: |
| LOW | fouroversix/scripts/ptq/evaluators/smoothquant.py | 196 | def get_calibration_experiments( |
| LOW | fouroversix/src/fouroversix/quantize/utils.py | 14 | def get_no_random_sign_vector(device: int) -> torch.Tensor: |
| LOW | fouroversix/src/fouroversix/model/quantize.py | 35 | def should_replace_existing_modules_in_model( |
| LOW | fouroversix/src/fouroversix/model/modules/linear.py | 271 | def _serialize_quantized_weight( |
| LOW | fouroversix/src/fouroversix/model/modules/linear.py | 322 | def quantized_weight_transposed(self) -> QuantizedTensor | None: |
| LOW | tests/test_i2v_teacher_forcing_context.py | 96 | def _load_causal_diffusion_with_stubs(): |
| LOW | utils/memory.py | 61 | def fake_diffusers_current_device(model: torch.nn.Module, target_device: torch.device): |
| LOW | utils/memory.py | 96 | def move_model_to_device_with_memory_preservation(model, target_device, preserved_memory_gb=0): |
| LOW | utils/memory.py | 112 | def offload_model_from_device_for_memory_preservation(model, target_device, preserved_memory_gb=0): |
| LOW | utils/torch_compile_utils.py | 86 | def configure_module_call_torch_compile( |
| LOW | utils/dataset.py | 200 | def _load_captions_from_folder(self, folder: Path): |
| LOW | utils/dataset.py | 422 | def _get_video_files_in_folder(self, folder_path): |
| LOW | utils/dataset.py | 451 | def _check_folder_has_enough_frames(self, folder_path): |
| LOW | utils/dataset.py | 533 | def _can_sample_from_position(self, total_frames, original_fps, num_frames, start_frame): |
| LOW | utils/dataset.py | 552 | def _can_complete_all_segments_without_wrap(self, video_files, start_video_idx, start_frame): |
| LOW | utils/dataset.py | 629 | def _sample_frames_from_video(self, video_path, num_frames, start_frame=0): |
| LOW | utils/dataset.py | 696 | def _sample_frames_from_video_ffmpeg(self, video_path, num_frames, start_frame=0): |
| LOW | utils/dataset.py | 750 | def _try_get_item_from_folder(self, folder_idx, deterministic: bool = False): |
| LOW | utils/lightvae_5b_wrapper.py | 18 | def _extract_checkpoint_state_dict(raw): |
| LOW | utils/lightvae_5b_wrapper.py | 31 | def _map_lightvae_key_to_wanvae(key): |
| LOW | utils/lightvae_5b_wrapper.py | 106 | def _normalize_vae_state_dict(raw_state): |
| LOW | utils/lightvae_5b_wrapper.py | 117 | def infer_lightvae_pruning_rate_from_ckpt(vae_path, full_decoder_conv1_out=1024): |
| LOW | utils/position_embedding_utils.py | 15 | def select_temporal_offset_for_sample( |
| LOW | utils/rope_triton.py | 84 | def _split_complex_to_cos_sin(freqs_complex: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor]: |
| LOW | utils/nvfp4_checkpoint.py | 38 | def unwrap_generator_state_dict(checkpoint: object, use_ema: bool = False) -> object: |
| LOW | utils/nvfp4_checkpoint.py | 50 | def clean_fsdp_state_dict_keys(state_dict: Mapping[str, torch.Tensor]) -> dict[str, torch.Tensor]: |
| LOW | utils/nvfp4_checkpoint.py | 55 | def build_model_quantization_config(config, keep_master_weights: bool = False): |
| LOW | utils/nvfp4_checkpoint.py | 134 | def drop_fouroversix_master_weights(model: nn.Module) -> list[str]: |
| LOW | utils/wan_5b_wrapper.py | 316 | def enable_gradient_checkpointing(self) -> None: |
| LOW | utils/quant.py | 90 | def _resolve_modules_to_not_convert( |
| LOW | utils/quant.py | 119 | def _get_default_filtered_modules(filter_profile: str | None) -> list[str]: |
| LOW | utils/quant.py | 137 | def _warn_for_te_config_mismatch(model_quant_config: ModelQuantizationConfig) -> None: |
| LOW | utils/quant.py | 273 | def _copy_tensor_into_parameter( |
| LOW | utils/quant.py | 350 | def quantize_model_with_optional_te( |
| LOW | utils/quant.py | 425 | def _materialize_transformer_engine_weights_for_inference( |
| LOW | utils/quant.py | 450 | def _materialize_mixed_quantized_weights_for_inference( |
| 52 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | pipeline/__init__.py | 1 | |
| LOW | pipeline/__init__.py | 2 | |
| LOW | pipeline/self_forcing_training.py | 12 | |
| LOW | fouroversix/scripts/generate_kernels.py | 1 | |
| LOW | fouroversix/scripts/resources.py | 1 | |
| LOW | fouroversix/scripts/test_on_modal.py | 1 | |
| LOW | fouroversix/scripts/speedtest/matmul.py | 1 | |
| LOW | fouroversix/scripts/speedtest/quantize.py | 1 | |
| LOW | fouroversix/scripts/ptq/utils.py | 1 | |
| LOW | fouroversix/scripts/ptq/coordinators/__init__.py | 1 | |
| LOW | fouroversix/scripts/ptq/coordinators/__init__.py | 2 | |
| LOW | fouroversix/scripts/ptq/evaluators/rtn.py | 1 | |
| LOW | fouroversix/scripts/ptq/evaluators/spinquant.py | 1 | |
| LOW | fouroversix/scripts/ptq/evaluators/__init__.py | 1 | |
| LOW | fouroversix/scripts/ptq/evaluators/awq.py | 1 | |
| LOW | fouroversix/scripts/ptq/evaluators/smoothquant.py | 1 | |
| LOW | fouroversix/scripts/ptq/evaluators/evaluator.py | 1 | |
| LOW | fouroversix/scripts/train/__main__.py | 1 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 3 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 4 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 4 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 4 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 4 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 4 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 11 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 11 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 11 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 12 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 12 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 12 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 12 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 12 | |
| LOW | fouroversix/src/fouroversix/__init__.py | 13 | |
| LOW | fouroversix/src/fouroversix/quantize/__init__.py | 1 | |
| LOW | fouroversix/src/fouroversix/quantize/__init__.py | 2 | |
| LOW | fouroversix/src/fouroversix/quantize/__init__.py | 3 | |
| LOW | fouroversix/src/fouroversix/quantize/__init__.py | 4 | |
| LOW | …oversix/src/fouroversix/quantize/transformer_engine.py | 33 | |
| LOW | fouroversix/src/fouroversix/quantize/cuda/backend.py | 29 | |
| LOW | fouroversix/src/fouroversix/quantize/cuda/__init__.py | 1 | |
| LOW | fouroversix/src/fouroversix/quantize/triton/kernel.py | 1 | |
| LOW | fouroversix/src/fouroversix/quantize/triton/__init__.py | 1 | |
| LOW | …uroversix/src/fouroversix/quantize/pytorch/__init__.py | 1 | |
| LOW | …roversix/src/fouroversix/quantize/pytorch/reference.py | 1 | |
| LOW | fouroversix/src/fouroversix/model/quantize.py | 1 | |
| LOW | fouroversix/src/fouroversix/model/__init__.py | 1 | |
| LOW | fouroversix/src/fouroversix/model/__init__.py | 1 | |
| LOW | fouroversix/src/fouroversix/model/__init__.py | 2 | |
| LOW | fouroversix/src/fouroversix/model/__init__.py | 3 | |
| LOW | fouroversix/src/fouroversix/model/__init__.py | 3 | |
| LOW | fouroversix/src/fouroversix/model/modules/__init__.py | 1 | |
| LOW | fouroversix/src/fouroversix/model/modules/__init__.py | 2 | |
| LOW | fouroversix/src/fouroversix/matmul/__init__.py | 1 | |
| LOW | fouroversix/src/fouroversix/matmul/cutlass/backend.py | 27 | |
| LOW | fouroversix/src/fouroversix/matmul/cutlass/__init__.py | 1 | |
| LOW | utils/dataset.py | 9 | |
| LOW | utils/dataset.py | 10 | |
| LOW | utils/dataset.py | 15 | |
| LOW | utils/rope_triton.py | 17 | |
| LOW | utils/nvfp4_checkpoint.py | 10 | |
| 45 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | fouroversix/scripts/ptq/coordinators/local.py | 0 | run any tasks that should be used to calibrate models for a given ptq method and set of parameters before running evalua |
| HIGH | fouroversix/scripts/ptq/coordinators/modal.py | 0 | run any tasks that should be used to calibrate models for a given ptq method and set of parameters before running evalua |
| HIGH | fouroversix/scripts/ptq/coordinators/base.py | 0 | run any tasks that should be used to calibrate models for a given ptq method and set of parameters before running evalua |
| HIGH | fouroversix/src/fouroversix/model/modules/linear.py | 0 | return high precision parameters to be quantized and deleted. |
| HIGH | fouroversix/src/fouroversix/model/modules/qwen.py | 0 | return high precision parameters to be quantized and deleted. |
| HIGH | fouroversix/src/fouroversix/model/modules/gpt_oss.py | 0 | return high precision parameters to be quantized and deleted. |
| HIGH | fouroversix/src/fouroversix/model/modules/gpt_oss.py | 0 | return high precision parameters to be quantized and deleted. |
| HIGH | fouroversix/src/fouroversix/model/modules/linear_ori.py | 0 | return high precision parameters to be quantized and deleted. |
| HIGH | fouroversix/src/fouroversix/model/modules/linear.py | 0 | get the size of a single element, in bytes, for a parameter. |
| HIGH | fouroversix/src/fouroversix/model/modules/qwen.py | 0 | get the size of a single element, in bytes, for a parameter. |
| HIGH | fouroversix/src/fouroversix/model/modules/gpt_oss.py | 0 | get the size of a single element, in bytes, for a parameter. |
| HIGH | fouroversix/src/fouroversix/model/modules/linear_ori.py | 0 | get the size of a single element, in bytes, for a parameter. |
| HIGH | fouroversix/src/fouroversix/model/modules/linear.py | 0 | prepare this layer for post-training quantization by quantizing the weight, storing the quantized weight, and deleting t |
| HIGH | fouroversix/src/fouroversix/model/modules/qwen.py | 0 | prepare this layer for post-training quantization by quantizing the weight, storing the quantized weight, and deleting t |
| HIGH | fouroversix/src/fouroversix/model/modules/gpt_oss.py | 0 | prepare this layer for post-training quantization by quantizing the weight, storing the quantized weight, and deleting t |
| HIGH | fouroversix/src/fouroversix/model/modules/linear_ori.py | 0 | prepare this layer for post-training quantization by quantizing the weight, storing the quantized weight, and deleting t |
| HIGH | wan_5b/textimage2video.py | 0 | configures a model object. this includes setting evaluation modes, applying distributed parallel strategy, and handling |
| HIGH | wan_5b/text2video.py | 0 | configures a model object. this includes setting evaluation modes, applying distributed parallel strategy, and handling |
| HIGH | wan_5b/image2video.py | 0 | configures a model object. this includes setting evaluation modes, applying distributed parallel strategy, and handling |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | pipeline/causal_diffusion_inference.py | 39 | # Step 1: Initialize all models |
| LOW | pipeline/causal_diffusion_inference.py | 67 | # Step 2: Initialize scheduler |
| LOW | pipeline/causal_diffusion_inference.py | 232 | # Step 1: Initialize KV cache to all zeros |
| LOW | pipeline/causal_diffusion_inference.py | 266 | # Step 2: Cache context feature |
| LOW | pipeline/causal_diffusion_inference.py | 437 | # Step 3: Temporal denoising loop |
| LOW | pipeline/causal_diffusion_inference.py | 741 | # Step 4: Decode the output |
| LOW | pipeline/self_forcing_training.py | 446 | # Step 1: Initialize KV cache to all zeros |
| LOW | pipeline/self_forcing_training.py | 468 | # Step 2: Cache context feature |
| LOW | pipeline/self_forcing_training.py | 486 | # Step 3: Temporal denoising loop |
| LOW | utils/dataset.py | 545 | # We need to check if we can get num_frames frames |
| LOW | model/diffusion.py | 30 | # Step 2: Initialize all hyperparameters |
| LOW | model/diffusion.py | 171 | # Step 2: Randomly sample a timestep and add noise to denoiser inputs |
| LOW | model/diffusion.py | 258 | # Step 3: Noise augmentation, also add small noise to clean context latents |
| LOW | model/diffusion.py | 331 | # Step 5: Store prediction errors into error buffer. |
| LOW | model/dmd.py | 47 | # Step 2: Initialize all dmd hyperparameters |
| LOW | model/dmd.py | 98 | # Step 1: Compute the fake score |
| LOW | model/dmd.py | 119 | # Step 2: Compute the real score |
| LOW | model/dmd.py | 138 | # Step 3: Compute the DMD gradient (DMD paper eq. 7). |
| LOW | model/dmd.py | 213 | # Step 1: Randomly sample timestep based on the given schedule and corresponding noise |
| LOW | model/dmd.py | 245 | # Step 2: Compute the KL grad |
| LOW | model/dmd.py | 289 | # Step 1: Unroll generator to obtain fake videos |
| LOW | model/dmd.py | 305 | # Step 2: Compute the DMD loss |
| LOW | model/dmd.py | 366 | # Step 1: Run generator on backward simulated noisy input |
| LOW | model/dmd.py | 408 | # Step 2: Compute the fake prediction |
| LOW | model/dmd.py | 446 | # Step 3: Compute the denoising loss for the fake critic |
| LOW | model/dmd.py | 483 | # Step 5: Debugging Log |
| LOW | trainer/diffusion.py | 60 | # Step 1: Initialize the distributed training environment (rank, seed, dtype, logging etc.) |
| LOW | trainer/diffusion.py | 152 | # Step 2: Initialize the model and optimizer |
| LOW | trainer/diffusion.py | 333 | # Step 3: Initialize the dataloader |
| LOW | trainer/diffusion.py | 736 | # Step 1: Get the next batch of text prompts |
| LOW | trainer/diffusion.py | 759 | # Step 2: Extract the conditional infos |
| LOW | trainer/diffusion.py | 792 | # Step 3: Train the generator |
| LOW | trainer/diffusion.py | 821 | # Step 4: Update EMA (if enabled and after start step) |
| LOW | trainer/diffusion.py | 856 | # Step 5: Logging |
| LOW | trainer/distillation.py | 41 | # Step 1: Initialize the distributed training environment (rank, seed, dtype, logging etc.) |
| LOW | trainer/distillation.py | 79 | # Step 2: Initialize the model |
| LOW | trainer/distillation.py | 406 | # Step 3: Set up EMA parameter containers |
| LOW | trainer/distillation.py | 430 | # Step 4: Initialize the optimizer |
| LOW | trainer/distillation.py | 447 | # Step 5: Initialize the dataloader |
| LOW | trainer/distillation.py | 531 | # Step 6: Initialize the validation dataloader for visualization (fixed prompts) |
| LOW | trainer/distillation.py | 1000 | # Step 1: Get the next batch of text prompts |
| LOW | trainer/distillation.py | 1040 | # Step 2: Extract the conditional infos |
| LOW | trainer/distillation.py | 1071 | # Step 3: Store gradients for the generator (if training the generator) |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | inference_sp.py | 49 | except Exception as exc: |
| LOW | inference.py | 84 | except Exception as e: |
| MEDIUM | pipeline/causal_diffusion_inference.py | 468 | def _vae_thread_fn(): |
| LOW | pipeline/causal_diffusion_inference.py | 356 | except Exception: |
| LOW | pipeline/causal_diffusion_inference.py | 361 | except Exception: |
| LOW | pipeline/causal_diffusion_inference.py | 366 | except Exception: |
| LOW | pipeline/causal_diffusion_inference.py | 936 | except Exception: |
| LOW | pipeline/causal_diffusion_inference.py | 946 | except Exception: |
| LOW | pipeline/causal_diffusion_inference.py | 951 | except Exception: |
| LOW | pipeline/causal_diffusion_inference.py | 965 | except Exception: |
| LOW | pipeline/causal_diffusion_inference.py | 974 | except Exception: |
| LOW | pipeline/causal_diffusion_inference.py | 64 | except Exception as exc: |
| LOW | pipeline/causal_diffusion_inference.py | 495 | except Exception as exc: |
| LOW | pipeline/self_forcing_training.py | 348 | except Exception: |
| LOW | pipeline/self_forcing_training.py | 353 | except Exception: |
| LOW | pipeline/self_forcing_training.py | 358 | except Exception: |
| LOW | pipeline/self_forcing_training.py | 381 | except Exception: |
| LOW | pipeline/self_forcing_training.py | 390 | except Exception: |
| LOW | pipeline/self_forcing_training.py | 764 | except Exception: |
| LOW | pipeline/self_forcing_training.py | 773 | except Exception: |
| LOW | utils/torch_compile_utils.py | 49 | except Exception as exc: |
| LOW | utils/torch_compile_utils.py | 75 | except Exception as exc: |
| LOW | utils/torch_compile_utils.py | 112 | except Exception as exc: |
| MEDIUM | utils/dataset.py | 433 | def get_numeric_key(path): |
| LOW | utils/dataset.py | 211 | except Exception: |
| LOW | utils/dataset.py | 230 | except Exception: |
| LOW | utils/dataset.py | 418 | except Exception: |
| LOW | utils/dataset.py | 469 | except Exception: |
| LOW | utils/dataset.py | 953 | except Exception as exc: |
| LOW | utils/lightvae_5b_wrapper.py | 123 | except Exception as exc: |
| LOW | utils/lightvae_5b_wrapper.py | 138 | except Exception: |
| LOW | utils/quant.py | 282 | except Exception: |
| LOW | utils/quant.py | 294 | except Exception as copy_exc: |
| LOW | utils/quant.py | 719 | except Exception as exc: # pragma: no cover - exercised only when extension is stale/missing |
| LOW | wan_5b/utils/prompt_extend.py | 185 | except Exception as e: |
| LOW | wan_5b/utils/prompt_extend.py | 248 | except Exception as e: |
| LOW | wan_5b/utils/utils.py | 53 | except Exception as e: |
| LOW | wan_5b/utils/utils.py | 75 | except Exception as e: |
| LOW | wan_5b/modules/causal_model_sp_ulysses.py | 46 | except Exception: |
| LOW | wan_5b/modules/attention.py | 11 | except Exception: |
| LOW | wan_5b/modules/attention.py | 36 | except Exception: |
| LOW | scripts/decode_vae_latents.py | 148 | except Exception as e: |
| LOW | scripts/decode_lightvae_latents.py | 324 | except Exception as exc: |
| LOW | scripts/decode_lightvae_latents.py | 334 | except Exception as exc: |
| LOW | model/dmd.py | 332 | except Exception: |
| LOW | model/dmd.py | 480 | except Exception: |
| LOW | trainer/diffusion.py | 50 | except Exception as e: |
| LOW | trainer/diffusion.py | 629 | except Exception as e: |
| LOW | trainer/diffusion.py | 646 | except Exception as e: |
| LOW | trainer/diffusion.py | 885 | except Exception: |
| LOW | trainer/distillation.py | 1265 | except Exception as e: |
| LOW | trainer/distillation.py | 908 | except Exception as e: |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | utils/dataset.py | 164 | # ------------------------------------------------------------------ |
| MEDIUM | utils/dataset.py | 166 | # ------------------------------------------------------------------ |
| MEDIUM | utils/dataset.py | 175 | # ------------------------------------------------------------------ |
| MEDIUM | utils/dataset.py | 177 | # ------------------------------------------------------------------ |
| MEDIUM | utils/dataset.py | 215 | # ------------------------------------------------------------------ |
| MEDIUM | utils/dataset.py | 217 | # ------------------------------------------------------------------ |
| MEDIUM | wan_5b/modules/causal_model.py | 235 | # ------------------------------- |
| MEDIUM | wan_5b/modules/causal_model.py | 238 | # ------------------------------- |
| MEDIUM | wan_5b/modules/causal_model.py | 253 | # ------------------------------- |
| MEDIUM | wan_5b/modules/causal_model.py | 258 | # ------------------------------- |
| MEDIUM | wan_5b/modules/causal_model.py | 1206 | # ========================================== |
| MEDIUM | wan_5b/modules/causal_model.py | 1208 | # ========================================== |
| MEDIUM | wan_5b/modules/causal_model.py | 1220 | # ========================================== |
| MEDIUM | wan_5b/modules/causal_model.py | 1222 | # ========================================== |
| MEDIUM | wan_5b/modules/causal_model.py | 1239 | # ========================================== |
| MEDIUM | wan_5b/modules/causal_model.py | 1241 | # ========================================== |
| MEDIUM | trainer/distillation.py | 601 | # ---------------------------------------------------------------------------------------------------------- |
| MEDIUM | trainer/distillation.py | 603 | # ---------------------------------------------------------------------------------------------------------- |
| MEDIUM | trainer/distillation.py | 1332 | # -------------------------------------------------------------------------------------------------------------- |
| MEDIUM | trainer/distillation.py | 1334 | # -------------------------------------------------------------------------------------------------------------- |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | inference_sp.py | 33 | |
| LOW | inference.py | 61 | |
| LOW | pipeline/causal_diffusion_inference.py | 167 | |
| LOW | pipeline/causal_diffusion_inference.py | 369 | |
| LOW | pipeline/causal_diffusion_inference.py | 468 | |
| LOW | pipeline/self_forcing_training.py | 111 | |
| LOW | pipeline/self_forcing_training.py | 409 | |
| LOW | fouroversix/setup.py | 52 | |
| LOW | fouroversix/tests/test_correctness.py | 52 | |
| LOW | fouroversix/scripts/ptq/evaluators/evaluator.py | 59 | |
| LOW | utils/dataset.py | 750 | |
| LOW | utils/position_embedding_utils.py | 56 | |
| LOW | utils/error_buffer.py | 239 | |
| LOW | utils/nvfp4_checkpoint.py | 134 | |
| LOW | utils/quant.py | 477 | |
| LOW | utils/lora_utils.py | 19 | |
| LOW | wan_5b/utils/fm_solvers.py | 412 | |
| LOW | wan_5b/utils/fm_solvers.py | 483 | |
| LOW | wan_5b/utils/utils.py | 106 | |
| LOW | wan_5b/utils/qwen_vl_utils.py | 85 | |
| LOW | wan_5b/utils/qwen_vl_utils.py | 328 | |
| LOW | wan_5b/modules/attention.py | 84 | |
| LOW | wan_5b/modules/vae2_1.py | 68 | |
| LOW | wan_5b/modules/vae2_1.py | 101 | |
| LOW | wan_5b/modules/t5.py | 27 | |
| LOW | wan_5b/modules/causal_model.py | 90 | |
| LOW | wan_5b/modules/causal_model.py | 316 | |
| LOW | wan_5b/modules/causal_model.py | 1360 | |
| LOW | wan_5b/modules/vae2_2.py | 73 | |
| LOW | wan_5b/modules/vae2_2.py | 111 | |
| LOW | scripts/decode_vae_latents.py | 72 | |
| LOW | scripts/decode_lightvae_latents.py | 104 | |
| LOW | model/diffusion.py | 15 | |
| LOW | trainer/diffusion.py | 25 | |
| LOW | trainer/diffusion.py | 56 | |
| LOW | trainer/diffusion.py | 552 | |
| LOW | trainer/diffusion.py | 578 | |
| LOW | trainer/diffusion.py | 637 | |
| LOW | trainer/diffusion.py | 911 | |
| LOW | trainer/diffusion.py | 1016 | |
| LOW | trainer/diffusion.py | 1080 | |
| LOW | trainer/distillation.py | 37 | |
| LOW | trainer/distillation.py | 831 | |
| LOW | trainer/distillation.py | 857 | |
| LOW | trainer/distillation.py | 994 | |
| LOW | trainer/distillation.py | 1152 | |
| LOW | trainer/distillation.py | 1271 |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | fouroversix/scripts/hadamard_code_gen.py | 1 | from pathlib import Path |
| LOW | …uroversix/src/fouroversix/quantize/quantized_tensor.py | 161 | f"Expected {expected_scale_factors} scale factors, got " |
| LOW | …uroversix/src/fouroversix/csrc/include/kernel_traits.h | 81 | // static constexpr int kSwizzleS = Is_nvfp4 ? 2 : 1; // 4 or 2 elements |
| LOW | …c/fouroversix/csrc/include/fp4_quant_launch_template.h | 21 | #define KERNEL_PARAM_MODIFIER __grid_constant__ |
| LOW | …uroversix/src/fouroversix/csrc/include/static_switch.h | 1 | // Inspired by |
| LOW | utils/__init__.py | 1 | # Marker file: turn `utils/` from a namespace package into a regular package. |
| LOW | utils/rope_triton.py | 21 | import triton.language as tl |
| LOW | utils/kernel/__init__.py | 1 | # Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES |
| LOW | wan_5b/modules/attention.py | 161 | # we're trying to replace). |
| LOW | wan_5b/modules/causal_model.py | 41 | # concatenated freqs tensor each time. LRU keeps memory bounded. |
| LOW | wan_5b/modules/causal_model.py | 61 | # when disabled, stores (freqs_i_complex, None, None). |
| LOW | wan_5b/modules/causal_model.py | 481 | global_sink_tokens = getattr(self, "global_sink_size", 0) * frame_seqlen |
| LOW | wan_5b/modules/causal_model.py | 661 | |
| LOW | model/diffusion.py | 41 | self.device = device |
| Severity | File | Line | Snippet |
|---|---|---|---|
| CRITICAL | fouroversix/scripts/ptq/evaluators/spinquant.py | 230 | model.lm_head.weight.data = model.model.embed_tokens.weight.data.clone() |
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | wan_5b/utils/utils.py | 80 | Convert a string to a boolean. Supported true values: 'yes', 'true', 't', 'y', '1' Supported false values: |
| HIGH | wan_5b/utils/qwen_vl_utils.py | 138 | calculate the number of frames for video used for model inputs. Args: ele (dict): a dict contains the confi |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | trainer/distillation.py | 1276 | # Define the specific modules we want to apply LoRA to |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | utils/dataset.py | 473 | # Check if we have enough frames for all segments |
| LOW | trainer/distillation.py | 1156 | # Check if we should train generator on this optimization step |