mrq 451726fdd5 added ability to disable activation checkpointing through the YAML (it is very VRAM intensive at double layer size)		2023-09-05 15:38:21 -05:00
..
__init__.py	added ability to disable activation checkpointing through the YAML (it is very VRAM intensive at double layer size)	2023-09-05 15:38:21 -05:00
adaln.py
ar.py	merged dedicated interleaved AR code with the normal AR code	2023-09-03 22:46:08 -05:00
base.py	added ability to disable activation checkpointing through the YAML (it is very VRAM intensive at double layer size)	2023-09-05 15:38:21 -05:00
nar.py	merged dedicated interleaved AR code with the normal AR code	2023-09-03 22:46:08 -05:00
retnet.py
transformer.py	added ability to disable activation checkpointing through the YAML (it is very VRAM intensive at double layer size)	2023-09-05 15:38:21 -05:00