File tree Expand file tree Collapse file tree 1 file changed +17
-0
lines changed
Expand file tree Collapse file tree 1 file changed +17
-0
lines changed Original file line number Diff line number Diff line change @@ -246,6 +246,7 @@ class MODEL_ARCH(IntEnum):
246246 QWEN2 = auto ()
247247 QWEN2MOE = auto ()
248248 QWEN2VL = auto ()
249+ QWEN3 = auto ()
249250 PHI2 = auto ()
250251 PHI3 = auto ()
251252 PHIMOE = auto ()
@@ -448,6 +449,7 @@ class MODEL_TENSOR(IntEnum):
448449 MODEL_ARCH .QWEN2 : "qwen2" ,
449450 MODEL_ARCH .QWEN2MOE : "qwen2moe" ,
450451 MODEL_ARCH .QWEN2VL : "qwen2vl" ,
452+ MODEL_ARCH .QWEN3 : "qwen3" ,
451453 MODEL_ARCH .PHI2 : "phi2" ,
452454 MODEL_ARCH .PHI3 : "phi3" ,
453455 MODEL_ARCH .PHIMOE : "phimoe" ,
@@ -923,6 +925,21 @@ class MODEL_TENSOR(IntEnum):
923925 MODEL_TENSOR .FFN_DOWN_SHEXP ,
924926 MODEL_TENSOR .FFN_UP_SHEXP ,
925927 ],
928+ MODEL_ARCH .QWEN3 : [
929+ MODEL_TENSOR .TOKEN_EMBD ,
930+ MODEL_TENSOR .OUTPUT_NORM ,
931+ MODEL_TENSOR .OUTPUT ,
932+ MODEL_TENSOR .ROPE_FREQS ,
933+ MODEL_TENSOR .ATTN_NORM ,
934+ MODEL_TENSOR .ATTN_Q ,
935+ MODEL_TENSOR .ATTN_K ,
936+ MODEL_TENSOR .ATTN_V ,
937+ MODEL_TENSOR .ATTN_OUT ,
938+ MODEL_TENSOR .FFN_NORM ,
939+ MODEL_TENSOR .FFN_GATE ,
940+ MODEL_TENSOR .FFN_DOWN ,
941+ MODEL_TENSOR .FFN_UP ,
942+ ],
926943 MODEL_ARCH .PLAMO : [
927944 MODEL_TENSOR .TOKEN_EMBD ,
928945 MODEL_TENSOR .OUTPUT_NORM ,
You can’t perform that action at this time.
0 commit comments