Views
No views yet
1import torch
2from transformers import AutoConfig, AutoModel, AutoProcessor
3
4
5MODEL_NAME = "Qwen/Qwen3-VL-Embedding-2B"
6path_tiny = "./qwen3_vl_embedding_tiny/"
7config = AutoConfig.from_pretrained(MODEL_NAME)
8config.text_config.intermediate_size = 10
9config.text_config.hidden_size = 4
10config.text_config.max_position_embeddings = 10
11config.text_config.num_hidden_layers = 2
12config.text_config.head_dim = 32
13config.text_config.num_attention_heads = 2
14config.text_config.num_key_value_heads = 2
15
16config.vision_config.hidden_size = 64
17config.vision_config.intermediate_size = 64
18config.vision_config.num_heads = 4
19
20
21hf_model = AutoModel.from_config(config)
22hf_model = hf_model.to(torch.float32)
23hf_model.save_pretrained(path_tiny)
24processor = AutoProcessor.from_pretrained(MODEL_NAME)
25processor.save_pretrained(path_tiny)