Kerassy commited on
Commit
6930bc6
·
verified ·
1 Parent(s): a2d0ea7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -77,7 +77,7 @@ inputs = tokenizer([prompt], return_tensors="pt").to("cuda")
77
 
78
  outputs = model.generate(
79
  **inputs,
80
- max_new_tokens=1024,
81
  use_cache=True,
82
  temperature=0.7,
83
  top_p=0.9,
@@ -103,7 +103,7 @@ inputs_direct = tokenizer([prompt_direct], return_tensors="pt").to("cuda")
103
 
104
  outputs_direct = model.generate(
105
  **inputs_direct,
106
- max_new_tokens=1024,
107
  eos_token_id=tokenizer.eos_token_id
108
  )
109
 
 
77
 
78
  outputs = model.generate(
79
  **inputs,
80
+ max_tokens=1024,
81
  use_cache=True,
82
  temperature=0.7,
83
  top_p=0.9,
 
103
 
104
  outputs_direct = model.generate(
105
  **inputs_direct,
106
+ max_tokens=1024,
107
  eos_token_id=tokenizer.eos_token_id
108
  )
109