Merge pull request #6457 from youkaichao/module-run

[misc] enable module run Former-commit-id: 813881a5d13dd1d5a526a85d41032196e0d46f04
2024-12-26 23:41:37 +08:00
parent b5fa1044b8 552816e04b
commit 5769a553d2
1 changed files with 3 additions and 0 deletions
--- a/src/llamafactory/cli.py
+++ b/src/llamafactory/cli.py
@@ -120,3 +120,6 @@ def main():
        print(USAGE)
    else:
        raise NotImplementedError(f"Unknown command: {command}.")
+
+if __name__ == "__main__":
+    main()