File size: 287 Bytes
c15c861
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from transformers import PixtralVisionConfig, VisionEncoderDecoderConfig


class PixtralVisionModelBatchConfig(PixtralVisionConfig):
    model_type = "pixtral_batch"


class VisionPixtralEncoderDecoderConfig(VisionEncoderDecoderConfig):
    model_type = "vision_pixtral_encoder_decoder"