.dockerignore
.gitignore
.isort.cfg
.pre-commit-config.yaml
.prettierignore
.pylintrc
.whitesource
CODEOWNERS
CONTRIBUTING.md
Dockerfile
LICENSE
README.md
SECURITY.md
code-of-conduct.md
prompt_tuning_parameter_selection.md
pyproject.toml
runtime_config.yaml
setup_requirements.txt
tox.ini
.github/dependabot.yml
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/ISSUE_TEMPLATE/user_story.md
.github/workflows/build-image.yml
.github/workflows/build-library.yml
.github/workflows/lint-code.yml
.github/workflows/publish-library.yml
benchmarks/README.md
benchmarks/logs/llama2-7b/20230905_183655.output
benchmarks/logs/llama2-7b/20230905_184809.output
benchmarks/logs/llama2-7b/20230905_191650.output
benchmarks/logs/llama2-7b/20230905_194133.output
benchmarks/logs/llama2-7b/20230906_135211.output
caikit_nlp/__init__.py
caikit_nlp/_version.py
caikit_nlp/version.py
caikit_nlp.egg-info/PKG-INFO
caikit_nlp.egg-info/SOURCES.txt
caikit_nlp.egg-info/dependency_links.txt
caikit_nlp.egg-info/requires.txt
caikit_nlp.egg-info/top_level.txt
caikit_nlp/config/__init__.py
caikit_nlp/config/config.yml
caikit_nlp/data_model/__init__.py
caikit_nlp/data_model/generation.py
caikit_nlp/model_management/__init__.py
caikit_nlp/model_management/tgis_auto_finder.py
caikit_nlp/modules/__init__.py
caikit_nlp/modules/text_classification/__init__.py
caikit_nlp/modules/text_classification/sequence_classification.py
caikit_nlp/modules/text_embedding/__init__.py
caikit_nlp/modules/text_embedding/embedding.py
caikit_nlp/modules/text_embedding/utils.py
caikit_nlp/modules/text_generation/__init__.py
caikit_nlp/modules/text_generation/peft_config.py
caikit_nlp/modules/text_generation/peft_prompt_tuning.py
caikit_nlp/modules/text_generation/peft_tgis_remote.py
caikit_nlp/modules/text_generation/text_generation_local.py
caikit_nlp/modules/text_generation/text_generation_tgis.py
caikit_nlp/modules/token_classification/__init__.py
caikit_nlp/modules/token_classification/filtered_span_classification.py
caikit_nlp/modules/tokenization/__init__.py
caikit_nlp/modules/tokenization/regex_sentence_splitter.py
caikit_nlp/resources/__init__.py
caikit_nlp/resources/pretrained_model/__init__.py
caikit_nlp/resources/pretrained_model/base.py
caikit_nlp/resources/pretrained_model/hf_auto_causal_lm.py
caikit_nlp/resources/pretrained_model/hf_auto_seq2seq_lm.py
caikit_nlp/resources/pretrained_model/hf_auto_seq_classifier.py
caikit_nlp/toolkit/__init__.py
caikit_nlp/toolkit/data_stream_wrapper.py
caikit_nlp/toolkit/data_type_utils.py
caikit_nlp/toolkit/task_specific_utils.py
caikit_nlp/toolkit/torch_run.py
caikit_nlp/toolkit/trainer_utils.py
caikit_nlp/toolkit/verbalizer_utils.py
caikit_nlp/toolkit/text_generation/__init__.py
caikit_nlp/toolkit/text_generation/model_run_utils.py
caikit_nlp/toolkit/text_generation/tgis_utils.py
examples/Caikit_Getting_Started.ipynb
examples/compare_local_vs_tgis_models.py
examples/evaluate_model.py
examples/kill-text-generation-launcher.sh
examples/load_and_run_distributed_peft.py
examples/run_fine_tuning.py
examples/run_peft_tuning.py
examples/text-generation-launcher
examples/utils.py
runtime_template/run_with_gateway.sh
scripts/dump_apis.sh
scripts/fmt.sh
scripts/run_local.sh
tests/__init__.py
tests/conftest.py
tests/data_model/test_generation.py
tests/fixtures/__init__.py
tests/fixtures/data_model/sample_objects.py
tests/fixtures/tiny_models/README.md
tests/fixtures/tiny_models/BertForSequenceClassification/config.json
tests/fixtures/tiny_models/BertForSequenceClassification/pytorch_model.bin
tests/fixtures/tiny_models/BertForSequenceClassification/special_tokens_map.json
tests/fixtures/tiny_models/BertForSequenceClassification/tf_model.h5
tests/fixtures/tiny_models/BertForSequenceClassification/tokenizer.json
tests/fixtures/tiny_models/BertForSequenceClassification/tokenizer_config.json
tests/fixtures/tiny_models/BertForSequenceClassification/vocab.txt
tests/fixtures/tiny_models/BloomForCausalLM/config.json
tests/fixtures/tiny_models/BloomForCausalLM/generation_config.json
tests/fixtures/tiny_models/BloomForCausalLM/pytorch_model.bin
tests/fixtures/tiny_models/BloomForCausalLM/special_tokens_map.json
tests/fixtures/tiny_models/BloomForCausalLM/tokenizer.json
tests/fixtures/tiny_models/BloomForCausalLM/tokenizer_config.json
tests/fixtures/tiny_models/T5ForConditionalGeneration/config.json
tests/fixtures/tiny_models/T5ForConditionalGeneration/generation_config.json
tests/fixtures/tiny_models/T5ForConditionalGeneration/pytorch_model.bin
tests/fixtures/tiny_models/T5ForConditionalGeneration/special_tokens_map.json
tests/fixtures/tiny_models/T5ForConditionalGeneration/tokenizer.json
tests/fixtures/tiny_models/T5ForConditionalGeneration/tokenizer_config.json
tests/model_management/__init__.py
tests/model_management/test_tgis_auto_finder.py
tests/modules/__init__.py
tests/modules/text_classification/test_sequence_classification.py
tests/modules/text_embedding/test_embedding.py
tests/modules/text_generation/__init__.py
tests/modules/text_generation/test_peft_config.py
tests/modules/text_generation/test_peft_prompt_tuning.py
tests/modules/text_generation/test_peft_tgis_remote.py
tests/modules/text_generation/test_text_generation_local.py
tests/modules/text_generation/test_text_generation_tgis.py
tests/modules/token_classification/test_filtered_span_classification.py
tests/modules/tokenization/test_regex_sentence_splitter.py
tests/resources/__init__.py
tests/resources/test_pretrained_model.py
tests/toolkit/__init__.py
tests/toolkit/test_data_stream_wrapper.py
tests/toolkit/test_data_type_utils.py
tests/toolkit/test_task_specific_utils.py
tests/toolkit/test_verbalizers.py
tests/toolkit/text_generation/test_model_run_utils.py
tests/toolkit/text_generation/test_tgis_utils.py