caikit / caikit/caikit-nlp

Add support for LoRA in prompt tuning module (Investigate)

Open
#136 4 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
12
Forks
55
PR merge metrics
No merged PRs in 30d

Description

## Is your feature request related to a problem? Please describe.

Currently we only support vanilla prompt tuning and multi-task prompt tuning from PEFT library. LoRA has shown to give good performance on variety of datasets. This story is to investigate support for LoRA into the module.

Contributor guide

Open the contributing guide

Research direction

Start by locating the prompt tuning module and its existing integrations with the PEFT library for vanilla and multi-task prompt tuning. Investigate what LoRA support would require and document a concrete scope, affected entry points, and validation approach; the issue does not name files or tests, so completion criteria remain to be established.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.