mrq ad7e290a5e ugh (ROCm seems to silently clamp any token value >= logits.shape[-1] for loss calculation, while cuda will throw an assert, making it hard to find this dumb fuckup)		2024-11-09 19:40:02 -06:00
..
arch	This better work	2024-11-09 18:04:59 -06:00
__init__.py
ar_nar.py	'borrowed' a sampling scheduler for NAR-len's RVQ level 0 (better than before, but still not good enough)	2024-11-07 21:19:14 -06:00
ar.py
base.py	ugh (ROCm seems to silently clamp any token value >= logits.shape[-1] for loss calculation, while cuda will throw an assert, making it hard to find this dumb fuckup)	2024-11-09 19:40:02 -06:00
experimental.py
lora.py
nar.py	I don't know why this fixes an assert thrown but it does	2024-11-09 19:04:13 -06:00