github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-01 01:06:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

[shardformer] register without auto policy

Browse Source
This commit is contained in:
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
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
"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"),
}