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"
|