KWRegan commited on
Commit
b61c367
·
1 Parent(s): b1af265

Add application file

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- # 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,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
- # 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
 
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