Update __init__.py

2026-09-11 12:30:16 +08:00 · 2022-11-26 18:59:50 +05:30
parent 6013c83dbc
commit d5c0c7e32c
1 changed files with 2 additions and 0 deletions
@@ -7,6 +7,8 @@ __version__ = "0.1.0.dev0"
 from .pet_model import (
    PETModel,
    PETModelForSequenceClassification,
+    PETModelForCausalLM,
+    PETModelForSeq2SeqLM,
    PromptEncoderType,
 )
 from .tuners import (