mirror of https://github.com/huggingface/diffusers.git synced 2025-12-11 23:14:37 +08:00

Files

Sayak Paul 137403ff31 [Core] Introduce class variants for Transformer2DModel (#7647 )

* init for patches

* finish patched model.

* continuous transformer

* vectorized transformer2d.

* style.

* inits.

* fix-copies.

* introduce DiTTransformer2DModel.

* fixes

* use REMAPPING as suggested by @DN6

* better logging.

* add pixart transformer model.

* inits.

* caption_channels.

* attention masking.

* fix use_additional_conditions.

* remove print.

* debug

* flatten

* fix: assertion for sigma

* handle remapping for modeling_utils

* add tests for dit transformer2d

* quality

* placeholder for pixart tests

* pixart tests

* add _no_split_modules

* add docs.

* check

* check

* check

* check

* fix tests

* fix tests

* move Transformer output to modeling_output

* move errors better and bring back use_additional_conditions attribute.

* add unnecessary things from DiT.

* clean up pixart

* fix remapping

* fix device_map things in pixart2d.

* replace Transformer2DModel with appropriate classes in dit, pixart tests

* empty

* legacy mixin classes./

* use a remapping dict for fetching class names.

* change to specifc model types in the pipeline implementations.

* move _fetch_remapped_cls_from_config to modeling_loading_utils.py

* fix dependency problems.

* add deprecation note.

2024-06-04 14:22:00 +04:00

763 B

Raw Blame History

DiTTransformer2D

A Transformer model for image-like data from DiT.

DiTTransformer2DModel

autodoc DiTTransformer2DModel

763 B Raw Blame History

DiTTransformer2D

DiTTransformer2DModel

763 B

Raw Blame History