[shardformer] register without auto policy

91850fe9 · klhhhhh · Hongxin Liu · f155ae89 · 91850fe9
Commit 91850fe9 authored Jul 18, 2023 by klhhhhh Committed by Hongxin Liu Aug 15, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 3 deletions

colossalai/shardformer/policies/auto_policy.py colossalai/shardformer/policies/auto_policy.py +0 -3

No files found.
--- a/colossalai/shardformer/policies/auto_policy.py
+++ b/colossalai/shardformer/policies/auto_policy.py
@@ -116,9 +116,6 @@ _POLICY_LIST = {
    # Sam
    "transformers.models.sam.modeling_sam.SamModel":
        PolicyLocation(file_name="sam", class_name="SamModelPolicy"),
-    # ChatGLM
-    "tests.kit.model_zoo.transformers.chatglm2_6b.modeling_chatglm.ChatGLMModel":
-        PolicyLocation(file_name="chatglm", class_name="ChatGLMModelPolicy"),
 }