dkleine commited on
Commit
8cfb465
1 Parent(s): ff07dc0

fixed generation_args in Sample inference code

Browse files

* fixed temperature for "do_sample": False (`temperature=0.0` would require sampling, which is turned off with `"do_sample": False`. therefore better to use `temperature=1.0` without sampling)

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -133,7 +133,7 @@ pipe = pipeline(
133
  generation_args = {
134
  "max_new_tokens": 500,
135
  "return_full_text": False,
136
- "temperature": 0.0,
137
  "do_sample": False,
138
  }
139
 
 
133
  generation_args = {
134
  "max_new_tokens": 500,
135
  "return_full_text": False,
136
+ "temperature": 1.0,
137
  "do_sample": False,
138
  }
139