Skip to content

Commit ef30651

Browse files
authored
fix typo
1 parent 9bab677 commit ef30651

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lmms_eval/models/tinyllava.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def loglikelihood(self, requests: List[Instance]) -> List[Tuple[float, bool]]:
216216
if type(image) is list:
217217
image = [_image.to(dtype=torch.float16, device=self.device) for _image in image]
218218
# as of 2024/06, tinyllava only accepts `images` input to be a tensor
219-
image_tensor = torch.stack(image_tensor)
219+
image = torch.stack(image)
220220
else:
221221
image = image.to(dtype=torch.float16, device=self.device)
222222
else:

0 commit comments

Comments
 (0)