generated from caikit/caikit-template
-
Notifications
You must be signed in to change notification settings - Fork 55
Open
Milestone
Description
Description
After refactoring #163 PT to take out prompt tuning related configuration to separate module. We want to add support for prompt tuning (specifically LoRA) in fine-tuning module, i.e text_generation. Changes:
- Add LoRA configuration support in the toolkit functionality created in Refactor peft module to take out common peft config functionality #163
- Add LoRA configuration and training from text-generation module (train).
- Expose parameter required for LoRA configuration via
.trainfunction - Add support for saving LoRA models with "merged weights". This is to be done in
.trainfunction itself, that way, themodelthat we configure to__init__function will look like any other transformers model.
Acceptance Criteria
- Unit tests cover new/changed code
- Examples build against new/changed code
- README updated here
- Example for training lora added in example script
Metadata
Metadata
Assignees
Labels
No labels
Type
Projects
Status
Ready for Review