justinj92 commited on
Commit
f4cb054
·
verified ·
1 Parent(s): c662eeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -17,15 +17,13 @@ subprocess.run(
17
  shell=True,
18
  )
19
 
20
- token = os.environ["HF_TOKEN"]
21
 
22
 
23
  model = AutoModelForCausalLM.from_pretrained(
24
- "microsoft/Phi-3-mini-128k-instruct",
25
- token=token,
26
  trust_remote_code=True,
27
  )
28
- tok = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct", token=token)
29
  terminators = [
30
  tok.eos_token_id,
31
  ]
 
17
  shell=True,
18
  )
19
 
 
20
 
21
 
22
  model = AutoModelForCausalLM.from_pretrained(
23
+ "justinj92/phi3-orpo",
 
24
  trust_remote_code=True,
25
  )
26
+ tok = AutoTokenizer.from_pretrained("justinj92/phi3-orpo", token=token)
27
  terminators = [
28
  tok.eos_token_id,
29
  ]