Update supervised.py

Former-commit-id: 788e8232fc4ed58ab2439a9bc2e38f64e12c6eb3
This commit is contained in:
hoshi-hiyouga 2024-06-07 03:38:23 +08:00 committed by GitHub
parent 21df5f0bd0
commit fd7bd911a6

View File

@ -4,7 +4,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Sequence, Tuple
from ...extras.constants import IGNORE_INDEX
from ...extras.logging import get_logger
from .mm_utils import get_paligemma_token_type_ids, get_pixel_values, greedy_knapsack
from .mm_utils import get_paligemma_token_type_ids, get_pixel_values
if TYPE_CHECKING: