Fix?

2026-01-20 14:44:40 +00:00
parent 11db77c2c3
commit 4dcdca7b71
1 changed files with 1 additions and 1 deletions
@@ -317,7 +317,7 @@ def tensor_auto_parallel(
    elif hasattr(model, "shard"):
        try:
            model.shard(group)  # type: ignore
-            return patch_distributed_model(model)
+            return model
        except (AttributeError, TypeError, NameError):
            pass