[chore] Update placement_policy.py

2024-05-17 13:42:18 +08:00 · 2024-05-17 13:42:18 +08:00 · e57812c672
parent 013690a86b
commit e57812c672
1 changed files with 0 additions and 5 deletions
--- a/colossalai/zero/gemini/placement_policy.py
+++ b/colossalai/zero/gemini/placement_policy.py
@ -45,11 +45,6 @@ class PlacementPolicy(ABC):
        raise NotImplementedError


-import os
-
-rank = int(os.environ["RANK"])
-
-
 class StaticPlacementPolicy(PlacementPolicy):
    def __init__(
        self,