Update supervised.py

This commit is contained in:
hoshi-hiyouga 2024-06-07 03:38:23 +08:00 committed by GitHub
parent 8cecade708
commit 788e8232fc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

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: