erikinfo commited on
Commit
269b640
·
1 Parent(s): a81cd46

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -18,11 +18,10 @@ Keyword Format: "Main Topic"."Subtopic1","Subtopic2","Subtopic3"
18
 
19
  Code Example:
20
  ```
21
- prompt = <BOS> + title + \
22
  <|SEP|> + keywords + <|SEP|>
23
 
24
  generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
25
 
26
  model.eval();
27
  ```
28
- Currently only works when resizing the model with model.resize_token_embeddings(len(tokenizer))
 
18
 
19
  Code Example:
20
  ```
21
+ prompt = <BOS> + title + \\
22
  <|SEP|> + keywords + <|SEP|>
23
 
24
  generated = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
25
 
26
  model.eval();
27
  ```