Skip to content

Commit 3e34e01

Browse files
Add DeepSeek-R1-Distill-Qwen-14B as a supported model in tt-transformers
1 parent 1349243 commit 3e34e01

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

models/tt_transformers/tt/model_config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -548,6 +548,7 @@ def __init__(
548548
"DeepSeek-R1-Distill-Llama-70B": {"N150": None, "N300": None, "T3K": 32, "TG": 128, "P150x4": 128},
549549
"Qwen2.5-7B": {"N150": 4, "N300": 64, "T3K": 128, "TG": 128, "P150x4": 128},
550550
"Qwen2.5-72B": {"N150": None, "N300": None, "T3K": 32, "TG": 128, "P150x4": 128},
551+
"DeepSeek-R1-Distill-Qwen-14B": {"N150": 4, "N300": 64, "T3K": 128, "TG": 128, "P150x4": 128},
551552
"Phi-3.5-mini-instruct": {"N150": 128, "N300": 128, "T3K": 128, "TG": 128, "P150x4": 128},
552553
"QwQ-32B": {"N150": None, "N300": None, "T3K": 64, "TG": 128, "P150x4": 128},
553554
"Qwen3-32B": {"N150": None, "N300": None, "T3K": 64, "TG": 128, "P150x4": 128},

0 commit comments

Comments
 (0)