From 64ac6ca396b98527364cfeeaba8d290ebfd16489 Mon Sep 17 00:00:00 2001 From: KUANGDD Date: Wed, 23 Oct 2024 15:32:33 +0800 Subject: [PATCH] rm import torch Former-commit-id: 561a0f8155afca20ac699e124320b0eaef6dac07 --- src/llamafactory/data/mm_plugin.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/llamafactory/data/mm_plugin.py b/src/llamafactory/data/mm_plugin.py index a138c058..9249d4ab 100644 --- a/src/llamafactory/data/mm_plugin.py +++ b/src/llamafactory/data/mm_plugin.py @@ -4,7 +4,6 @@ from io import BytesIO from typing import TYPE_CHECKING, Dict, List, Optional, Sequence, Tuple, TypedDict, Union import numpy as np -import torch from transformers.image_utils import get_image_size, to_numpy_array from typing_extensions import override