Name		Name	Last commit message	Last commit date
Latest commit History 225 Commits
.github		.github
caikit_nlp		caikit_nlp
examples		examples
protos/training-management-protos		protos/training-management-protos
runtime_template		runtime_template
scripts		scripts
tests		tests
.dockerignore		.dockerignore
.gitignore		.gitignore
.isort.cfg		.isort.cfg
.pre-commit-config.yaml		.pre-commit-config.yaml
.prettierignore		.prettierignore
.pylintrc		.pylintrc
.whitesource		.whitesource
CODEOWNERS		CODEOWNERS
CONTRIBUTING.md		CONTRIBUTING.md
LICENSE		LICENSE
README.md		README.md
SECURITY.md		SECURITY.md
code-of-conduct.md		code-of-conduct.md
parameter_selection.md		parameter_selection.md
pyproject.toml		pyproject.toml
redacted_history.txt		redacted_history.txt
runtime_config.yaml		runtime_config.yaml
setup_requirements.txt		setup_requirements.txt
tox.ini		tox.ini

Repository files navigation

Caikit NLP

caikit_nlp is a Caikit library that currently provides PEFT prompt tuning and MPT (multi-task prompt tuning) functionalities.

More information on MPT can be found at: https://arxiv.org/abs/2303.02861

Currently causal language models and sequence-to-sequence models are supported.

The data model for text generative capabilities is baked into this repository itself at caikit_nlp/data_model/generation.py.