diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py index 4ed43642..9f605532 100644 --- a/src/llamafactory/extras/constants.py +++ b/src/llamafactory/extras/constants.py @@ -87,6 +87,8 @@ STAGES_USE_PAIR_DATA = {"rm", "dpo"} SUPPORTED_CLASS_FOR_S2ATTN = {"llama"} +SWANLAB_CONFIG = "swanlab_public_config.json" + VIDEO_PLACEHOLDER = os.environ.get("VIDEO_PLACEHOLDER", "