NorGLM commited on
Commit
edbd164
1 Parent(s): 41016f5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -39,8 +39,8 @@ tokenizer.pad_token = tokenizer.eos_token
39
  model = PeftModel.from_pretrained(model, peft_model_id)
40
  ```
41
 
42
- ## Evaluation on test set
43
- Load the model to evaluate on the last 20\% of NO-Alpaca dataset:
44
  ```python
45
  def generate_texts(model, tokenizer, prompts, max_seq_length=200, do_sample=True, top_p=0.95, top_k=10):
46
  # prompts are a list of news articles
 
39
  model = PeftModel.from_pretrained(model, peft_model_id)
40
  ```
41
 
42
+ ## Inference on test set
43
+ Load the model to evaluate on the test set of NO-CNN-DailyMail dataset:
44
  ```python
45
  def generate_texts(model, tokenizer, prompts, max_seq_length=200, do_sample=True, top_p=0.95, top_k=10):
46
  # prompts are a list of news articles