Update supervised.py

Former-commit-id: 49993c4f4e1f871a22ff0196afe60026b668a4dc
This commit is contained in:
hoshi-hiyouga 2024-06-07 03:38:23 +08:00 committed by GitHub
parent fdeec6db52
commit be1e5f9d62

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.constants import IGNORE_INDEX
from ...extras.logging import get_logger 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: if TYPE_CHECKING: