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

rename chatglm to chatglm2 (#4484)

Browse Source
This commit is contained in:
Jianghai 2023-08-22 14:13:31 +08:00 committed by GitHub
parent 1c7df566e2
commit 5545114fd8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
colossalai/shardformer/modeling/chatglm.py → colossalai/shardformer/modeling/chatglm2.py
View File

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

@ -125,9 +125,9 @@ _POLICY_LIST = {
# ChatGLM # ChatGLM
"colossalai.shardformer.modeling.chatglm2_6b.modeling_chatglm.ChatGLMModel": "colossalai.shardformer.modeling.chatglm2_6b.modeling_chatglm.ChatGLMModel":
PolicyLocation(file_name="chatglm", class_name="ChatGLMModelPolicy"), PolicyLocation(file_name="chatglm2", class_name="ChatGLMModelPolicy"),
"colossalai.shardformer.modeling.chatglm2_6b.modeling_chatglm.ChatGLMForConditionalGeneration": "colossalai.shardformer.modeling.chatglm2_6b.modeling_chatglm.ChatGLMForConditionalGeneration":
PolicyLocation(file_name="chatglm", class_name="ChatGLMForConditionalGenerationPolicy"), PolicyLocation(file_name="chatglm2", class_name="ChatGLMForConditionalGenerationPolicy"),
} }

4
colossalai/shardformer/policies/chatglm.py → colossalai/shardformer/policies/chatglm2.py
View File

@ -7,7 +7,7 @@ from transformers.modeling_outputs import BaseModelOutputWithPast
import colossalai.shardformer.layer as col_nn import colossalai.shardformer.layer as col_nn
from colossalai.pipeline.stage_manager import PipelineStageManager from colossalai.pipeline.stage_manager import PipelineStageManager
from colossalai.shardformer.modeling.chatglm import ChatGLMPipelineForwards from colossalai.shardformer.modeling.chatglm2 import ChatGLMPipelineForwards
from colossalai.shardformer.modeling.chatglm2_6b.configuration_chatglm import ChatGLMConfig from colossalai.shardformer.modeling.chatglm2_6b.configuration_chatglm import ChatGLMConfig
from colossalai.shardformer.modeling.chatglm2_6b.modeling_chatglm import ( from colossalai.shardformer.modeling.chatglm2_6b.modeling_chatglm import (
ChatGLMForConditionalGeneration, ChatGLMForConditionalGeneration,
@ -15,7 +15,7 @@ from colossalai.shardformer.modeling.chatglm2_6b.modeling_chatglm import (
GLMBlock, GLMBlock,
) )
from ..modeling.chatglm import get_flash_core_attention_forward, get_jit_fused_glm_block_forward from ..modeling.chatglm2 import get_flash_core_attention_forward, get_jit_fused_glm_block_forward
from ..modeling.jit import get_jit_fused_dropout_add_func from ..modeling.jit import get_jit_fused_dropout_add_func
from .base_policy import ModulePolicyDescription, Policy, SubModuleReplacementDescription from .base_policy import ModulePolicyDescription, Policy, SubModuleReplacementDescription

2
tests/kit/model_zoo/transformers/__init__.py
View File

@ -2,7 +2,7 @@ from .albert import *
from .bert import * from .bert import *
from .blip2 import * from .blip2 import *
from .bloom import * from .bloom import *
from .chatglm import * from .chatglm2 import *
from .gpt import * from .gpt import *
from .llama import * from .llama import *
from .opt import * from .opt import *

0
tests/kit/model_zoo/transformers/chatglm.py → tests/kit/model_zoo/transformers/chatglm2.py
View File

0
tests/test_shardformer/test_model/test_shard_chatglm.py → tests/test_shardformer/test_model/test_shard_chatglm2.py
View File