from transformers import PixtralVisionConfig, VisionEncoderDecoderConfig | |
class PixtralVisionModelBatchConfig(PixtralVisionConfig): | |
model_type = "pixtral_batch" | |
class VisionPixtralEncoderDecoderConfig(VisionEncoderDecoderConfig): | |
model_type = "vision_pixtral_encoder_decoder" | |