mirror of
https://github.com/wassname/peft.git
synced 2026-09-25 13:50:20 +08:00
38 lines
980 B
Python
38 lines
980 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 .mapping import MODEL_TYPE_TO_PET_MODEL_MAPPING, PET_TYPE_TO_CONFIG_MAPPING, get_pet_config, get_pet_model
|
|
from .pet_model import (
|
|
PETModel,
|
|
PETModelForCausalLM,
|
|
PETModelForSeq2SeqLM,
|
|
PETModelForSequenceClassification,
|
|
PETModelForTokenClassification,
|
|
)
|
|
from .tuners import (
|
|
LoRAConfig,
|
|
LoRAModel,
|
|
PrefixEncoder,
|
|
PrefixTuningConfig,
|
|
PromptEmbedding,
|
|
PromptEncoder,
|
|
PromptEncoderConfig,
|
|
PromptEncoderReparameterizationType,
|
|
PromptTuningConfig,
|
|
PromptTuningInit,
|
|
)
|
|
from .utils import (
|
|
PETConfig,
|
|
PETType,
|
|
PromptLearningConfig,
|
|
TaskType,
|
|
bloom_model_postprocess_past_key_value,
|
|
get_pet_model_state_dict,
|
|
pet_model_load_and_dispatch,
|
|
set_pet_model_state_dict,
|
|
shift_tokens_right,
|
|
)
|