File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -178,7 +178,7 @@ single completion:
178
178
output = generator.generate(
179
179
prompt = " Five good reasons to adopt a cat:" ,
180
180
max_new_tokens = 200 ,
181
- add_bos = True )
181
+ add_bos = True ,
182
182
)
183
183
184
184
print (output)
@@ -214,7 +214,7 @@ outputs = generator.generate(
214
214
max_new_tokens = 200 ,
215
215
stop_conditions = [tokenizer.eos_token_id],
216
216
gen_settings = gen_settings,
217
- add_bos = True )
217
+ add_bos = True
218
218
)
219
219
220
220
for o in outputs:
@@ -287,4 +287,4 @@ depending on the job settings. See the docstring for details.
287
287
288
288
## To be continued
289
289
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