File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -178,7 +178,7 @@ single completion:
178178output = generator.generate(
179179 prompt = " Five good reasons to adopt a cat:" ,
180180 max_new_tokens = 200 ,
181- add_bos = True )
181+ add_bos = True ,
182182)
183183
184184print (output)
@@ -214,7 +214,7 @@ outputs = generator.generate(
214214 max_new_tokens = 200 ,
215215 stop_conditions = [tokenizer.eos_token_id],
216216 gen_settings = gen_settings,
217- add_bos = True )
217+ add_bos = True
218218)
219219
220220for o in outputs:
@@ -287,4 +287,4 @@ depending on the job settings. See the docstring for details.
287287
288288## To be continued
289289
290- I may expand on this in the future, maybe with some performance benchmarks or something. Here's a cat: 🐈
290+ I may expand on this in the future, maybe with some performance benchmarks or something. Here's a cat: 🐈
You can’t perform that action at this time.
0 commit comments