File tree Expand file tree Collapse file tree 5 files changed +7
-7
lines changed Expand file tree Collapse file tree 5 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -250,11 +250,12 @@ def _collate(x):
250
250
gen_kwargs ["top_p" ] = None
251
251
if "num_beams" not in gen_kwargs :
252
252
gen_kwargs ["num_beams" ] = 1
253
+ do_sample = True if gen_kwargs ["temperature" ] > 0 else False
253
254
try :
254
255
cont = self .model .generate (
255
256
** inputs ,
256
- do_sample = True if gen_kwargs [ "temperature" ] > 0 else False ,
257
- temperature = gen_kwargs ["temperature" ],
257
+ do_sample = do_sample ,
258
+ temperature = gen_kwargs ["temperature" ] if do_sample else None ,
258
259
top_p = gen_kwargs ["top_p" ],
259
260
num_beams = gen_kwargs ["num_beams" ],
260
261
max_new_tokens = gen_kwargs ["max_new_tokens" ],
Original file line number Diff line number Diff line change @@ -357,11 +357,12 @@ def _collate(x):
357
357
gen_kwargs ["top_p" ] = None
358
358
if "num_beams" not in gen_kwargs :
359
359
gen_kwargs ["num_beams" ] = 1
360
+ do_sample = True if gen_kwargs ["temperature" ] > 0 else False
360
361
try :
361
362
cont = self .model .generate (
362
363
** inputs ,
363
- do_sample = True if gen_kwargs [ "temperature" ] > 0 else False ,
364
- temperature = gen_kwargs ["temperature" ],
364
+ do_sample = do_sample ,
365
+ temperature = gen_kwargs ["temperature" ] if do_sample else None ,
365
366
top_p = gen_kwargs ["top_p" ],
366
367
num_beams = gen_kwargs ["num_beams" ],
367
368
max_new_tokens = gen_kwargs ["max_new_tokens" ],
Original file line number Diff line number Diff line change @@ -2,7 +2,6 @@ dataset_path: yerevann/coco-karpathy
2
2
dataset_kwargs :
3
3
token : True
4
4
task : " coco_karpathy_test"
5
- group : " coco_karpathy"
6
5
test_split : test
7
6
output_type : generate_until
8
7
doc_to_visual : !function utils.coco_doc_to_visual
Original file line number Diff line number Diff line change @@ -2,7 +2,6 @@ dataset_path: yerevann/coco-karpathy
2
2
dataset_kwargs :
3
3
token : True
4
4
task : " coco_karpathy_val"
5
- group : " coco_karpathy"
6
5
test_split : validation
7
6
output_type : generate_until
8
7
doc_to_visual : !function utils.coco_doc_to_visual_karpathy
Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ classifiers = [
19
19
" License :: OSI Approved :: MIT License" ,
20
20
" Operating System :: OS Independent" ,
21
21
]
22
- requires-python = " >=3.12 "
22
+ requires-python = " >=3.8 "
23
23
dependencies = [
24
24
" accelerate>=0.29.1" ,
25
25
" black>=24.1.0" ,
You can’t perform that action at this time.
0 commit comments