zjkarina commited on
Commit
8629e06
·
1 Parent(s): 4f34161

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -3,7 +3,8 @@
3
  <!-- Provide a longer summary of what this model is. -->
4
 
5
  ## Uses
6
- <!-- gen_kwargs = {
 
7
  "max_new_tokens": 100,
8
  "top_k": 70,
9
  "top_p": 0.8,
@@ -26,5 +27,4 @@ model.to(device)
26
  prepared = tokenizer.encode(inp, return_tensors='pt').to(model.device)
27
  out = model.generate(input_ids=prepared, **gen_kwargs)
28
  generated = tokenizer.decode(out[0])
29
- -->
30
-
 
3
  <!-- Provide a longer summary of what this model is. -->
4
 
5
  ## Uses
6
+ ```
7
+ gen_kwargs = {
8
  "max_new_tokens": 100,
9
  "top_k": 70,
10
  "top_p": 0.8,
 
27
  prepared = tokenizer.encode(inp, return_tensors='pt').to(model.device)
28
  out = model.generate(input_ids=prepared, **gen_kwargs)
29
  generated = tokenizer.decode(out[0])
30
+ ```