diff --git a/pyproject.toml b/pyproject.toml index 6076feb..fd7fd5d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -48,7 +48,7 @@ train = [ "trl>=0.8.0", "peft>=0.10.0", "deepspeed>=0.14.0", - "wandb>=0.16.0", + "wandb>=0.25.1", "swanlab>=0.3.0", ] dev = ["pytest>=8.0", "ruff>=0.3.0"]