Switch to supported models

config.py

@@ -5,10 +5,10 @@ from dotenv import load_dotenv
 load_dotenv()
 
 
-BASE_MODEL = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+BASE_MODEL = "meta-llama/Llama-3.1-8B-Instruct"
 # Other options:
-# MODEL = "meta-llama/Llama-2-7b-chat-hf"
-# MODEL = "openlm-research/open_llama_3b"
+# MODEL = "HuggingFaceTB/SmolLM3-3B"
+# MODEL = "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
 
 # If you finetune the model or change it in any way, save it to huggingface hub, then set MY_MODEL to your model ID. The model ID is in the format "your-username/your-model-name".
 MY_MODEL = None