MANIFEST.in
README.md
requirements.txt
setup.py
acids_msprior.egg-info/PKG-INFO
acids_msprior.egg-info/SOURCES.txt
acids_msprior.egg-info/dependency_links.txt
acids_msprior.egg-info/entry_points.txt
acids_msprior.egg-info/requires.txt
acids_msprior.egg-info/top_level.txt
msprior/__init__.py
msprior/attention.py
msprior/dataset.py
msprior/preprocessor.py
msprior/rwkv.py
msprior/scripted.py
msprior/task.py
msprior/utils.py
msprior/configs/decoder_only.gin
msprior/configs/flattened.gin
msprior/configs/modulated_alibi.gin
msprior/configs/recurrent.gin
msprior/configs/rwkv.gin
msprior/configs/rwkv_semantic.gin
msprior/cuda/wkv_cuda.cu
msprior/cuda/wkv_op.cpp
msprior_scripts/__init__.py
msprior_scripts/combine.py
msprior_scripts/compact.py
msprior_scripts/export.py
msprior_scripts/main_cli.py
msprior_scripts/preprocess.py
msprior_scripts/train.py
tests/__init__.py
tests/test_attention.py
tests/test_cache.py
tests/test_configs.py