Update __init__.py

2025-02-21 11:13:15 -05:00
parent 3dd5c6f8ec
commit f68aedd1f8
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/config/models/input/v0_4_1/init.py
+++ b/src/axolotl/utils/config/models/input/v0_4_1/init.py
@@ -1377,7 +1377,7 @@ class AxolotlInputConfig(
    @classmethod
    def check_fsdp_tp(cls, data):
        if data.get("fsdp") and data.get("tensor_parallel"):
-            raise ValueError("FSDP is not compatible with tensor parallelism")
+            raise ValueError("FSDP with tensor parallelism is not supported yet.")
        return data

    @model_validator(mode="after")