mrq 06e948aec1 suppress warning on exit about distributed not being cleaned up (because I updated my system)		2024-07-25 16:50:47 -05:00
..
arch	mamba2-hf using vasqu/mamba2-torch because it lets me use mamba2 without triton ops (training with my 4xV100s are not happy with mamba2 because of triton)	2024-06-14 19:42:17 -05:00
__init__.py	sanity cleanup: moved experimental features under its own thing	2024-06-30 10:37:33 -05:00
ar_nar.py	suppress warning on exit about distributed not being cleaned up (because I updated my system)	2024-07-25 16:50:47 -05:00
base.py	added experimental training setting to perform token dropout to MAYBE compensate for errors from the preceding RVQ level (two types: token error offset, token dropout embedding replace)	2024-07-24 19:35:17 -05:00
experimental.py	sanity cleanup	2024-07-04 15:58:08 -05:00
lora.py	some weird fixes for an equally weird regression with LoRA loading	2024-07-22 20:47:24 -05:00
nar.py	cleaned up demo page creation, added option to pass in RVQ level sampling distribution for training	2024-07-21 19:12:03 -05:00