support report custom args

Former-commit-id: d41254c40a1c5cacf9377096adb27efa9bdb79ea
2024-12-19 14:57:09 +00:00
parent adff887659
commit a897d46049
20 changed files with 164 additions and 124 deletions
--- a/setup.py
+++ b/setup.py
@@ -61,6 +61,7 @@ extra_require = {
    "qwen": ["transformers_stream_generator"],
    "modelscope": ["modelscope"],
    "openmind": ["openmind"],
+    "swanlab": ["swanlab"],
    "dev": ["pre-commit", "ruff", "pytest"],
 }