Update __init__.py

This commit is contained in:
Sourab Mangrulkar
2022-11-29 18:20:23 +05:30
parent a92a7876e0
commit 7c21cb456c
+1 -8
View File
@@ -4,14 +4,7 @@
__version__ = "0.1.0.dev0"
from .pet_model import (
PETModel,
PETModelForCausalLM,
PETModelForSeq2SeqLM,
PETModelForSequenceClassification,
PETPluginBase,
PETType,
)
from .pet_model import PETModel, PETModelForCausalLM, PETModelForSeq2SeqLM, PETModelForSequenceClassification
from .task_mapping import MODEL_TYPE_TO_PROMPT_MODEL_MAPPING
from .tuners import (
PrefixEncoder,