| from transformers.configuration_utils import PretrainedConfig |
| from transformers.models.qwen2_5_vl.configuration_qwen2_5_vl import Qwen2_5_VLVisionConfig |
| from transformers.models.qwen2.configuration_qwen2 import Qwen2Config |
|
|
|
|
| class OpenCUAConfig(PretrainedConfig): |
| """OpenCUA-2.5-32B model configuration. |
| |
| Args: |
| vision_config: Configuration for the vision model.Qwen2_5_VLVisionConfig |
| text_config: Configuration for the text model. Qwen2Config |
| pad_token_id: The token ID to use for padding. |
| """ |
|
|
| model_type = "opencua" |
|
|
| def __init__( |
| self, |
| vision_config: dict | Qwen2_5_VLVisionConfig | None = None, |
| text_config: dict | Qwen2Config | None = None, |
| ignore_index: int = -100, |
| media_placeholder_token_id: int = 151664, |
| pad_token_id: int = 0, |
| **kwargs |
| ): |
| if isinstance(vision_config, dict): |
| vision_config = Qwen2_5_VLVisionConfig(**vision_config) |
| self.vision_config = vision_config |
|
|
| if isinstance(text_config, dict): |
| text_config = Qwen2Config(**text_config) |
| self.text_config = text_config |
|
|
| self.ignore_index = ignore_index |
| self.media_placeholder_token_id = media_placeholder_token_id |
|
|
| super().__init__(pad_token_id=pad_token_id, **kwargs) |
|
|
