Skip to content

GitLab

Switch branch/tag
  1. 17 Nov, 2025 3 commits
  3. 15 Nov, 2025 1 commit
  5. 14 Nov, 2025 1 commit
    • David El Malih's avatar
      Improve docstrings and type hints in scheduling_euler_discrete.py (#12654) · 63dd6017
      David El Malih authored 
      * refactor: enhance type hints and documentation in EulerDiscreteScheduler

Updated type hints for function parameters and return types in the EulerDiscreteScheduler class to improve code clarity and maintainability. Enhanced docstrings for several methods to provide clearer descriptions of their functionality and expected arguments. This includes specifying Literal types for certain parameters and ensuring consistent return type annotations across the class.

* refactor: enhance type hints and documentation across multiple schedulers

Updated type hints and improved docstrings in various scheduler classes, including CMStochasticIterativeScheduler, CosineDPMSolverMultistepScheduler, and others. This includes specifying parameter types, return types, and providing clearer descriptions of method functionalities. Notable changes include the addition of default values in the begin_index argument and enhanced explanations for noise addition methods. These improvements aim to enhance code clarity and maintainability across the scheduling module.

* refactor: update docstrings to clarify noise schedule construction

Revised docstrings across multiple scheduler classes to enhance clarity regarding the construction of noise schedules. Updated references to relevant papers, ensuring accurate citations for the methodologies used. This includes changes in DEISMultistepScheduler, DPMSolverMultistepInverseScheduler, and others, improving documentation consistency and readability.
      63dd6017
  7. 13 Nov, 2025 4 commits
    • David El Malih's avatar
      Improve docstrings and type hints in scheduling_ddpm.py (#12651) · 3c1ca869
      David El Malih authored 
      * Enhance type hints and docstrings in scheduling_ddpm.py

- Added type hints for function parameters and return types across the DDPMScheduler class and related functions.
- Improved docstrings for clarity, including detailed descriptions of parameters and return values.
- Updated the alpha_transform_type and beta_schedule parameters to use Literal types for better type safety.
- Refined the _get_variance and previous_timestep methods with comprehensive documentation.

* Refactor docstrings and type hints in scheduling_ddpm.py

- Cleaned up whitespace in the rescale_zero_terminal_snr function.
- Enhanced the variance_type parameter in the DDPMScheduler class with improved formatting for better readability.
- Updated the docstring for the compute_variance method to maintain consistency and clarity in parameter descriptions and return values.

* Apply `make fix-copies`

* Refactor type hints across multiple scheduler files

- Updated type hints to include `Literal` for improved type safety in various scheduling files.
- Ensured consistency in type hinting for parameters and return types across the affected modules.
- This change enhances code clarity and maintainability.
      3c1ca869
    • David El Malih's avatar
      Improve docstrings and type hints in scheduling_ddim.py (#12622) · 6fe4a6ff
      David El Malih authored 
      * Improve docstrings and type hints in scheduling_ddim.py

- Add complete type hints for all function parameters
- Enhance docstrings to follow project conventions
- Add missing parameter descriptions

Fixes #9567

* Enhance docstrings and type hints in scheduling_ddim.py

- Update parameter types and descriptions for clarity
- Improve explanations in method docstrings to align with project standards
- Add optional annotations for parameters where applicable

* Refine type hints and docstrings in scheduling_ddim.py

- Update parameter types to use Literal for specific string options
- Enhance docstring descriptions for clarity and consistency
- Ensure all parameters have appropriate type annotations and defaults

* Apply review feedback on scheduling_ddim.py

- Replace "prevent singularities" with "avoid numerical instability" for better clarity
- Add backticks around `alpha_bar` variable name for consistent formatting
- Convert Imagen Video paper URLs to Hugging Face papers references

* Propagate changes using 'make fix-copies'

* Add missing Literal
      6fe4a6ff
    • dg845's avatar
      [WIP]Add Wan2.2 Animate Pipeline (Continuation of #12442 by tolgacangoz) (#12526) · d8e48058
      dg845 authored 
      



---------
Co-authored-by: default avatarTolga Cangöz <mtcangoz@gmail.com>
Co-authored-by: default avatarTolga Cangöz <46008593+tolgacangoz@users.noreply.github.com>
      d8e48058
    • David El Malih's avatar
      Improve docstrings and type hints in scheduling_amused.py (#12623) · 44c31016
      David El Malih authored 
      * Improve docstrings and type hints in scheduling_amused.py

- Add complete type hints for helper functions (gumbel_noise, mask_by_random_topk)
- Enhance AmusedSchedulerOutput with proper Optional typing
- Add comprehensive docstrings for AmusedScheduler class
- Improve __init__, set_timesteps, step, and add_noise methods
- Fix type hints to match documentation conventions
- All changes follow project standards from issue #9567

* Enhance type hints and docstrings in scheduling_amused.py

- Update type hints for `prev_sample` and `pred_original_sample` in `AmusedSchedulerOutput` to reflect their tensor types.
- Improve docstring for `gumbel_noise` to specify the output tensor's dtype and device.
- Refine `AmusedScheduler` class documentation, including detailed descriptions of the masking schedule and temperature parameters.
- Adjust type hints in `set_timesteps` and `step` methods for better clarity and consistency.

* Apply review feedback on scheduling_amused.py

- Replace generic [Amused] reference with specific [`AmusedPipeline`] reference for consistency with project documentation conventions
      44c31016
  9. 12 Nov, 2025 6 commits
    • YiYi Xu's avatar
      [modular] add a check (#12628) · d6c63bb9
      YiYi Xu authored 
      * add

* fix
      d6c63bb9
    • Quentin Gallouédec's avatar
      ArXiv -> HF Papers (#12583) · f3db38c1
      Quentin Gallouédec authored 
      * Update pipeline_skyreels_v2_i2v.py

* Update README.md

* Update torch_utils.py

* Update torch_utils.py

* Update guider_utils.py

* Update pipeline_ltx.py

* Update pipeline_bria.py

* Apply suggestion from @qgallouedec

* Update autoencoder_kl_qwenimage.py

* Update pipeline_prx.py

* Update pipeline_wan_vace.py

* Update pipeline_skyreels_v2.py

* Update pipeline_skyreels_v2_diffusion_forcing.py

* Update pipeline_bria_fibo.py

* Update pipeline_skyreels_v2_diffusion_forcing_i2v.py

* Update pipeline_ltx_condition.py

* Update pipeline_ltx_image2video.py

* Update regional_prompting_stable_diffusion.py

* make style

* style

* style
      f3db38c1
    • Sayak Paul's avatar
      [modular] add tests for qwen modular (#12585) · f5e5f348
      Sayak Paul authored 
      * add tests for qwenimage modular.

* qwenimage edit.

* qwenimage edit plus.

* empty

* align with the latest structure

* up

* up

* reason

* up

* fix multiple issues.

* up

* up

* fix

* up

* make it similar to the original pipeline.
      f5e5f348
    • YiYi Xu's avatar
      fix dispatch_attention_fn check (#12636) · 093cd3f0
      YiYi Xu authored 
      * fix

* fix
      093cd3f0
    • a120092009's avatar
      Add MLU Support. (#12629) · aecf0c53
      a120092009 authored 
      

* Add MLU Support.

* fix comment.

* rename is_mlu_available to is_torch_mlu_available

* Apply style fixes

---------
Co-authored-by: default avatargithub-actions[bot] <github-actions[bot]@users.noreply.github.com>
      aecf0c53
    • YiYi Xu's avatar
      fix copies (#12637) · 0c758929
      YiYi Xu authored 
      * fix

* remoce cocpies instead
      0c758929
  11. 11 Nov, 2025 1 commit
  13. 10 Nov, 2025 4 commits
  15. 07 Nov, 2025 2 commits
  17. 06 Nov, 2025 3 commits
  19. 04 Nov, 2025 1 commit
  21. 03 Nov, 2025 2 commits
  23. 02 Nov, 2025 3 commits
  25. 31 Oct, 2025 2 commits
  27. 30 Oct, 2025 2 commits
  29. 28 Oct, 2025 5 commits
    • galbria's avatar
      Bria fibo (#12545) · 84e16575
      galbria authored 
      

* Bria FIBO pipeline

* style fixs

* fix CR

* Refactor BriaFibo classes and update pipeline parameters

- Updated BriaFiboAttnProcessor and BriaFiboAttention classes to reflect changes from Flux equivalents.
- Modified the _unpack_latents method in BriaFiboPipeline to improve clarity.
- Increased the default max_sequence_length to 3000 and added a new optional parameter do_patching.
- Cleaned up test_pipeline_bria_fibo.py by removing unused imports and skipping unsupported tests.

* edit the docs of FIBO

* Remove unused BriaFibo imports and update CPU offload method in BriaFiboPipeline

* Refactor FIBO classes to BriaFibo naming convention

- Updated class names from FIBO to BriaFibo for consistency across the module.
- Modified instances of FIBOEmbedND, FIBOTimesteps, TextProjection, and TimestepProjEmbeddings to reflect the new naming.
- Ensured all references in the BriaFiboTransformer2DModel are updated accordingly.

* Add BriaFiboTransformer2DModel import to transformers module

* Remove unused BriaFibo imports from modular pipelines and add BriaFiboTransformer2DModel and BriaFiboPipeline classes to dummy objects for enhanced compatibility with torch and transformers.

* Update BriaFibo classes with copied documentation and fix import typo in pipeline module

- Added documentation comments indicating the source of copied code in BriaFiboTransformerBlock and _pack_latents methods.
- Corrected the import statement for BriaFiboPipeline in the pipelines module.

* Remove unused BriaFibo imports from __init__.py to streamline modular pipelines.

* Refactor documentation comments in BriaFibo classes to indicate inspiration from existing implementations

- Updated comments in BriaFiboAttnProcessor, BriaFiboAttention, and BriaFiboPipeline to reflect that the code is inspired by other modules rather than copied.
- Enhanced clarity on the origins of the methods to maintain proper attribution.

* change Inspired by to Based on

* add reference link and fix trailing whitespace

* Add BriaFiboTransformer2DModel documentation and update comments in BriaFibo classes

- Introduced a new documentation file for BriaFiboTransformer2DModel.
- Updated comments in BriaFiboAttnProcessor, BriaFiboAttention, and BriaFiboPipeline to clarify the origins of the code, indicating copied sources for better attribution.

---------
Co-authored-by: default avatarsayakpaul <spsayakpaul@gmail.com>
      84e16575
    • Wang, Yi's avatar
      fix crash if tiling mode is enabled (#12521) · dc622a95
      Wang, Yi authored 
      

* fix crash in tiling mode is enabled
Signed-off-by: default avatarWang, Yi A <yi.a.wang@intel.com>

* fmt
Signed-off-by: default avatarWang, Yi A <yi.a.wang@intel.com>

---------
Signed-off-by: default avatarWang, Yi A <yi.a.wang@intel.com>
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
      dc622a95
    • Dhruv Nair's avatar
      [Pipelines] Enable Wan VACE to run since single transformer (#12428) · ecfbc8f9
      Dhruv Nair authored 
      * update

* update

* update

* update

* update
      ecfbc8f9
    • Sayak Paul's avatar
      support latest few-step wan LoRA. (#12541) · df0e2a4f
      Sayak Paul authored 
      * support latest few-step wan LoRA.

* up

* up
      df0e2a4f
    • G.O.D's avatar
      Improve pos embed for Flux.1 inference on Ascend NPU (#12534) · 303efd2b
      G.O.D authored 
      

improve pos embed for ascend npu
Co-authored-by: default avatarfelix01.yu <felix01.yu@vipshop.com>
      303efd2b