adelelsayed1991 commited on
Commit
64c1aa0
·
verified ·
1 Parent(s): 3434a32

Fix: point PeftModel.from_pretrained at the correct adapter subfolder (rl/seed_42/best)

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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