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

Fix commandr import; format code

Browse Source
This commit is contained in:
Lianmin Zheng
2024-04-16 18:10:12 +00:00
parent db611066ad
commit 65501a9cf1
4 changed files with 83 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/sglang/srt/models/dbrx.py
View File

@@ -171,7 +171,6 @@ class DbrxExperts(nn.Module):
class DbrxAttention(nn.Module):
def __init__(
self,
config: DbrxConfig,
@@ -251,7 +250,6 @@ class DbrxAttention(nn.Module):
class DbrxFusedNormAttention(nn.Module):
def __init__(
self,
config: DbrxConfig,
@@ -284,7 +282,6 @@ class DbrxFusedNormAttention(nn.Module):
class DbrxBlock(nn.Module):
def __init__(
self,
config: DbrxConfig,
@@ -312,7 +309,6 @@ class DbrxBlock(nn.Module):
class DbrxModel(nn.Module):
def __init__(
self,
config: DbrxConfig,
@@ -351,7 +347,6 @@ class DbrxModel(nn.Module):
class DbrxForCausalLM(nn.Module):
def __init__(
self,
config: DbrxConfig,