Classification prompt tuning#144

Closed

Ssukriti wants to merge 72 commits intocaikit:mainfrom

Ssukriti:classification_prompt_tuning

Commits on Aug 24, 2023

classification prompt tuning wrapper
Ssukriti
committed
:recycle: Refactor trainer logic and move it to resources

gkumbhat
authored and
Ssukriti
committed
:construction: Work in progress causal-lm trainer

gkumbhat
authored and
Ssukriti
committed
:art: Fix linting and formatting

gkumbhat
authored and
Ssukriti
committed
:bug: Fix seq2seq training arguments

gkumbhat
authored and
Ssukriti
committed
:bug: Remove task ids from resource tokenization functions

gkumbhat
authored and
Ssukriti
committed
:white_check_mark: Add cuda device fixture to get around cuda unit testing when available

gkumbhat
authored and
Ssukriti
committed
:art: Fix formatting

gkumbhat
authored and
Ssukriti
committed
:white_check_mark::sparkles: Make fine-tuning work for causal lm models and make tests pass

gkumbhat
authored and
Ssukriti
committed
:wrench: Make review changes and add docstring for arguments

gkumbhat
authored and
Ssukriti
committed
:sparkles: Add support for model.generate right after training by saving and loading the model

gkumbhat
authored and
Ssukriti
committed
:art: Fix formatting and linting

gkumbhat
authored and
Ssukriti
committed
:bug: Fix default verbalizer declaration in fine_tuning module

gkumbhat
authored and
Ssukriti
committed
:wrench: Update parameters for trainer and add random seed

gkumbhat
authored and
Ssukriti
committed
:rewind: Revert back dump_api script changes

gkumbhat
authored and
Ssukriti
committed
Apply suggestions from code review

authored and
Ssukriti
committed
text generation local block
Ssukriti
committed
formatter fixes
Ssukriti
committed
refactoring text geenration blocks
Ssukriti
committed
:truck: Fix interface imports

gkumbhat
authored and
Ssukriti
committed
:art: Fix linting

gkumbhat
authored and
Ssukriti
committed
:coffin: Remove unused code

gkumbhat
authored and
Ssukriti
committed
:bug: Remove default for artifact_path from config in text-gen model load

gkumbhat
authored and
Ssukriti
committed
Update README.md

rawkintrevo
authored and
Ssukriti
committed
Update README.md

authored and
Ssukriti
committed
Update README.md

authored and
Ssukriti
committed
Update README.md

authored and
Ssukriti
committed
Update README.md

rawkintrevo
authored and
Ssukriti
committed
:sparkles: Add fine-tuning support in text generation local module

gkumbhat
authored and
Ssukriti
committed
:white_check_mark: Add test for save and load of fine-tuned model

gkumbhat
authored and
Ssukriti
committed
:art: Fix linting and fine-tuning test statements

gkumbhat
authored and
Ssukriti
committed
:fire: Remove separate fine-tuning module

gkumbhat
authored and
Ssukriti
committed
:recycle: Update text generation tgis and its tests to work with remote TGIS concept

gkumbhat
authored and
Ssukriti
committed
:recycle: Refactor load and save functions to provide portability

gkumbhat
authored and
Ssukriti
committed
:sparkles: Fix fine-tuning script to make it optionally work with remote TGIS for evaluation

gkumbhat
authored and
Ssukriti
committed
:bug::sparkles: Fix dtype issue and add truncate input tokens param

gkumbhat
authored and
Ssukriti
committed
Apply suggestions from code review

authored and
Ssukriti
committed
:bug: Remove temp hack for handling truncation in TGIS long

gkumbhat
authored and
Ssukriti
committed
:bug: Fix token truncation input condition

gkumbhat
authored and
Ssukriti
committed
:art: Move artifact path to use os path join

gkumbhat
authored and
Ssukriti
committed
:bug: Fix training arguments for service generation to work correctly

gkumbhat
authored and
Ssukriti
committed
TgisAutoFinder: Initial implementation of TGISAutoFinder

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Add remote_tgis finder to runtime_config

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Remove connection test

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Bug fixes in value checks

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Add the ability to give a specific TGIS backend to use from the finder

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Skip peft training tests on arm processors

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Happy path unit tests for the auto finder

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Bump caikit to allow 0.15

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Don't require instance_name in finder constructor

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Bump caikit-tgis-backend for connection testing

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Add model_id in mock TGISConnection

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Handle bad tgis connections

gabe-l-hart
authored and
Ssukriti
committed
TgisAutoFinder: Linting fixes

gabe-l-hart
authored and
Ssukriti
committed
fixed simple typo (#128)

olson-ibm
authored and
Ssukriti
committed
:technologist: Update fine-tuning example to allow local save of the model and flexible target seq len

gkumbhat
authored and
Ssukriti
committed
:art: fix formatting

gkumbhat
authored and
Ssukriti
committed
:arrow_up: Update caikit[runtime-grpc,runtime-http] requirement

dependabot[bot]
authored and
Ssukriti
committed
⬆️ upgrade caikit

tharapalanivel
authored and
Ssukriti
committed
🚧 Rebase on caikit interfaces

tharapalanivel
authored and
Ssukriti
committed
⬆️ Bump caikit-tgis-backend

tharapalanivel
authored and
Ssukriti
committed
Update CODEOWNERS

gkumbhat
authored and
Ssukriti
committed
add unit tests
Ssukriti
committed
load and save tests
Ssukriti
committed
classification prompt tuning block
Ssukriti
committed
Closes #134

jolson-ibm
authored and
Ssukriti
committed
Closes #134

jolson-ibm
authored and
Ssukriti
committed
Closes #134

jolson-ibm
authored and
Ssukriti
committed
:construction: Debug streaming API generation

gkumbhat
authored and
Ssukriti
committed
:package: Update caikit

gkumbhat
authored and
Ssukriti
committed
:memo: Add comment explaining wip comment for prompt tuning

gkumbhat
authored and
Ssukriti
committed
:arrow_up: Update caikit[runtime-grpc,runtime-http] requirement

dependabot[bot]
authored and
Ssukriti
committed