Fix: point PeftModel.from_pretrained at the correct adapter subfolder (rl/seed_42/best)
Browse files
app.py
CHANGED
|
@@ -14,6 +14,11 @@ from peft import PeftModel
|
|
| 14 |
|
| 15 |
BASE_MODEL = "Qwen/Qwen2.5-Coder-14B-Instruct"
|
| 16 |
ADAPTER = "adelelsayed1991/fhirsql-reasoning-sql-adapters"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
ABSTENTION_TOKEN = "UNANSWERABLE"
|
| 18 |
MAX_NEW_TOKENS = 512
|
| 19 |
|
|
@@ -44,7 +49,7 @@ SYSTEM_PROMPT = SYSTEM_PROMPT_TEMPLATE.format(
|
|
| 44 |
|
| 45 |
tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
|
| 46 |
base_model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, torch_dtype="auto")
|
| 47 |
-
model = PeftModel.from_pretrained(base_model, ADAPTER)
|
| 48 |
model.to("cuda")
|
| 49 |
model.eval()
|
| 50 |
|
|
|
|
| 14 |
|
| 15 |
BASE_MODEL = "Qwen/Qwen2.5-Coder-14B-Instruct"
|
| 16 |
ADAPTER = "adelelsayed1991/fhirsql-reasoning-sql-adapters"
|
| 17 |
+
# The adapters repo holds six checkpoints (sft/rl x seed_42/43/44), each in
|
| 18 |
+
# its own subfolder rather than at the repo root. rl/seed_42/best is served
|
| 19 |
+
# here as the project's most advanced trained artifact (SFT+RL); change
|
| 20 |
+
# this to any of sft|rl/seed_{42,43,44}/best to serve a different one.
|
| 21 |
+
ADAPTER_SUBFOLDER = "rl/seed_42/best"
|
| 22 |
ABSTENTION_TOKEN = "UNANSWERABLE"
|
| 23 |
MAX_NEW_TOKENS = 512
|
| 24 |
|
|
|
|
| 49 |
|
| 50 |
tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
|
| 51 |
base_model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, torch_dtype="auto")
|
| 52 |
+
model = PeftModel.from_pretrained(base_model, ADAPTER, subfolder=ADAPTER_SUBFOLDER)
|
| 53 |
model.to("cuda")
|
| 54 |
model.eval()
|
| 55 |
|