sreeramajay commited on
Commit
4e5b5d4
·
1 Parent(s): ca99adc

checkpoint update

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ eval_tokenizer.pad_token = eval_tokenizer.eos_token
27
  # Load Peft Weights
28
  from peft import PeftModel
29
 
30
- ft_model = PeftModel.from_pretrained(model, "mistral-samsum-finetune/checkpoint-100")
31
 
32
  # Format the Sample Input
33
  def formatting_func(example):
 
27
  # Load Peft Weights
28
  from peft import PeftModel
29
 
30
+ ft_model = PeftModel.from_pretrained(model, "mistral-samsum-finetune/checkpoint-150")
31
 
32
  # Format the Sample Input
33
  def formatting_func(example):