Shriti09 commited on
Commit
5044361
·
verified ·
1 Parent(s): 95775e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
  # Base model and adapter paths (updated for Hugging Face repo)
10
  base_model_name = "microsoft/phi-2" # Pull from HF Hub directly
11
- adapter_path = "Shriti09/phi2-qlora-adapter" # Update with your Hugging Face repo path
12
 
13
  print("🔧 Loading base model...")
14
  base_model = AutoModelForCausalLM.from_pretrained(
 
8
 
9
  # Base model and adapter paths (updated for Hugging Face repo)
10
  base_model_name = "microsoft/phi-2" # Pull from HF Hub directly
11
+ adapter_path = "Shriti09/Microsoft-Phi-QLora" # Update with your Hugging Face repo path
12
 
13
  print("🔧 Loading base model...")
14
  base_model = AutoModelForCausalLM.from_pretrained(