github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[shardformer] register without auto policy

pull/4445/head
klhhhhh 2023-07-18 12:33:12 +08:00 committed by Hongxin Liu
parent f155ae89c4
commit 91850fe984
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/shardformer/policies/auto_policy.py
View File

@ -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"),
} }