seq2seq (#309)

huggingface · Oct 20, 2023 · 78491a6 · 78491a6
1 parent 6559d99
commit 78491a6
Show file tree

Hide file tree

Showing 8 changed files with 825 additions and 2 deletions.
diff --git a/src/autotrain/cli/autotrain.py b/src/autotrain/cli/autotrain.py
@@ -6,6 +6,7 @@
 from .run_dreambooth import RunAutoTrainDreamboothCommand
 from .run_image_classification import RunAutoTrainImageClassificationCommand
 from .run_llm import RunAutoTrainLLMCommand
+from .run_seq2seq import RunAutoTrainSeq2SeqCommand
 from .run_setup import RunSetupCommand
 from .run_spacerunner import RunAutoTrainSpaceRunnerCommand
 from .run_tabular import RunAutoTrainTabularCommand
@@ -31,6 +32,7 @@ def main():
     RunAutoTrainImageClassificationCommand.register_subcommand(commands_parser)
     RunAutoTrainTabularCommand.register_subcommand(commands_parser)
     RunAutoTrainSpaceRunnerCommand.register_subcommand(commands_parser)
+    RunAutoTrainSeq2SeqCommand.register_subcommand(commands_parser)
 
     args = parser.parse_args()