Add application file
Browse files
app.py
CHANGED
@@ -216,8 +216,8 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
|
|
216 |
# %% Constants and Configuration
|
217 |
# MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
|
218 |
# MODEL_NAME = "google/gemma-3-4b-it"
|
219 |
-
|
220 |
-
MODEL_NAME = "Qwen/Qwen3-4B"
|
221 |
# MODEL_NAME = "Qwen/Qwen3-0.6B"
|
222 |
# MODEL_NAME = "microsoft/Phi-4-mini-instruct"
|
223 |
#MODEL_NAME = "meta-llama/Llama-2-7b-hf"
|
@@ -286,8 +286,8 @@ def main(INPUT_FILE, OUTPUT_FILE, PROMPT_ID, PROMPT_TOPIC, MULTI_PV, NUM_WORDS,
|
|
286 |
# LegalNumberOfMove_str = input(" Enter Max Number of moves [Default: 10]: ") or "10"
|
287 |
LegalNumberOfMove = NUM_WORDS # int(LegalNumberOfMove_str)
|
288 |
# EngineID = f"DeepSeek R1 1.5B Qwen-Distil Greedy ({DEVICE.upper()})" # Updated EngineID
|
289 |
-
|
290 |
-
EngineID = f"Qwen/Qwen3-4B"
|
291 |
# EngineID = f"Qwen/Qwen3-0.6B"
|
292 |
# EngineID = f"Gemma-3-4b-it ({DEVICE.upper()})" # Indicate CPU in EngineID
|
293 |
Depth = 1
|
|
|
216 |
# %% Constants and Configuration
|
217 |
# MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
|
218 |
# MODEL_NAME = "google/gemma-3-4b-it"
|
219 |
+
MODEL_NAME = "Qwen/Qwen3-1.7B"
|
220 |
+
# MODEL_NAME = "Qwen/Qwen3-4B"
|
221 |
# MODEL_NAME = "Qwen/Qwen3-0.6B"
|
222 |
# MODEL_NAME = "microsoft/Phi-4-mini-instruct"
|
223 |
#MODEL_NAME = "meta-llama/Llama-2-7b-hf"
|
|
|
286 |
# LegalNumberOfMove_str = input(" Enter Max Number of moves [Default: 10]: ") or "10"
|
287 |
LegalNumberOfMove = NUM_WORDS # int(LegalNumberOfMove_str)
|
288 |
# EngineID = f"DeepSeek R1 1.5B Qwen-Distil Greedy ({DEVICE.upper()})" # Updated EngineID
|
289 |
+
EngineID = f"Qwen/Qwen3-1.7B"
|
290 |
+
# EngineID = f"Qwen/Qwen3-4B"
|
291 |
# EngineID = f"Qwen/Qwen3-0.6B"
|
292 |
# EngineID = f"Gemma-3-4b-it ({DEVICE.upper()})" # Indicate CPU in EngineID
|
293 |
Depth = 1
|