Files
peft/src/pet/__init__.py
T
2022-11-26 18:59:50 +05:30

21 lines
499 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,
PETModelForSequenceClassification,
PETModelForCausalLM,
PETModelForSeq2SeqLM,
PromptEncoderType,
)
from .tuners import (
PrefixEncoder,
PromptEmbedding,
PromptEncoder,
PromptEncoderReparameterizationType,
PromptTuningInit,
)