Skip to content

GitLab

Switch branch/tag
  1. 13 Nov, 2025 2 commits
    • 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
  3. 12 Nov, 2025 7 commits
  5. 11 Nov, 2025 2 commits
  7. 10 Nov, 2025 8 commits
  9. 07 Nov, 2025 3 commits
  11. 06 Nov, 2025 4 commits
  13. 04 Nov, 2025 3 commits
  15. 03 Nov, 2025 2 commits
  17. 02 Nov, 2025 3 commits
  19. 31 Oct, 2025 2 commits
  21. 30 Oct, 2025 2 commits
  23. 28 Oct, 2025 2 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
    • Sayak Paul's avatar
      [ci] don't run sana layerwise casting tests in CI. (#12551) · 55d49d43
      Sayak Paul authored 
      * don't run sana layerwise casting tests in CI.

* up
      55d49d43