Jedrzej Kosinski 4195dfb032 Merge branch 'master' into patch_hooks_improved_memory		2024-11-11 11:26:27 -06:00
..
diffusionmodules	Refactored code to store wrappers and callbacks in transformer_options, added apply_model and diffusion_model.forward wrappers	2024-11-03 06:22:48 -06:00
distributions	…
encoders	…
attention.py	Mochi VAE encoder.	2024-11-01 17:33:09 -04:00
ema.py	…
sub_quadratic_attention.py	Switch some more prints to logging.	2024-03-11 16:34:58 -04:00
temporal_ae.py	…