Update __init__.py (huggingface#1602)

burtenshaw · Apr 30, 2024 · d88ec14 · d88ec14
1 parent 6c18e40
commit d88ec14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trl/commands/__init__.py b/trl/commands/__init__.py
@@ -20,7 +20,7 @@
 
 
 _import_structure = {
-    "cli_utils": ["SftArgumentParser", "init_zero_verbose", "DPOScriptArguments", "TrlParser", "YamlConfigParser"],
+    "cli_utils": ["SFTScriptArguments", "init_zero_verbose", "DPOScriptArguments", "TrlParser", "YamlConfigParser"],
 }