Skip to content

GitLab

Commit 91850fe9 authored by klhhhhh's avatar klhhhhh Committed by Hongxin Liu
Browse files

[shardformer] register without auto policy

parent f155ae89
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
colossalai/shardformer/policies/auto_policy.py
View file @ 91850fe9
...@@ -116,9 +116,6 @@ _POLICY_LIST = { ...@@ -116,9 +116,6 @@ _POLICY_LIST = {
# Sam # Sam
"transformers.models.sam.modeling_sam.SamModel": "transformers.models.sam.modeling_sam.SamModel":
PolicyLocation(file_name="sam", class_name="SamModelPolicy"), 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"),
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment