EITD commited on
Commit
4edc451
·
1 Parent(s): aaec37b

update model

Browse files
Files changed (2) hide show
  1. .gitignore +3 -0
  2. app.py +2 -3
.gitignore ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ lora_model*
2
+ Copy*
3
+ unsloth*
app.py CHANGED
@@ -7,10 +7,9 @@ For more information on `huggingface_hub` Inference API support, please check th
7
  # client = InferenceClient("EITD/lora_model", token=os.getenv("HF_TOKEN"))
8
 
9
  model = AutoPeftModelForCausalLM.from_pretrained(
10
- "EITD/lora_model_1", # YOUR MODEL YOU USED FOR TRAINING
11
- load_in_4bit = False,
12
  )
13
- tokenizer = AutoTokenizer.from_pretrained("EITD/lora_model_1")
14
 
15
  def respond(
16
  message,
 
7
  # client = InferenceClient("EITD/lora_model", token=os.getenv("HF_TOKEN"))
8
 
9
  model = AutoPeftModelForCausalLM.from_pretrained(
10
+ "EITD/orpo_llama", # YOUR MODEL YOU USED FOR TRAINING
 
11
  )
12
+ tokenizer = AutoTokenizer.from_pretrained("EITD/orpo_llama")
13
 
14
  def respond(
15
  message,