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
__init__.py
ar_nar.py
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