Skip to content

Commit 9935012

Browse files
authored
[Fix] Minor fix on some warning messages (#704)
* Lower python deps * Fix warnings * Fix warnings during generation
1 parent c4a3128 commit 9935012

File tree

5 files changed

+7
-7
lines changed

5 files changed

+7
-7
lines changed

lmms_eval/models/aero.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -250,11 +250,12 @@ def _collate(x):
250250
gen_kwargs["top_p"] = None
251251
if "num_beams" not in gen_kwargs:
252252
gen_kwargs["num_beams"] = 1
253+
do_sample = True if gen_kwargs["temperature"] > 0 else False
253254
try:
254255
cont = self.model.generate(
255256
**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,
258259
top_p=gen_kwargs["top_p"],
259260
num_beams=gen_kwargs["num_beams"],
260261
max_new_tokens=gen_kwargs["max_new_tokens"],

lmms_eval/models/llava_hf.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -357,11 +357,12 @@ def _collate(x):
357357
gen_kwargs["top_p"] = None
358358
if "num_beams" not in gen_kwargs:
359359
gen_kwargs["num_beams"] = 1
360+
do_sample = True if gen_kwargs["temperature"] > 0 else False
360361
try:
361362
cont = self.model.generate(
362363
**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,
365366
top_p=gen_kwargs["top_p"],
366367
num_beams=gen_kwargs["num_beams"],
367368
max_new_tokens=gen_kwargs["max_new_tokens"],

lmms_eval/tasks/coco_cap/coco_karpathy_test.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ dataset_path: yerevann/coco-karpathy
22
dataset_kwargs:
33
token: True
44
task: "coco_karpathy_test"
5-
group : "coco_karpathy"
65
test_split: test
76
output_type: generate_until
87
doc_to_visual: !function utils.coco_doc_to_visual

lmms_eval/tasks/coco_cap/coco_karpathy_val.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ dataset_path: yerevann/coco-karpathy
22
dataset_kwargs:
33
token: True
44
task: "coco_karpathy_val"
5-
group : "coco_karpathy"
65
test_split: validation
76
output_type: generate_until
87
doc_to_visual: !function utils.coco_doc_to_visual_karpathy

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ classifiers = [
1919
"License :: OSI Approved :: MIT License",
2020
"Operating System :: OS Independent",
2121
]
22-
requires-python = ">=3.12"
22+
requires-python = ">=3.8"
2323
dependencies = [
2424
"accelerate>=0.29.1",
2525
"black>=24.1.0",

0 commit comments

Comments
 (0)