From 5b377e914134c34c8f20069afaf9a9fe1e39a42a Mon Sep 17 00:00:00 2001
From: vatsal <31882705+vatsalkshah@users.noreply.github.com>
Date: Tue, 28 Jan 2025 22:45:45 +0530
Subject: [PATCH] chore: add deepseek distilled models to constant.py (#70)

* chore: add deepseek distilled models to constant.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 src/core/constant.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/core/constant.py b/src/core/constant.py
index 317dd14..fa7a0fc 100644
--- a/src/core/constant.py
+++ b/src/core/constant.py
@@ -97,4 +97,11 @@
     "microsoft/Phi-3.5-mini-instruct",
     "microsoft/Phi-3-mini-4k-instruct",
     "microsoft/Phi-3-medium-4k-instruct",
+    # deepseek
+    "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
+    "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B",
+    "deepseek-ai/DeepSeek-R1-Distill-Llama-8B",
+    "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B",
+    "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
+    "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
 ]