fix typo · dadwadw233/lmms-eval@7d3536e (original) (raw)
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -216,7 +216,7 @@ def loglikelihood(self, requests: List[Instance]) -> List[Tuple[float, bool]]: | ||
| 216 | 216 | if type(image) is list: |
| 217 | 217 | image = [_image.to(dtype=torch.float16, device=self.device) for _image in image] |
| 218 | 218 | # 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) | |
| 220 | 220 | else: |
| 221 | 221 | image = image.to(dtype=torch.float16, device=self.device) |
| 222 | 222 | else: |