wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-26 14:00:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor

Browse Source
This commit is contained in:
Sourab Mangrulkar
2022-11-29 18:40:41 +05:30
parent 7c21cb456c
commit 91deee81e2
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pet/__init__.py
+1 -1
View File
@@ -5,7 +5,7 @@
__version__ = "0.1.0.dev0"
from .pet_model import PETModel, PETModelForCausalLM, PETModelForSeq2SeqLM, PETModelForSequenceClassification
from .task_mapping import MODEL_TYPE_TO_PROMPT_MODEL_MAPPING
from .mapping import MODEL_TYPE_TO_PET_MODEL_MAPPING, PET_TYPE_TO_CONFIG_MAPPING, get_pet_config, get_pet_model
from .tuners import (
PrefixEncoder,
PrefixTuningConfig,