SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer
1225 matches across 16 categories. Click a row to expand file-level details.
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | environment_setup.sh | 2 | # ----------------------------------------------------------------------------- |
| MEDIUM | environment_setup.sh | 13 | # ----------------------------------------------------------------------------- |
| MEDIUM | diffusion/post_training/rewards.py | 15 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/post_training/rewards.py | 17 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/post_training/rewards.py | 81 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/post_training/rewards.py | 83 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/post_training/rewards.py | 107 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/post_training/rewards.py | 109 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/post_training/rewards.py | 185 | # ========================================================================= |
| MEDIUM | diffusion/post_training/rewards.py | 187 | # ========================================================================= |
| MEDIUM | diffusion/post_training/rewards.py | 319 | # ========================================================================= |
| MEDIUM | diffusion/post_training/rewards.py | 321 | # ========================================================================= |
| MEDIUM | …post_training/diffusers_patch/pipeline_with_logprob.py | 32 | # --------------------------------------------------------------------------- |
| MEDIUM | …post_training/diffusers_patch/pipeline_with_logprob.py | 34 | # --------------------------------------------------------------------------- |
| MEDIUM | …post_training/diffusers_patch/pipeline_with_logprob.py | 197 | # --------------------------------------------------------------------------- |
| MEDIUM | …post_training/diffusers_patch/pipeline_with_logprob.py | 199 | # --------------------------------------------------------------------------- |
| MEDIUM | …post_training/diffusers_patch/pipeline_with_logprob.py | 346 | # --------------------------------------------------------------------------- |
| MEDIUM | …post_training/diffusers_patch/pipeline_with_logprob.py | 348 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/utils/action_overlay.py | 43 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/utils/action_overlay.py | 45 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/utils/action_overlay.py | 121 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/utils/action_overlay.py | 123 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/edm_sample.py | 28 | # ---------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/edm_sample.py | 83 | # ---------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/qwen/qwen_vl.py | 77 | # =================================================================== |
| MEDIUM | diffusion/model/qwen/qwen_vl.py | 79 | # =================================================================== |
| MEDIUM | diffusion/model/qwen/qwen_vl.py | 144 | # =================================================================== |
| MEDIUM | diffusion/model/qwen/qwen_vl.py | 146 | # =================================================================== |
| MEDIUM | diffusion/model/qwen/qwen_vl.py | 208 | # =================================================================== |
| MEDIUM | diffusion/model/qwen/qwen_vl.py | 210 | # =================================================================== |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 256 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 258 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 268 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 270 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 240 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 242 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 296 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 298 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 315 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 317 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 48 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 50 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 68 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 70 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 268 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 270 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 464 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 466 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 551 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_camctrl_blocks.py | 553 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 57 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 64 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 171 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 173 | # --------------------------------------------------------------------------- |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 270 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 272 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 279 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 281 | # ========================================================================= |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 428 | # ------------------------------------------------------------------ |
| MEDIUM | diffusion/model/nets/sana_gdn_camctrl_blocks.py | 430 | # ------------------------------------------------------------------ |
| 140 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tools/convert_sana_wm_refiner_to_diffusers.py | 35 | |
| LOW | tools/metrics/geneval/generation/diffusers_generate.py | 15 | |
| LOW | tools/metrics/pytorch-fid/src/pytorch_fid/inception.py | 7 | |
| LOW | tools/metrics/pytorch-fid/src/pytorch_fid/inception.py | 9 | |
| LOW | tools/controlnet/inference_controlnet.py | 37 | |
| LOW | tools/controlnet/inference_controlnet.py | 37 | |
| LOW | tools/convert_scripts/convert_sana_to_svdquant.py | 2 | |
| LOW | tools/convert_scripts/convert_sana_to_diffusers.py | 2 | |
| LOW | …ols/convert_scripts/convert_sana_video_to_diffusers.py | 2 | |
| LOW | train_video_scripts/train_video_ivjoint_chunk.py | 36 | |
| LOW | train_video_scripts/train_video_ivjoint.py | 31 | |
| LOW | train_video_scripts/train_video_ivjoint.py | 35 | |
| LOW | train_video_scripts/train_video_ivjoint.py | 38 | |
| LOW | train_video_scripts/train_video_ivjoint.py | 47 | |
| LOW | app/app_sana_multithread.py | 17 | |
| LOW | app/sana_pipeline.py | 28 | |
| LOW | app/sana_pipeline.py | 28 | |
| LOW | app/sana_pipeline.py | 28 | |
| LOW | app/sana_pipeline.py | 28 | |
| LOW | app/sana_controlnet_pipeline.py | 31 | |
| LOW | app/sana_controlnet_pipeline.py | 31 | |
| LOW | app/sana_controlnet_pipeline.py | 31 | |
| LOW | app/sana_controlnet_pipeline.py | 31 | |
| LOW | app/sana_pipeline_inpaint.py | 20 | |
| LOW | app/sana_pipeline_inpaint.py | 31 | |
| LOW | app/sana_pipeline_inpaint.py | 31 | |
| LOW | app/sana_pipeline_inpaint.py | 31 | |
| LOW | app/sana_pipeline_inpaint.py | 31 | |
| LOW | app/sana_pipeline_inpaint.py | 41 | |
| LOW | app/safety_check.py | 19 | |
| LOW | app/safety_check.py | 21 | |
| LOW | app/safety_check.py | 21 | |
| LOW | app/app_sana_4bit.py | 21 | |
| LOW | app/app_sana.py | 17 | |
| LOW | app/app_sana.py | 32 | |
| LOW | app/sana_sprint_pipeline.py | 32 | |
| LOW | app/sana_sprint_pipeline.py | 32 | |
| LOW | app/app_sana_inpaint.py | 8 | |
| LOW | app/app_sana_sprint.py | 17 | |
| LOW | sana/tools/__init__.py | 1 | |
| LOW | sana/tools/__init__.py | 2 | |
| LOW | sana/tools/__init__.py | 2 | |
| LOW | diffusion/__init__.py | 3 | |
| LOW | diffusion/__init__.py | 4 | |
| LOW | diffusion/__init__.py | 4 | |
| LOW | diffusion/__init__.py | 5 | |
| LOW | diffusion/__init__.py | 6 | |
| LOW | diffusion/__init__.py | 7 | |
| LOW | diffusion/__init__.py | 8 | |
| LOW | diffusion/__init__.py | 9 | |
| LOW | diffusion/post_training/rewards.py | 5 | |
| LOW | diffusion/post_training/dataset/pickscore/prpocess.py | 3 | |
| LOW | diffusion/refiner/diffusers_ltx2_refiner.py | 31 | |
| LOW | diffusion/guiders/__init__.py | 1 | |
| LOW | diffusion/scheduler/flow_euler_sampler.py | 26 | |
| LOW | diffusion/utils/misc.py | 30 | |
| LOW | diffusion/utils/action_overlay.py | 33 | |
| LOW | diffusion/utils/config_wan.py | 4 | |
| LOW | diffusion/utils/optimizer.py | 25 | |
| LOW | diffusion/model/edm_sample.py | 26 | |
| 233 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | app/app_sana_multithread.py | 0 | <div style='display: flex; align-items: center; justify-content: center; text-align: center;'> <img src="https://raw.git |
| HIGH | app/app_sana.py | 0 | <div style='display: flex; align-items: center; justify-content: center; text-align: center;'> <img src="https://raw.git |
| HIGH | app/app_sana_inpaint.py | 0 | <div style='display: flex; align-items: center; justify-content: center; text-align: center;'> <img src="https://raw.git |
| HIGH | scripts/interface.py | 0 | <div style='display: flex; align-items: center; justify-content: center; text-align: center;'> <img src="https://raw.git |
| HIGH | diffusion/scheduler/scm_scheduler.py | 0 | output class for the scheduler's `step` function output. args: prev_sample (`torch.floattensor` of shape `(batch_size, n |
| HIGH | diffusion/scheduler/lcm_scheduler.py | 0 | output class for the scheduler's `step` function output. args: prev_sample (`torch.floattensor` of shape `(batch_size, n |
| HIGH | diffusion/scheduler/trigflow_scheduler.py | 0 | output class for the scheduler's `step` function output. args: prev_sample (`torch.floattensor` of shape `(batch_size, n |
| HIGH | diffusion/scheduler/scm_scheduler.py | 0 | sets the discrete timesteps used for the diffusion chain (to be run before inference). args: num_inference_steps (`int`) |
| HIGH | diffusion/scheduler/lcm_scheduler.py | 0 | sets the discrete timesteps used for the diffusion chain (to be run before inference). args: num_inference_steps (`int`) |
| HIGH | diffusion/scheduler/trigflow_scheduler.py | 0 | sets the discrete timesteps used for the diffusion chain (to be run before inference). args: num_inference_steps (`int`) |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute log(alpha_t) of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute log(alpha_t) of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/sa_solver.py | 0 | compute log(alpha_t) of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute alpha_t of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute alpha_t of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/sa_solver.py | 0 | compute alpha_t of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute sigma_t of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute sigma_t of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/sa_solver.py | 0 | compute sigma_t of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute lambda_t = log(alpha_t) - log(sigma_t) of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute lambda_t = log(alpha_t) - log(sigma_t) of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/sa_solver.py | 0 | compute lambda_t = log(alpha_t) - log(sigma_t) of a given continuous-time label t in [0, t]. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute the continuous-time label t in [0, t] of a given half-logsnr lambda_t. |
| HIGH | diffusion/model/dpm_solver.py | 0 | compute the continuous-time label t in [0, t] of a given half-logsnr lambda_t. |
| HIGH | diffusion/model/sa_solver.py | 0 | compute the continuous-time label t in [0, t] of a given half-logsnr lambda_t. |
| HIGH | diffusion/model/nets/sana_multi_scale_controlnet.py | 0 | forward pass of sana. x: (n, c, h, w) tensor of spatial inputs (images or latent representations of images) t: (n,) tens |
| HIGH | diffusion/model/nets/sana_multi_scale_adaln.py | 0 | forward pass of sana. x: (n, c, h, w) tensor of spatial inputs (images or latent representations of images) t: (n,) tens |
| HIGH | diffusion/model/nets/sana_multi_scale.py | 0 | forward pass of sana. x: (n, c, h, w) tensor of spatial inputs (images or latent representations of images) t: (n,) tens |
| HIGH | diffusion/model/nets/sana.py | 0 | forward pass of sana. x: (n, c, h, w) tensor of spatial inputs (images or latent representations of images) t: (n,) tens |
| HIGH | diffusion/model/nets/sana_multi_scale_controlnet.py | 0 | this method allows the object to be called like a function. it simply calls the forward method. |
| HIGH | diffusion/model/nets/sana_multi_scale_video.py | 0 | this method allows the object to be called like a function. it simply calls the forward method. |
| HIGH | diffusion/model/nets/sana_multi_scale.py | 0 | this method allows the object to be called like a function. it simply calls the forward method. |
| HIGH | diffusion/model/nets/sana.py | 0 | this method allows the object to be called like a function. it simply calls the forward method. |
| HIGH | diffusion/model/nets/sana_multi_scale_video.py | 0 | a sana block with global shared adaptive layer norm zero (adaln-zero) conditioning. |
| HIGH | diffusion/model/nets/sana_multi_scale_video_camctrl.py | 0 | a sana block with global shared adaptive layer norm zero (adaln-zero) conditioning. |
| HIGH | diffusion/model/nets/sana_multi_scale.py | 0 | a sana block with global shared adaptive layer norm zero (adaln-zero) conditioning. |
| HIGH | diffusion/model/nets/sana_multi_scale_video.py | 0 | forward pass of sana. x: (n, c, t, h, w) tensor of spatial inputs (images or latent representations of images) t: (n,) t |
| HIGH | diffusion/model/nets/sana_multi_scale_video.py | 0 | forward pass of sana. x: (n, c, t, h, w) tensor of spatial inputs (images or latent representations of images) t: (n,) t |
| HIGH | diffusion/model/nets/sana_multi_scale_video_camctrl.py | 0 | forward pass of sana. x: (n, c, t, h, w) tensor of spatial inputs (images or latent representations of images) t: (n,) t |
| HIGH | diffusion/model/nets/sana_blocks.py | 0 | embeds class labels into vector representations. also handles label dropout for classifier-free guidance. |
| HIGH | diffusion/model/nets/sana_blocks.py | 0 | embeds class labels into vector representations. also handles label dropout for classifier-free guidance. |
| HIGH | diffusion/model/nets/sana_blocks.py | 0 | embeds class labels into vector representations. also handles label dropout for classifier-free guidance. |
| HIGH | …inear/modules/triton_lite_mla_kernels/pad_vk_mm_fwd.py | 0 | kernel for computing the matmul c = a x b. a has shape (m, k), b has shape (k, n) and c has shape (m, n) |
| HIGH | …ear/modules/triton_lite_mla_kernels/linear_relu_fwd.py | 0 | kernel for computing the matmul c = a x b. a has shape (m, k), b has shape (k, n) and c has shape (m, n) |
| HIGH | …ear/modules/triton_lite_mla_kernels/proj_divide_bwd.py | 0 | kernel for computing the matmul c = a x b. a has shape (m, k), b has shape (k, n) and c has shape (m, n) |
| HIGH | …/nets/fastlinear/modules/triton_lite_mla_kernels/mm.py | 0 | kernel for computing the matmul c = a x b. a has shape (m, k), b has shape (k, n) and c has shape (m, n) |
| HIGH | diffusion/model/wan/model.py | 0 | args: x(tensor): shape [b, l, c] seq_lens(tensor): shape [b] grid_sizes(tensor): shape [b, 3], the second dimension cont |
| HIGH | diffusion/model/wan/model.py | 0 | args: x(tensor): shape [b, l, c] seq_lens(tensor): shape [b] grid_sizes(tensor): shape [b, 3], the second dimension cont |
| HIGH | diffusion/model/wan/model.py | 0 | args: x(tensor): shape [b, l, c] seq_lens(tensor): shape [b] grid_sizes(tensor): shape [b, 3], the second dimension cont |
| HIGH | scripts/inference_geneval.py | 0 | \ @article{ghosh2024geneval, title={geneval: an object-focused framework for evaluating text-to-image alignment}, author |
| HIGH | scripts/inference_geneval_diffusers.py | 0 | \ @article{ghosh2024geneval, title={geneval: an object-focused framework for evaluating text-to-image alignment}, author |
| HIGH | scripts/inference_sana_sprint_geneval.py | 0 | \ @article{ghosh2024geneval, title={geneval: an object-focused framework for evaluating text-to-image alignment}, author |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tools/convert_sana_wm_refiner_to_diffusers.py | 123 | def convert_connectors_to_diffusers(checkpoint: dict[str, object]) -> dict[str, object]: |
| LOW | tools/metrics/geneval/prompts/create_prompts.py | 38 | def generate_single_object_sample(rng: np.random.Generator, size: int = None): |
| LOW | tools/metrics/geneval/prompts/create_prompts.py | 62 | def generate_two_object_sample(rng: np.random.Generator): |
| LOW | tools/metrics/geneval/prompts/create_prompts.py | 128 | def generate_color_attribution_sample(rng: np.random.Generator): |
| LOW | tools/metrics/clip-score/clip_score.py | 231 | def calculate_clip_score_official(dataloader): |
| LOW | tools/metrics/pytorch-fid/compute_fid.py | 90 | def calculate_frechet_distance(mu1, sigma1, mu2, sigma2, eps=1e-6): |
| LOW | tools/metrics/pytorch-fid/compute_fid.py | 123 | def calculate_activation_statistics(files, model, batch_size=50, dims=2048, device="cpu", num_workers=1): |
| LOW | tools/metrics/pytorch-fid/compute_fid.py | 130 | def compute_statistics_of_path(path, model, batch_size, dims, device, num_workers=1, flag="ref"): |
| LOW | tools/metrics/pytorch-fid/compute_fid.py | 168 | def calculate_fid_given_paths(paths, batch_size, device, dims, num_workers=1): |
| LOW | tools/metrics/pytorch-fid/tests/test_fid_score.py | 13 | def test_calculate_fid_given_statistics(mocker, tmp_path, device): |
| LOW | tools/metrics/pytorch-fid/tests/test_fid_score.py | 41 | def test_compute_statistics_of_path(mocker, tmp_path, device): |
| LOW | tools/metrics/pytorch-fid/tests/test_fid_score.py | 62 | def test_compute_statistics_of_path_from_file(mocker, tmp_path, device): |
| LOW | tools/metrics/pytorch-fid/src/pytorch_fid/fid_score.py | 153 | def calculate_frechet_distance(mu1, sigma1, mu2, sigma2, eps=1e-6): |
| LOW | tools/metrics/pytorch-fid/src/pytorch_fid/fid_score.py | 206 | def calculate_activation_statistics(files, model, batch_size=50, dims=2048, device="cpu", num_workers=1): |
| LOW | tools/metrics/pytorch-fid/src/pytorch_fid/fid_score.py | 230 | def compute_statistics_of_path(path, model, batch_size, dims, device, num_workers=1): |
| LOW | tools/metrics/pytorch-fid/src/pytorch_fid/fid_score.py | 242 | def calculate_fid_given_paths(paths, batch_size, device, dims, num_workers=1): |
| LOW | app/sana_pipeline.py | 52 | def classify_height_width_bin(height: int, width: int, ratios: dict) -> Tuple[int, int]: |
| LOW | app/sana_controlnet_pipeline.py | 54 | def classify_height_width_bin(height: int, width: int, ratios: dict) -> Tuple[int, int]: |
| LOW | app/sana_pipeline_inpaint.py | 56 | def classify_height_width_bin(height: int, width: int, ratios: dict) -> Tuple[int, int]: |
| LOW | app/sana_sprint_pipeline.py | 40 | def classify_height_width_bin(height: int, width: int, ratios: dict) -> Tuple[int, int]: |
| LOW | diffusion/post_training/rewards.py | 97 | def _get_clip_image_transform(processor): |
| LOW | diffusion/post_training/rewards.py | 114 | def _patch_imagereward_compat(): |
| LOW | diffusion/post_training/rewards.py | 124 | def _apply_chunking_to_forward(forward_fn, chunk_size, chunk_dim, *input_tensors): |
| LOW | diffusion/post_training/rewards.py | 136 | def _find_pruneable_heads_and_indices(heads, n_heads, head_size, already_pruned_heads): |
| LOW | …post_training/diffusers_patch/pipeline_with_logprob.py | 203 | def pipeline_with_logprob_flux( |
| LOW | …post_training/diffusers_patch/pipeline_with_logprob.py | 352 | def pipeline_with_logprob_sana( |
| LOW | diffusion/post_training/diffusers_patch/solver.py | 299 | def dpm_solver_first_order_update( |
| LOW | diffusion/post_training/diffusers_patch/solver.py | 318 | def multistep_dpm_solver_second_order_update( |
| LOW | diffusion/refiner/diffusers_ltx2_refiner.py | 63 | def _load_diffusers_components(self) -> tuple[nn.Module, nn.Module]: |
| LOW | diffusion/refiner/diffusers_ltx2_refiner.py | 331 | def _streaming_self_attention( |
| LOW | diffusion/scheduler/flow_euler_sampler.py | 91 | def add_noise_to_image_conditioning_latents( |
| LOW | diffusion/scheduler/lcm_scheduler.py | 88 | def rescale_zero_terminal_snr(betas): |
| LOW | diffusion/scheduler/lcm_scheduler.py | 310 | def get_scalings_for_boundary_condition_discrete(self, t): |
| LOW | diffusion/scheduler/longlive_flow_euler_sampler.py | 214 | def enable_gradient_checkpointing(self): |
| LOW | diffusion/scheduler/longlive_flow_euler_sampler.py | 361 | def _initialize_cached_modules(self): |
| LOW | diffusion/scheduler/longlive_flow_euler_sampler.py | 401 | def _create_autoregressive_segments(self, total_frames: int, base_chunk_frames: int) -> List[int]: |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 414 | def get_coefficients_exponential_negative(self, order, interval_start, interval_end): |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 438 | def get_coefficients_exponential_positive(self, order, interval_start, interval_end, tau): |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 482 | def lagrange_polynomial_coefficient(self, order, lambda_list): |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 606 | def stochastic_adams_bashforth_update( |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 698 | def stochastic_adams_moulton_update( |
| LOW | diffusion/utils/lr_scheduler.py | 62 | def get_cosine_decay_to_constant_with_warmup( |
| LOW | diffusion/utils/camctrl_config.py | 145 | def model_video_camctrl_init_config(config: SanaVideoCamCtrlConfig, latent_size: int = 32): |
| LOW | diffusion/utils/checkpoint.py | 286 | def load_ckpt_with_auto_reshape(model, state_dict, strict=False): |
| LOW | diffusion/utils/checkpoint.py | 367 | def load_ckpt_with_auto_reshape(model, state_dict, strict=False): |
| LOW | diffusion/utils/chunk_utils.py | 71 | def chunk_index_from_chunk_size( |
| LOW | diffusion/utils/chunk_utils.py | 134 | def get_chunk_index_from_config(config: Any, num_frames: Optional[int] = None) -> Optional[List[int]]: |
| LOW | diffusion/utils/logger.py | 130 | def rename_file_with_creation_time(file_path): |
| LOW | diffusion/utils/optimizer.py | 584 | def _dequantize_state_first_step(self, quantized_chunks): |
| LOW | diffusion/utils/cam_utils.py | 89 | def random_rotation_matrix_quaternion(dtype=torch.float32, device="cpu"): |
| LOW | diffusion/utils/cam_utils.py | 214 | def _normalize_poses_identity_unit_distance( |
| LOW | diffusion/utils/import_utils.py | 37 | def is_triton_module_available(): |
| LOW | diffusion/model/model_growth_utils.py | 109 | def _get_num_layers_from_state_dict(self, state_dict): |
| LOW | diffusion/model/model_growth_utils.py | 122 | def _copy_non_transformer_params(self): |
| LOW | diffusion/model/respace.py | 82 | def truncated_normal_icdf_sample(n, mu, sigma, a, b, device, dtype): |
| LOW | diffusion/model/respace.py | 113 | def compute_density_for_timestep_sampling( |
| LOW | diffusion/model/respace.py | 200 | def _masked_multinomial_from_logweights(self, logw_col: th.Tensor, starts: th.Tensor, ends: th.Tensor) -> th.Tensor: |
| LOW | diffusion/model/respace.py | 226 | def sample_step_sequence_batch( |
| LOW | diffusion/model/respace.py | 465 | def training_losses_diffusers(self, model, *args, **kwargs): # pylint: disable=signature-differs |
| LOW | diffusion/model/dpm_solver.py | 891 | def get_orders_and_timesteps_for_singlestep_solver(self, steps, order, skip_type, t_T, t_0, device): |
| 118 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | environment_setup.sh | 1 | #!/usr/bin/env bash |
| LOW | tools/create_wids_metadata.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | tools/download.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | tools/convert_sana_wm_refiner_to_diffusers.py | 1 | #!/usr/bin/env python3 |
| LOW | tools/controlnet/inference_controlnet.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | train_video_scripts/train_video_ivjoint_chunk.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | train_video_scripts/train_video_ivjoint.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | app/app_sana_multithread.py | 1 | #!/usr/bin/env python |
| LOW | app/sana_pipeline.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | app/sana_controlnet_pipeline.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | app/sana_pipeline_inpaint.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | app/safety_check.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | app/app_sana_4bit.py | 1 | #!/usr/bin/env python |
| LOW | app/app_sana.py | 1 | #!/usr/bin/env python |
| LOW | app/sana_sprint_pipeline.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | app/app_sana_sprint.py | 1 | #!/usr/bin/env python |
| LOW | asset/examples.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | asset/docs/quantize/8bit_sana.md | 1 | <!-- Copyright 2024 The HuggingFace Team. All rights reserved. |
| LOW | asset/docs/quantize/4bit_sana.md | 1 | <!--Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | sana/tools/hf_utils.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | sana/cli/run.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | sana/cli/upload2hf.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES (authored by @Lyken17) |
| LOW | diffusion/post_training/diffusers_patch/text_encode.py | 1 | #!/usr/bin/env python |
| LOW | diffusion/refiner/diffusers_ltx2_refiner.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/guiders/adaptive_projected_guidance.py | 1 | # Copyright 2025 The HuggingFace Team. All rights reserved. |
| LOW | diffusion/scheduler/dpm_solver.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/scheduler/scm_scheduler.py | 1 | # Copyright 2023 Stanford University Team and The HuggingFace Team. All rights reserved. |
| LOW | diffusion/scheduler/flow_euler_sampler.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/scheduler/iddpm.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/scheduler/sa_sampler.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/scheduler/lcm_scheduler.py | 1 | # Copyright 2023 Stanford University Team and The HuggingFace Team. All rights reserved. |
| LOW | diffusion/scheduler/trigflow_scheduler.py | 1 | # Copyright 2023 Stanford University Team and The HuggingFace Team. All rights reserved. |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 1 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| LOW | diffusion/utils/lr_scheduler.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/camctrl_config.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/misc.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/config.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/checkpoint.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/chunk_utils.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/logger.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/action_overlay.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/optimizer.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/cam_utils.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/utils/dist_utils.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/model_growth_utils.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/respace.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/dpm_solver.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/registry.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/timestep_sampler.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/diffusion_utils.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/edm_sample.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/builder.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/act.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/utils.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/liger_norms.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/gaussian_diffusion.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/norms.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/sa_solver.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/nets/sana_multi_scale_controlnet.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| LOW | diffusion/model/nets/sana_multi_scale_video.py | 1 | # Copyright 2024 NVIDIA CORPORATION & AFFILIATES |
| 111 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tools/create_wids_metadata.py | 32 | |
| LOW | tools/metrics/utils.py | 4 | |
| LOW | tools/metrics/geneval/evaluation/evaluate_images.py | 151 | |
| LOW | tools/metrics/geneval/generation/diffusers_generate.py | 84 | |
| LOW | tools/metrics/clip-score/clip_score.py | 85 | |
| LOW | tools/metrics/clip-score/clip_score.py | 106 | |
| LOW | tools/metrics/dpg_bench/compute_dpg_bench.py | 132 | |
| LOW | tools/metrics/pytorch-fid/compute_fid.py | 130 | |
| LOW | tools/convert_scripts/convert_sana_to_svdquant.py | 47 | |
| LOW | tools/convert_scripts/convert_sana_to_diffusers.py | 47 | |
| LOW | …ols/convert_scripts/convert_sana_video_to_diffusers.py | 33 | |
| LOW | …t_scripts/convert_ImgDataset_to_WebDatasetMS_format.py | 13 | |
| LOW | train_video_scripts/train_video_ivjoint_chunk.py | 105 | |
| LOW | train_video_scripts/train_video_ivjoint_chunk.py | 278 | |
| LOW | train_video_scripts/train_video_ivjoint_chunk.py | 854 | |
| LOW | train_video_scripts/train_video_ivjoint_chunk.py | 119 | |
| LOW | train_video_scripts/train_video_ivjoint.py | 240 | |
| LOW | train_video_scripts/train_video_ivjoint.py | 739 | |
| LOW | app/sana_pipeline.py | 167 | |
| LOW | app/sana_controlnet_pipeline.py | 190 | |
| LOW | app/sana_pipeline_inpaint.py | 324 | |
| LOW | app/app_sana_inpaint.py | 44 | |
| LOW | app/app_sana_inpaint.py | 278 | |
| LOW | sana/cli/upload2hf.py | 50 | |
| LOW | diffusion/post_training/ema.py | 27 | |
| LOW | diffusion/post_training/diffusers_patch/solver.py | 15 | |
| LOW | diffusion/guiders/adaptive_projected_guidance.py | 106 | |
| LOW | diffusion/scheduler/lcm_scheduler.py | 175 | |
| LOW | diffusion/scheduler/longlive_flow_euler_sampler.py | 420 | |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 138 | |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 334 | |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 414 | |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 438 | |
| LOW | diffusion/scheduler/sa_solver_diffusers.py | 482 | |
| LOW | diffusion/utils/misc.py | 317 | |
| LOW | diffusion/utils/checkpoint.py | 137 | |
| LOW | diffusion/utils/logger.py | 197 | |
| LOW | diffusion/utils/optimizer.py | 52 | |
| LOW | diffusion/utils/optimizer.py | 325 | |
| LOW | diffusion/utils/optimizer.py | 501 | |
| LOW | diffusion/utils/optimizer.py | 623 | |
| LOW | diffusion/utils/optimizer.py | 771 | |
| LOW | diffusion/utils/import_utils.py | 48 | |
| LOW | diffusion/utils/data_sampler.py | 105 | |
| LOW | diffusion/model/model_growth_utils.py | 134 | |
| LOW | diffusion/model/model_growth_utils.py | 240 | |
| LOW | diffusion/model/model_growth_utils.py | 297 | |
| LOW | diffusion/model/respace.py | 33 | |
| LOW | diffusion/model/respace.py | 113 | |
| LOW | diffusion/model/respace.py | 320 | |
| LOW | diffusion/model/dpm_solver.py | 266 | |
| LOW | diffusion/model/dpm_solver.py | 389 | |
| LOW | diffusion/model/dpm_solver.py | 439 | |
| LOW | diffusion/model/dpm_solver.py | 824 | |
| LOW | diffusion/model/dpm_solver.py | 1545 | |
| LOW | diffusion/model/dpm_solver.py | 1789 | |
| LOW | diffusion/model/edm_sample.py | 88 | |
| LOW | diffusion/model/builder.py | 138 | |
| LOW | diffusion/model/builder.py | 195 | |
| LOW | diffusion/model/builder.py | 320 | |
| 106 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | tools/metrics/clip-score/clip_score.py | 199 | except Exception as e: |
| MEDIUM | tools/metrics/clip-score/clip_score.py | 179 | def process_sample(sample): |
| LOW | tools/metrics/dpg_bench/compute_dpg_bench.py | 246 | except Exception as e: |
| LOW | tools/metrics/pytorch-fid/compute_fid.py | 42 | except Exception as e: |
| LOW | …t_scripts/convert_ImgDataset_to_WebDatasetMS_format.py | 52 | except Exception as e: |
| MEDIUM | …t_scripts/convert_ImgDataset_to_WebDatasetMS_format.py | 53 | print(f"Error processing file {filename}: {e}") |
| LOW | train_video_scripts/train_video_ivjoint.py | 370 | except Exception as e: |
| MEDIUM | train_video_scripts/train_video_ivjoint.py | 371 | print(f"Error in vae_encode: {e}") |
| LOW | train_video_scripts/train_video_ivjoint.py | 483 | except Exception: |
| LOW | app/app_sana_inpaint.py | 253 | except Exception as e: |
| LOW | app/app_sana_inpaint.py | 264 | except Exception as e: |
| LOW | app/app_sana_inpaint.py | 469 | except Exception as e: |
| LOW | sana/tools/hf_utils.py | 97 | except Exception as e: |
| MEDIUM | sana/tools/hf_utils.py | 98 | print(f"Error downloading file: {e}") |
| LOW | diffusion/scheduler/longlive_flow_euler_sampler.py | 307 | except Exception: |
| LOW | diffusion/scheduler/longlive_flow_euler_sampler.py | 351 | except Exception: |
| LOW | diffusion/utils/git.py | 167 | except Exception as e: |
| LOW | diffusion/utils/action_overlay.py | 138 | except Exception: |
| LOW | diffusion/model/nets/sana_multi_scale_video.py | 1010 | except Exception as e: |
| MEDIUM | diffusion/model/nets/sana_multi_scale_video.py | 1011 | print(f"Error loading {key}: {e}") |
| LOW | diffusion/model/nets/sana_multi_scale_video_camctrl.py | 1526 | except Exception as e: |
| MEDIUM | diffusion/model/nets/sana_multi_scale_video_camctrl.py | 1527 | print(f"Error loading {key}: {e}") |
| LOW | diffusion/model/nets/sana.py | 402 | except Exception as e: |
| LOW | diffusion/model/dc_ae/efficientvit/apps/utils/image.py | 69 | except Exception: |
| LOW | diffusion/model/dc_ae/efficientvit/apps/utils/image.py | 177 | except Exception: |
| LOW | diffusion/model/dc_ae/efficientvit/apps/utils/dist.py | 43 | except Exception: |
| LOW | diffusion/model/ops/fused_gdn_chunkwise.py | 1156 | except Exception: |
| LOW | …line/sana_inference_interactive_pipeline_long_chunk.py | 29 | except Exception: |
| LOW | …line/sana_inference_interactive_pipeline_long_chunk.py | 176 | except Exception: |
| LOW | …line/sana_inference_interactive_pipeline_long_chunk.py | 198 | except Exception: |
| LOW | …line/sana_inference_interactive_pipeline_long_chunk.py | 211 | except Exception: |
| LOW | …line/sana_inference_interactive_pipeline_long_chunk.py | 247 | except Exception as e: |
| LOW | …line/sana_inference_interactive_pipeline_long_chunk.py | 250 | except Exception: |
| LOW | …ngsana/pipeline/sana_inference_interactive_pipeline.py | 32 | except Exception: |
| LOW | …ngsana/pipeline/sana_inference_interactive_pipeline.py | 192 | except Exception: |
| LOW | …ngsana/pipeline/sana_inference_interactive_pipeline.py | 249 | except Exception: |
| LOW | …ngsana/pipeline/sana_inference_interactive_pipeline.py | 263 | except Exception: |
| LOW | …ngsana/pipeline/sana_inference_interactive_pipeline.py | 301 | except Exception as e: |
| LOW | …ngsana/pipeline/sana_inference_interactive_pipeline.py | 306 | except Exception: |
| LOW | diffusion/longsana/pipeline/sana_inference_pipeline.py | 43 | except Exception: |
| LOW | diffusion/longsana/pipeline/sana_training_pipeline.py | 67 | except Exception: |
| LOW | diffusion/longsana/pipeline/sana_training_pipeline.py | 619 | except Exception: |
| LOW | diffusion/longsana/pipeline/sana_training_pipeline.py | 638 | except Exception: |
| LOW | …ion/longsana/pipeline/sana_switch_training_pipeline.py | 111 | except Exception as e: |
| LOW | …ion/longsana/pipeline/sana_switch_training_pipeline.py | 114 | except Exception: |
| LOW | diffusion/longsana/utils/model_wrapper.py | 147 | except Exception: |
| LOW | diffusion/longsana/utils/model_wrapper.py | 232 | except Exception: |
| LOW | diffusion/longsana/model/ode_regression_sana.py | 60 | except Exception: |
| LOW | diffusion/longsana/model/ode_regression_sana.py | 112 | except Exception: |
| LOW | diffusion/longsana/model/ode_regression_sana.py | 144 | except Exception: |
| LOW | diffusion/longsana/model/ode_regression_sana.py | 190 | except Exception: |
| LOW | diffusion/longsana/model/dmd_sana.py | 109 | except Exception: |
| LOW | diffusion/longsana/model/dmd_sana.py | 129 | except Exception: |
| LOW | diffusion/longsana/model/dmd_sana.py | 143 | except Exception: |
| LOW | diffusion/longsana/model/dmd_sana.py | 224 | except Exception: |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 400 | except Exception as e: |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 661 | except Exception as _e: |
| LOW | diffusion/longsana/trainer/longsana_trainer.py | 470 | except Exception as e: |
| LOW | diffusion/longsana/trainer/longsana_trainer.py | 476 | except Exception as e: |
| LOW | diffusion/longsana/trainer/ode.py | 302 | except Exception as e: |
| 22 more matches not shown… | |||
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | …t_scripts/convert_ImgDataset_to_WebDatasetMS_format.py | 57 | # Create a TAR file and include all files |
| MEDIUM | app/safety_check.py | 16 | # This file is copy from https://github.com/mit-han-lab/hart/blob/main/hart/utils/safety_check.py |
| MEDIUM | diffusion/utils/cam_utils.py | 77 | # Create the base (N, 4, 4) tensor (identity matrix padded) |
| MEDIUM | diffusion/model/dpm_solver.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana_multi_scale_controlnet.py | 19 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana_multi_scale_video.py | 18 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana_blocks.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana_multi_scale_adaln.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana_gdn_blocks.py | 49 | """This function is intended to align with the l2norm implementation in the FLA library.""" |
| MEDIUM | diffusion/model/nets/sana_U_shape.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana_multi_scale_video_camctrl.py | 18 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/basic_modules.py | 19 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana_U_shape_multi_scale.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana_multi_scale.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/model/nets/sana.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/data/datasets/sana_data_multi_scale.py | 18 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/data/datasets/sana_data.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/data/datasets/utils.py | 17 | # This file is modified from https://github.com/PixArt-alpha/PixArt-sigma |
| MEDIUM | diffusion/data/wids/wids_tar.py | 17 | # This file is copied from https://github.com/NVlabs/VILA/tree/main/llava/wids |
| MEDIUM | diffusion/data/wids/wids_tar.py | 49 | # Create the index |
| MEDIUM | diffusion/data/wids/wids_tar.py | 59 | # Create an empty list for the index |
| MEDIUM | diffusion/data/wids/wids_mmtar.py | 17 | # This file is copied from https://github.com/NVlabs/VILA/tree/main/llava/wids |
| MEDIUM | diffusion/data/wids/__init__.py | 2 | # This file is part of the WebDataset library. |
| MEDIUM | diffusion/data/wids/wids_specs.py | 17 | # This file is copied from https://github.com/NVlabs/VILA/tree/main/llava/wids |
| MEDIUM | diffusion/data/wids/wids.py | 17 | # This file is modified from https://github.com/NVlabs/VILA/tree/main/llava/wids |
| MEDIUM | diffusion/data/wids/wids_dl.py | 17 | # This file is copied from https://github.com/NVlabs/VILA/tree/main/llava/wids |
| MEDIUM | diffusion/data/wids/wids_lru.py | 17 | # This file is copied from https://github.com/NVlabs/VILA/tree/main/llava/wids |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | app/sana_video_refiner_pipeline_diffusers.py | 53 | # Step 1: Generate latent using Sana Video Pipeline |
| LOW | app/sana_video_refiner_pipeline_diffusers.py | 93 | # Step 2: Load LTX2 Pipeline |
| LOW | app/sana_video_refiner_pipeline_diffusers.py | 168 | # Step 3: Manually pack video latent (skip diffusers' normalize, consistent with original code) |
| LOW | app/sana_video_refiner_pipeline_diffusers.py | 180 | # Step 4: Create audio latent (becomes zero after normalize, consistent with original code) |
| LOW | app/sana_video_refiner_pipeline_diffusers.py | 202 | # Step 5: LTX2 Stage-2 Refinement |
| LOW | app/sana_video_refiner_pipeline_diffusers.py | 233 | # Step 6: Save output |
| LOW | diffusion/longsana/model/ode_regression_sana.py | 105 | # Step 1: Initialize all models |
| LOW | diffusion/longsana/model/ode_regression_sana.py | 126 | # Step 2: Initialize all hyperparameters |
| LOW | diffusion/longsana/model/ode_regression_sana.py | 327 | # Step 1: Run generator on noisy latents |
| LOW | diffusion/longsana/model/ode_regression_sana.py | 342 | # Step 2: Compute the regression loss |
| LOW | diffusion/longsana/model/dmd_sana.py | 575 | # Step 1: Sample noise and backward simulate the generator's input |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 30 | # Step 1: Initialize the distributed training environment (rank, seed, dtype, logging etc.) |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 68 | # Step 2: Initialize the model |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 71 | # Step 2: Initialize the model and optimizer |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 143 | # Step 4: Initialize the optimizer |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 157 | # Step 5: Initialize the dataloader |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 172 | # Step 6: Initialize the validation dataloader for visualization (fixed prompts) |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 446 | # Step 1: Get the next batch of text prompts |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 462 | # Step 2: Extract the conditional infos for sana |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 489 | # Step 3: Store gradients for the generator (if training the generator) |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 510 | # Step 4: Store gradients for the critic (if training the critic) |
| LOW | diffusion/longsana/trainer/ode.py | 32 | # Step 1: Initialize the distributed training environment (rank, seed, dtype, logging etc.) |
| LOW | diffusion/longsana/trainer/ode.py | 69 | # Step 2: Initialize the model and optimizer |
| LOW | diffusion/longsana/trainer/ode.py | 92 | # Step 4: Initialize the optimizer |
| LOW | diffusion/longsana/trainer/ode.py | 100 | # Step 5: Initialize the dataloader |
| LOW | diffusion/longsana/trainer/ode.py | 111 | # Step 6: Initialize the validation dataloader for visualization (fixed prompts) |
| LOW | diffusion/longsana/trainer/ode.py | 466 | # Step 1: Get the next batch of text prompts |
| LOW | diffusion/longsana/trainer/ode.py | 471 | # Step 2: Extract the conditional infos |
| LOW | diffusion/longsana/trainer/ode.py | 475 | # Step 3: Train the generator |
| LOW | diffusion/longsana/trainer/ode.py | 512 | # Step 4: Visualization |
| LOW | diffusion/longsana/trainer/ode.py | 565 | # Step 5: Logging |
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | diffusion/utils/chunk_utils.py | 76 | Convert chunk_size to chunk_index list with a split strategy. Args: T: Number of latent frames. chu |
| HIGH | diffusion/utils/chunk_utils.py | 135 | Resolve chunk_index from a config, supporting chunk_size and strategy. Priority: 1) config.model.chunk_index |
| HIGH | diffusion/utils/chunk_utils.py | 214 | Return frame-time positions belonging to size-1 (singleton) chunks. A size-1 chunk has no intra-chunk lookahead, so |
| HIGH | diffusion/utils/chunk_utils.py | 384 | Normalize chunk_index and detect if uniform. This function handles all the complex logic for: 1. Converting chu |
| HIGH | diffusion/model/model_growth_utils.py | 70 | Args: strategy: init strategy name **kwargs: strategy specific parameters Ret |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | environment_setup.sh | 16 | # Check if we should skip environment setup entirely (used by CI). |
| LOW | app/app_sana_inpaint.py | 449 | # Check if this is a new image by comparing with stored original |
| LOW | diffusion/utils/chunk_utils.py | 342 | # Check if tail (sizes[1:]) is uniform |
| LOW | diffusion/utils/chunk_utils.py | 359 | # Check if tail (sizes[1:]) is uniform |
| LOW | diffusion/utils/chunk_utils.py | 418 | # Check if uniform (requires chunk_size for comparison) |
| LOW | diffusion/utils/chunk_utils.py | 446 | # Check if uniform |
| LOW | diffusion/utils/data_sampler.py | 262 | # Assign samples to each bucket |
| LOW | diffusion/model/utils.py | 136 | # Check if resizing is needed |
| LOW | diffusion/longsana/utils/dataset.py | 294 | # Check if all samples list length is consistent |
| LOW | diffusion/longsana/model/streaming_sana_long.py | 376 | # Check if previous_frames can be used for overlap and auto-compute overlap frame count |
| LOW | diffusion/longsana/trainer/self_forcing_trainer.py | 613 | # Check if we've reached max iterations |
| LOW | diffusion/longsana/trainer/longsana_trainer.py | 282 | # Check if we should train generator on this optimization step |
| LOW | diffusion/data/wids/wids.py | 644 | # Check if we're missing the cache too often. |
| Severity | File | Line | Snippet |
|---|---|---|---|
| HIGH | asset/docs/sana_lora_dreambooth.md | 111 | - `report_to="wandb` will ensure the training runs are tracked on [Weights and Biases](https://wandb.ai/site). To use it |
| HIGH | docs/sana_lora_dreambooth.md | 111 | - `report_to="wandb` will ensure the training runs are tracked on [Weights and Biases](https://wandb.ai/site). To use it |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | environment_setup.sh | 8 | # Usage: |
| LOW | app/app_sana_inpaint.py | 315 | ### How to use: |
| Severity | File | Line | Snippet |
|---|---|---|---|
| MEDIUM | diffusion/longsana/model/dmd_sana.py | 217 | # TODO: Need to make the path more robust. |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | …t_scripts/convert_ImgDataset_to_WebDatasetMS_format.py | 13 | def process_data(input_dir, output_tar_name="output.tar"): |
| Severity | File | Line | Snippet |
|---|---|---|---|
| LOW | train_scripts/train_dreambooth_lora_sana.py | 1016 | # make sure to pop weight so that corresponding model is not saved again |