3

app.py

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig,
 from peft import LoraConfig, get_peft_model, prepare_model_for_kbit_training
 
 # Constants
-MODEL_NAME = "deepseek-math"
+MODEL_NAME = "deepseek-ai/deepseek-math-7b-base"
 SAVE_PATH = "finetuned_deepseek_math"
 
 @st.cache_resource(show_spinner=False)