File size: 225 Bytes
032e687 |
1 2 3 4 5 6 |
# Copyright (c) OpenMMLab. All rights reserved.
from xtuner.model.utils import *
from typing import List, Optional
import torch
from transformers import PreTrainedModel
from xtuner.utils import IGNORE_INDEX, IMAGE_TOKEN_INDEX |