.gitignore
LICENSE
README.md
pyproject.toml
requirements.txt
.github/workflows/python-publish.yml
configs/_base_/default_runtime.py
configs/_base_/datasets/bert.py
configs/_base_/datasets/gpt.py
configs/_base_/models/bert.py
configs/_base_/models/gpt.py
configs/_base_/schedules/default_schedule.py
configs/bert/bert.py
configs/gpt/gpt.py
src/nami/__init__.py
src/nami/registry.py
src/nami/version.py
src/nami/apis/__init__.py
src/nami/apis/app.py
src/nami/apis/pipeline.py
src/nami/apis/train.py
src/nami/datasets/__init__.py
src/nami/datasets/base_dataset.py
src/nami/datasets/dataset_gpt.py
src/nami/datasets/dataset_rm.py
src/nami/datasets/collate/__init__.py
src/nami/datasets/collate/collate_gpt.py
src/nami/datasets/collate/collate_rm.py
src/nami/datasets/transforms/__init__.py
src/nami/evaluation/__init__.py
src/nami/evaluation/metrics/__init__.py
src/nami/evaluation/metrics/metrics_gpt.py
src/nami/evaluation/metrics/metrics_rm.py
src/nami/hooks/__init__.py
src/nami/hooks/resize_token_embedding_hook.py
src/nami/hooks/save_pretrained_hook.py
src/nami/losses/__init__.py
src/nami/losses/cross_entropy_loss.py
src/nami/losses/rank_loss.py
src/nami/losses/utils.py
src/nami/models/__init__.py
src/nami/models/bert/__init__.py
src/nami/models/bert/configuration_bert.py
src/nami/models/bert/modeling_bert.py
src/nami/models/gpt2/__init__.py
src/nami/models/gpt2/configuration_gpt2.py
src/nami/models/gpt2/modeling_gpt2.py
src/nami/models/utils/__init__.py
src/nami/models/utils/base_model.py
src/nami/models/utils/builder.py
src/nami/models/utils/layers.py
src/nami/pipelines/__init__.py
src/pynami.egg-info/PKG-INFO
src/pynami.egg-info/SOURCES.txt
src/pynami.egg-info/dependency_links.txt
src/pynami.egg-info/requires.txt
src/pynami.egg-info/top_level.txt
tools/dist_train.sh
tools/slurm_train.sh