File size: 861 Bytes
5550028
 
 
119344f
 
 
 
 
 
 
0932bd5
119344f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
---
license: apache-2.0
---

```python
import torch
from transformers import AutoConfig, AutoModel, AutoProcessor


MODEL_NAME = "Qwen/Qwen3-VL-Embedding-2B"
path_tiny = "./qwen3_vl_embedding_tiny/"
config = AutoConfig.from_pretrained(MODEL_NAME)
config.text_config.intermediate_size = 10
config.text_config.hidden_size = 4
config.text_config.max_position_embeddings = 10
config.text_config.num_hidden_layers = 2
config.text_config.head_dim = 32
config.text_config.num_attention_heads = 2
config.text_config.num_key_value_heads = 2

config.vision_config.hidden_size = 64
config.vision_config.intermediate_size = 64
config.vision_config.num_heads = 4


hf_model = AutoModel.from_config(config)
hf_model = hf_model.to(torch.float32)
hf_model.save_pretrained(path_tiny)
processor = AutoProcessor.from_pretrained(MODEL_NAME)
processor.save_pretrained(path_tiny)
```