Spaces:
Sleeping
Sleeping
piyushgrover
commited on
Commit
·
e05d589
1
Parent(s):
7d65571
bug fix
Browse files- app.py +1 -0
- 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
|