Vivek16 commited on
Commit
e2451da
·
verified ·
1 Parent(s): 6104606

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -3,11 +3,12 @@ import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  from peft import PeftModel
5
 
 
 
6
  # --- Configuration ---
7
- # ⚠️ REPLACE 'YOUR_HF_USERNAME' with your actual username
8
  BASE_MODEL_ID = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
9
- ADAPTER_MODEL_ID = "YOUR_HF_USERNAME/Root_Math-TinyLlama-CPU"
10
-
11
  # Define the instruction template used during fine-tuning (Step 5)
12
  INSTRUCTION_TEMPLATE = "<|system|>\nSolve the following math problem:</s>\n<|user|>\n{}</s>\n<|assistant|>"
13
 
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  from peft import PeftModel
5
 
6
+ # app.py snippet
7
+
8
  # --- Configuration ---
9
+ # ⚠️ YOUR USERNAME IS NOW SET
10
  BASE_MODEL_ID = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
11
+ ADAPTER_MODEL_ID = "Vivek16/Root_Math-TinyLlama-CPU" # <-- Replace 'Vivek16' with your actual username
 
12
  # Define the instruction template used during fine-tuning (Step 5)
13
  INSTRUCTION_TEMPLATE = "<|system|>\nSolve the following math problem:</s>\n<|user|>\n{}</s>\n<|assistant|>"
14