EITD commited on
Commit
be3b008
·
1 Parent(s): 0225df5
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,10 +11,10 @@ For more information on `huggingface_hub` Inference API support, please check th
11
 
12
  from peft import AutoPeftModelForCausalLM
13
  model = AutoPeftModelForCausalLM.from_pretrained(
14
- "orpo_model", # YOUR MODEL YOU USED FOR TRAINING
15
  load_in_4bit = False,
16
  )
17
- tokenizer = AutoTokenizer.from_pretrained("orpo_model")
18
 
19
  def respond(
20
  message,
 
11
 
12
  from peft import AutoPeftModelForCausalLM
13
  model = AutoPeftModelForCausalLM.from_pretrained(
14
+ "EITD/orpo_model", # YOUR MODEL YOU USED FOR TRAINING
15
  load_in_4bit = False,
16
  )
17
+ tokenizer = AutoTokenizer.from_pretrained("EITD/orpo_model")
18
 
19
  def respond(
20
  message,