piyushgrover commited on
Commit
e05d589
·
1 Parent(s): 7d65571
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -19,6 +19,7 @@ new_model = "piyushgrover/phi-2-qlora-adapter-custom"
19
  model = PeftModel.from_pretrained(base_model, new_model)
20
  model = model.merge_and_unload()
21
 
 
22
  # Reload tokenizer to save it
23
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
24
  tokenizer.pad_token = tokenizer.eos_token
 
19
  model = PeftModel.from_pretrained(base_model, new_model)
20
  model = model.merge_and_unload()
21
 
22
+ model.eval()
23
  # Reload tokenizer to save it
24
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
25
  tokenizer.pad_token = tokenizer.eos_token
requirements.txt CHANGED
@@ -10,3 +10,4 @@ accelerate
10
  git+https://github.com/huggingface/peft.git
11
  bitsandbytes
12
  einops
 
 
10
  git+https://github.com/huggingface/peft.git
11
  bitsandbytes
12
  einops
13
+ scipy