mirror of
https://github.com/wassname/peft.git
synced 2026-09-25 13:50:20 +08:00
20 lines
710 B
Python
20 lines
710 B
Python
# flake8: noqa
|
|
# There's no way to ignore "F401 '...' imported but unused" warnings in this
|
|
# module, but to preserve other warnings. So, don't check this module at all.
|
|
|
|
__version__ = "0.1.0.dev0"
|
|
|
|
from .pet_model import PETModel, PETModelForCausalLM, PETModelForSeq2SeqLM, PETModelForSequenceClassification
|
|
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,
|
|
PromptEmbedding,
|
|
PromptEncoder,
|
|
PromptEncoderConfig,
|
|
PromptEncoderReparameterizationType,
|
|
PromptTuningConfig,
|
|
PromptTuningInit,
|
|
)
|
|
from .utils import PETConfig, PETType, PromptLearningConfig, TaskType
|