EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Format code (#118)

Browse Source
This commit is contained in:
Lianmin Zheng
2024-01-29 17:08:12 -08:00
committed by GitHub
parent 6b0af2853c
commit 4ea92f8307
6 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/sglang/srt/models/llama2.py
View File

@@ -319,4 +319,5 @@ class LlamaForCausalLM(nn.Module):
weight_loader = getattr(param, "weight_loader", default_weight_loader)
weight_loader(param, loaded_weight)
EntryClass = LlamaForCausalLM

1
python/sglang/srt/models/llava.py
View File

@@ -331,4 +331,5 @@ def monkey_path_clip_vision_embed_forward():
clip_vision_embed_forward,
)
EntryClass = LlavaLlamaForCausalLM

1
python/sglang/srt/models/mixtral.py
View File

@@ -377,4 +377,5 @@ class MixtralForCausalLM(nn.Module):
weight_loader = getattr(param, "weight_loader", default_weight_loader)
weight_loader(param, loaded_weight)
EntryClass = MixtralForCausalLM

1
python/sglang/srt/models/qwen.py
View File

@@ -259,4 +259,5 @@ class QWenLMHeadModel(nn.Module):
weight_loader = getattr(param, "weight_loader", default_weight_loader)
weight_loader(param, loaded_weight)
EntryClass = QWenLMHeadModel

1
python/sglang/srt/models/qwen2.py
View File

@@ -315,4 +315,5 @@ class Qwen2ForCausalLM(nn.Module):
weight_loader = getattr(param, "weight_loader", default_weight_loader)
weight_loader(param, loaded_weight)
EntryClass = Qwen2ForCausalLM