rm import torch

hiyouga · Oct 23, 2024 · a857184 · a857184
1 parent 341a79f
commit a857184
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/llamafactory/data/mm_plugin.py b/src/llamafactory/data/mm_plugin.py
@@ -4,7 +4,6 @@
 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