cleanup

Signed-off-by: Mayank Mishra <[email protected]>
IBM · Oct 7, 2024 · 3b31f56 · 3b31f56
1 parent 5623a28
commit 3b31f56
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dolomite_engine/hf_models/modeling_utils_TP/TP.py b/dolomite_engine/hf_models/modeling_utils_TP/TP.py
@@ -97,6 +97,7 @@ def modify_state_dict_to_dtensor_dict(module: nn.Module, state_dict: dict, prefi
             device_mesh = param.device_mesh
             placements = param.placements
             result[key] = DTensor.from_local(tensor, device_mesh=device_mesh, placements=placements)
+
     return result