Upload 47 files
Browse files- .gitattributes +4 -0
- gradio_demo.py +127 -0
- internvl3_1b_ax650/internvl3_1b_vit.axmodel +3 -0
- internvl3_1b_ax650/model.embed_tokens.weight.bfloat16.bin +3 -0
- internvl3_1b_ax650/qwen2_p128_l0_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l10_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l11_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l12_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l13_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l14_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l15_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l16_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l17_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l18_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l19_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l1_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l20_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l21_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l22_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l23_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l2_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l3_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l4_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l5_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l6_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l7_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l8_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_p128_l9_together.axmodel +3 -0
- internvl3_1b_ax650/qwen2_post.axmodel +3 -0
- internvl3_tokenizer.py +186 -0
- internvl3_tokenizer/added_tokens.json +33 -0
- internvl3_tokenizer/config.json +224 -0
- internvl3_tokenizer/generation_config.json +4 -0
- internvl3_tokenizer/merges.txt +0 -0
- internvl3_tokenizer/preprocessor_config.json +19 -0
- internvl3_tokenizer/special_tokens_map.json +31 -0
- internvl3_tokenizer/tokenizer.json +0 -0
- internvl3_tokenizer/tokenizer_config.json +281 -0
- internvl3_tokenizer/vocab.json +0 -0
- main_api_ax650 +3 -0
- main_api_axcl_x86 +3 -0
- main_ax650 +3 -0
- main_axcl_x86 +3 -0
- post_config.json +14 -0
- run_internvl_3_1b_448_api_ax650.sh +10 -0
- run_internvl_3_1b_448_api_axcl_x86.sh +11 -0
- run_internvl_3_1b_448_ax650.sh +11 -0
- run_internvl_3_1b_448_axcl_x86.sh +12 -0
.gitattributes
CHANGED
@@ -34,3 +34,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
*.axmodel filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
*.axmodel filter=lfs diff=lfs merge=lfs -text
|
37 |
+
main_api_ax650 filter=lfs diff=lfs merge=lfs -text
|
38 |
+
main_api_axcl_x86 filter=lfs diff=lfs merge=lfs -text
|
39 |
+
main_ax650 filter=lfs diff=lfs merge=lfs -text
|
40 |
+
main_axcl_x86 filter=lfs diff=lfs merge=lfs -text
|
gradio_demo.py
ADDED
@@ -0,0 +1,127 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import mimetypes
|
2 |
+
import os
|
3 |
+
import gradio as gr
|
4 |
+
import requests
|
5 |
+
import json, time
|
6 |
+
|
7 |
+
base_url = "http://127.0.0.1:8000"
|
8 |
+
|
9 |
+
def upload_image(file_path):
|
10 |
+
if file_path is None:
|
11 |
+
return None
|
12 |
+
# Gradio File component returns a tempfile-like object
|
13 |
+
# file_path = image.name
|
14 |
+
filename = os.path.basename(file_path)
|
15 |
+
# Guess MIME type
|
16 |
+
mime_type, _ = mimetypes.guess_type(filename)
|
17 |
+
mime_type = mime_type or 'application/octet-stream'
|
18 |
+
# Open file in binary mode for upload
|
19 |
+
with open(file_path, 'rb') as f:
|
20 |
+
file_bytes = f.read()
|
21 |
+
# Prepare multipart form data
|
22 |
+
files = {
|
23 |
+
'image': (filename, file_bytes, mime_type)
|
24 |
+
}
|
25 |
+
# Send to upload endpoint
|
26 |
+
resp = requests.post(
|
27 |
+
f'{base_url}/api/upload',
|
28 |
+
files=files
|
29 |
+
)
|
30 |
+
resp.raise_for_status()
|
31 |
+
data = resp.json()
|
32 |
+
return data.get('file_path')
|
33 |
+
|
34 |
+
def stop_generation():
|
35 |
+
try:
|
36 |
+
requests.get(f'{base_url}/api/stop')
|
37 |
+
except:
|
38 |
+
pass
|
39 |
+
|
40 |
+
def respond(prompt, image:gr.Image, temp, rep_penalty, tp, tk, history=None):
|
41 |
+
if history is None:
|
42 |
+
history = []
|
43 |
+
if not prompt.strip():
|
44 |
+
return history
|
45 |
+
# append empty response to history
|
46 |
+
if image is None:
|
47 |
+
file_path = None
|
48 |
+
else:
|
49 |
+
file_path = upload_image(image)
|
50 |
+
history.append((f'', None))
|
51 |
+
relative_path = os.path.relpath(file_path)
|
52 |
+
# html = f"<img src='{relative_path}' style='max-width:300px;'/>"
|
53 |
+
# history.append((html, None))
|
54 |
+
# print(relative_path)
|
55 |
+
|
56 |
+
|
57 |
+
history.append((prompt, ""))
|
58 |
+
yield history
|
59 |
+
# stream updates
|
60 |
+
|
61 |
+
|
62 |
+
payload = {
|
63 |
+
"prompt": prompt,
|
64 |
+
"temperature": temp,
|
65 |
+
"repetition_penalty": rep_penalty,
|
66 |
+
"top-p": tp,
|
67 |
+
"top-k": tk
|
68 |
+
}
|
69 |
+
if file_path:
|
70 |
+
payload["file_path"] = file_path
|
71 |
+
|
72 |
+
response = requests.post(
|
73 |
+
f'{base_url}/api/generate',
|
74 |
+
json=payload
|
75 |
+
)
|
76 |
+
response.raise_for_status()
|
77 |
+
|
78 |
+
|
79 |
+
while True:
|
80 |
+
time.sleep(0.01)
|
81 |
+
response = requests.get(
|
82 |
+
f'{base_url}/api/generate_provider'
|
83 |
+
)
|
84 |
+
data = response.json()
|
85 |
+
chunk:str = data.get("response", "")
|
86 |
+
done = data.get("done", False)
|
87 |
+
if done:
|
88 |
+
break
|
89 |
+
if chunk.strip() == "":
|
90 |
+
continue
|
91 |
+
history[-1] = (prompt, history[-1][1] + chunk)
|
92 |
+
yield history
|
93 |
+
|
94 |
+
print("end")
|
95 |
+
|
96 |
+
|
97 |
+
|
98 |
+
|
99 |
+
def chat_interface():
|
100 |
+
with gr.Blocks(theme=gr.themes.Soft(font="Consolas"), fill_width=True) as demo:
|
101 |
+
gr.Markdown("## Chat with LLM\nUpload an image and chat with the model!")
|
102 |
+
with gr.Row():
|
103 |
+
image = gr.Image(label="Upload Image", type="filepath")
|
104 |
+
with gr.Column(scale=3):
|
105 |
+
chatbot = gr.Chatbot(height=600)
|
106 |
+
prompt = gr.Textbox(placeholder="Type your message...", label="Prompt", value="描述一下这张图片")
|
107 |
+
with gr.Row():
|
108 |
+
btn_chat = gr.Button("Chat", variant="primary")
|
109 |
+
btn_stop = gr.Button("Stop", variant="stop")
|
110 |
+
|
111 |
+
with gr.Column(scale=1):
|
112 |
+
temperature = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, value=0.7, label="Temperature")
|
113 |
+
repetition_penalty = gr.Slider(minimum=1.0, maximum=2.0, step=0.01, value=1.0, label="Repetition Penalty")
|
114 |
+
top_p = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, value=0.9, label="Top-p Sampling")
|
115 |
+
top_k = gr.Slider(minimum=0, maximum=100, step=1, value=40, label="Top-k Sampling")
|
116 |
+
|
117 |
+
btn_stop.click(fn=stop_generation, inputs=None, outputs=None)
|
118 |
+
btn_chat.click(
|
119 |
+
fn=respond,
|
120 |
+
inputs=[prompt, image, temperature, repetition_penalty, top_p, top_k, chatbot],
|
121 |
+
outputs=chatbot
|
122 |
+
)
|
123 |
+
|
124 |
+
demo.launch(server_name="0.0.0.0", server_port=7860)
|
125 |
+
|
126 |
+
if __name__ == "__main__":
|
127 |
+
chat_interface()
|
internvl3_1b_ax650/internvl3_1b_vit.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0603400f56632d7580312bbc6dee7f25c61f43f57bc1e19a63fff20241ca99c6
|
3 |
+
size 368057444
|
internvl3_1b_ax650/model.embed_tokens.weight.bfloat16.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9cecf2aa3a9d0762e525990561e75f08ab5df970d918f53e1b72dd73c81a6d34
|
3 |
+
size 271799808
|
internvl3_1b_ax650/qwen2_p128_l0_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eead6ae0b49ad5a4c554fb85c345f18bed3104289ea4f612aaeea581aa708576
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l10_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9926dd7996dbe34e4fb117aa130d0ba2dd1d63d9b338e02e2d62dd4b62fbba80
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l11_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3c9f09f97c5cd35fafc260e4e2fb29849fb1006b61b7a5158cdcadbc1688be33
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l12_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3ca530f6bc2c4a9d9656d5368fd547e2e75c76126e8980e6057520be384554aa
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l13_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8a420ab0e159a13b349338faed80a33c63103dc489da58d3756a743943cfd8bb
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l14_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0e1a05a44e10c3f163cb1b53d955195efe10eea4a034d264391991ebd9b165ae
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l15_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8fa98c1c6645edde2f3917a006915d14fccea19ab89dd4d921cb724fbf421e9c
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l16_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d7af902e12c1c4164f9f1d176d211568a86c3f218f0ff3acf7f4e46e070fe64b
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l17_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b979a366484f19c6cfa19e4fcf67f117612412e356e45533a2488d3634bd4c5d
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l18_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5d6d979c9dc5ee4fde522fcf2eb5b425c31a8f119281f40e424b424e41fb7782
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l19_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:72d8b0463bae2e9c38eaf94c12f3c693df79a2154e36ac156768ed4d6ca7a4cf
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l1_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6555c32c31e2ba89cbf4906517d4fee636cfa29145707f7a502af3b5a19933e5
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l20_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:531c9e292abcd467f6d99fbacfaa7cee1b93450b0da974ca67ff8c47dec3c79a
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l21_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e4d8b1aba7f6a57d755be7b2340881dd525aeab804ad26e2db7843764e20456
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l22_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:624fc3ca9319c83252c830a5c97f68e90b4550363c9ff3f6c55f827ea42bca29
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l23_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:72e3922f62e6b33721aa9b2fef57b720a206265ca2b90d5f886132c62b76b546
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l2_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:45d5e98ed1110457594d77ef379786eb0838b6d9f63b21e18ec67965ec218fd6
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l3_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1abf7d7beeb56ef269abc2c1c6804e7f740d1a16432a86e8cecca926e29a027a
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l4_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f2bb08583ab396116d38a39f1b66da3b6b75a72ea87f98e4b326f34c832ec6f3
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l5_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:664596f8641b49121957e1d740ac668b6245831a39e35327e8f7ee460d906840
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l6_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c8764ce12a92255d85c0a4cd86d9aaadb3ef14964e2108921d898b757402970b
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l7_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c705aed26bc995e361147a0d79fa9cfb7bb735f5903ba551726b38066fef5b2e
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l8_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3cd57814326331fd677adfdd903e585b078bf4a1b13169474029ebed1ea12aeb
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_p128_l9_together.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:11eab50591e174e21d6c45dacc6fc0eb5a8eb543497d2d71f99c2f0a6ffbc133
|
3 |
+
size 20071994
|
internvl3_1b_ax650/qwen2_post.axmodel
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:889f1bee82603ad07d5239fbcfccbea71dda8f5a3188b3ad1799f2f7608fcaff
|
3 |
+
size 147705143
|
internvl3_tokenizer.py
ADDED
@@ -0,0 +1,186 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import AutoTokenizer
|
2 |
+
from http.server import HTTPServer, BaseHTTPRequestHandler
|
3 |
+
import json
|
4 |
+
import argparse
|
5 |
+
|
6 |
+
|
7 |
+
class Tokenizer_Http():
|
8 |
+
|
9 |
+
def __init__(self):
|
10 |
+
|
11 |
+
path = 'internvl3_tokenizer'
|
12 |
+
self.tokenizer = AutoTokenizer.from_pretrained(path,
|
13 |
+
trust_remote_code=True,
|
14 |
+
use_fast=False)
|
15 |
+
|
16 |
+
def encode(self, content):
|
17 |
+
prompt = f"<|im_start|>system\n你是由上海人工智能实验室联合商汤科技开发的书生多模态大模型,英文名叫InternVL, 是一个有用无害的人工智能助手。<|im_end|><|im_start|>user\n{content}<|im_end|><|im_start|>assistant\n"
|
18 |
+
input_ids = self.tokenizer.encode(prompt)
|
19 |
+
return input_ids
|
20 |
+
|
21 |
+
def encode_with_image(self, question, num_of_images, imgsz) -> list:
|
22 |
+
prompt = "<|im_start|>system\n你是书生·万象, 英文名是InternVL, 是由上海人工智能实验室、清华大学及多家合作单位联合开发的多模态大语言模型.<|im_end|>\n"
|
23 |
+
|
24 |
+
prompt += "<|im_start|>user\n" + question
|
25 |
+
|
26 |
+
context_len = 64
|
27 |
+
if imgsz == 448:
|
28 |
+
context_len = 256
|
29 |
+
elif imgsz == 224:
|
30 |
+
context_len = 64
|
31 |
+
else:
|
32 |
+
print(f"imgsz is {imgsz}")
|
33 |
+
return
|
34 |
+
print("context_len is ", context_len)
|
35 |
+
|
36 |
+
if num_of_images > 0:
|
37 |
+
for idx in range(num_of_images):
|
38 |
+
prompt += "\n<img>" + "<IMG_CONTEXT>" * context_len + "</img>\n"
|
39 |
+
|
40 |
+
prompt += "<|im_end|>\n<|im_start|>assistant"
|
41 |
+
print(f"prompt is {prompt}")
|
42 |
+
token_ids = self.tokenizer.encode(prompt)
|
43 |
+
return token_ids
|
44 |
+
|
45 |
+
|
46 |
+
def decode(self, token_ids):
|
47 |
+
return self.tokenizer.decode(token_ids,
|
48 |
+
clean_up_tokenization_spaces=False, skip_special_tokens=True)
|
49 |
+
|
50 |
+
@property
|
51 |
+
def bos_id(self):
|
52 |
+
return self.tokenizer.bos_token_id
|
53 |
+
|
54 |
+
@property
|
55 |
+
def eos_id(self):
|
56 |
+
return self.tokenizer.eos_token_id
|
57 |
+
|
58 |
+
@property
|
59 |
+
def bos_token(self):
|
60 |
+
return self.tokenizer.bos_token
|
61 |
+
|
62 |
+
@property
|
63 |
+
def eos_token(self):
|
64 |
+
return self.tokenizer.eos_token
|
65 |
+
|
66 |
+
@property
|
67 |
+
def img_start_token(self):
|
68 |
+
return self.tokenizer.encode("<img>")[0]
|
69 |
+
|
70 |
+
@property
|
71 |
+
def img_context_token(self):
|
72 |
+
return self.tokenizer.encode("<IMG_CONTEXT>")[0]
|
73 |
+
|
74 |
+
tokenizer = Tokenizer_Http()
|
75 |
+
|
76 |
+
print(tokenizer.bos_id, tokenizer.bos_token, tokenizer.eos_id,
|
77 |
+
tokenizer.eos_token, tokenizer.img_start_token, tokenizer.img_context_token)
|
78 |
+
token_ids = tokenizer.encode_with_image("你好", 1, 448)
|
79 |
+
print(token_ids)
|
80 |
+
print(len(token_ids))
|
81 |
+
token_ids = tokenizer.encode("hello world")
|
82 |
+
print(token_ids)
|
83 |
+
print(len(token_ids))
|
84 |
+
|
85 |
+
|
86 |
+
class Request(BaseHTTPRequestHandler):
|
87 |
+
#通过类继承,新定义类
|
88 |
+
timeout = 5
|
89 |
+
server_version = 'Apache'
|
90 |
+
|
91 |
+
def do_GET(self):
|
92 |
+
print(self.path)
|
93 |
+
#在新类中定义get的内容(当客户端向该服务端使用get请求时,本服务端将如下运行)
|
94 |
+
self.send_response(200)
|
95 |
+
self.send_header("type", "get") #设置响应头,可省略或设置多个
|
96 |
+
self.end_headers()
|
97 |
+
|
98 |
+
if self.path == '/bos_id':
|
99 |
+
bos_id = tokenizer.bos_id
|
100 |
+
# print(bos_id)
|
101 |
+
# to json
|
102 |
+
if bos_id is None:
|
103 |
+
msg = json.dumps({'bos_id': -1})
|
104 |
+
else:
|
105 |
+
msg = json.dumps({'bos_id': bos_id})
|
106 |
+
elif self.path == '/eos_id':
|
107 |
+
eos_id = tokenizer.eos_id
|
108 |
+
if eos_id is None:
|
109 |
+
msg = json.dumps({'eos_id': -1})
|
110 |
+
else:
|
111 |
+
msg = json.dumps({'eos_id': eos_id})
|
112 |
+
elif self.path == '/img_start_token':
|
113 |
+
img_start_token = tokenizer.img_start_token
|
114 |
+
if img_start_token is None:
|
115 |
+
msg = json.dumps({'img_start_token': -1})
|
116 |
+
else:
|
117 |
+
msg = json.dumps({'img_start_token': img_start_token})
|
118 |
+
elif self.path == '/img_context_token':
|
119 |
+
img_context_token = tokenizer.img_context_token
|
120 |
+
if img_context_token is None:
|
121 |
+
msg = json.dumps({'img_context_token': -1})
|
122 |
+
else:
|
123 |
+
msg = json.dumps({'img_context_token': img_context_token})
|
124 |
+
else:
|
125 |
+
msg = 'error'
|
126 |
+
|
127 |
+
print(msg)
|
128 |
+
msg = str(msg).encode() #转为str再转为byte格式
|
129 |
+
|
130 |
+
self.wfile.write(msg) #将byte格式的信息返回给客户端
|
131 |
+
|
132 |
+
def do_POST(self):
|
133 |
+
#在新类中定义post的内容(当客户端向该服务端使用post请求时,本服务端将如下运行)
|
134 |
+
data = self.rfile.read(int(
|
135 |
+
self.headers['content-length'])) #获取从客户端传入的参数(byte格式)
|
136 |
+
data = data.decode() #将byte格式转为str格式
|
137 |
+
|
138 |
+
self.send_response(200)
|
139 |
+
self.send_header("type", "post") #设置响应头,可��略或设置多个
|
140 |
+
self.end_headers()
|
141 |
+
|
142 |
+
if self.path == '/encode':
|
143 |
+
req = json.loads(data)
|
144 |
+
print(req)
|
145 |
+
prompt = req['text']
|
146 |
+
b_img_prompt = False
|
147 |
+
if 'img_prompt' in req:
|
148 |
+
b_img_prompt = req['img_prompt']
|
149 |
+
if b_img_prompt:
|
150 |
+
num_img = req['num_img']
|
151 |
+
imgsz = req['imgsz']
|
152 |
+
token_ids = tokenizer.encode_with_image(prompt, num_img, imgsz)
|
153 |
+
else:
|
154 |
+
token_ids = tokenizer.encode(prompt)
|
155 |
+
if token_ids is None:
|
156 |
+
msg = json.dumps({'token_ids': -1})
|
157 |
+
else:
|
158 |
+
msg = json.dumps({'token_ids': token_ids})
|
159 |
+
|
160 |
+
elif self.path == '/decode':
|
161 |
+
req = json.loads(data)
|
162 |
+
token_ids = req['token_ids']
|
163 |
+
text = tokenizer.decode(token_ids)
|
164 |
+
if text is None:
|
165 |
+
msg = json.dumps({'text': ""})
|
166 |
+
else:
|
167 |
+
msg = json.dumps({'text': text})
|
168 |
+
else:
|
169 |
+
msg = 'error'
|
170 |
+
print(msg)
|
171 |
+
msg = str(msg).encode() #转为str再转为byte格式
|
172 |
+
|
173 |
+
self.wfile.write(msg) #将byte格式的信息返回给客户端
|
174 |
+
|
175 |
+
|
176 |
+
if __name__ == "__main__":
|
177 |
+
|
178 |
+
args = argparse.ArgumentParser()
|
179 |
+
args.add_argument('--host', type=str, default='0.0.0.0')
|
180 |
+
args.add_argument('--port', type=int, default=12345)
|
181 |
+
args = args.parse_args()
|
182 |
+
|
183 |
+
host = (args.host, args.port) #设定地址与端口号,'localhost'等价于'127.0.0.1'
|
184 |
+
print('http://%s:%s' % host)
|
185 |
+
server = HTTPServer(host, Request) #根据地址端口号和新定义的类,创建服务器实例
|
186 |
+
server.serve_forever() #开启服务
|
internvl3_tokenizer/added_tokens.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"</box>": 151673,
|
3 |
+
"</img>": 151666,
|
4 |
+
"</quad>": 151669,
|
5 |
+
"</ref>": 151671,
|
6 |
+
"</tool_call>": 151658,
|
7 |
+
"<IMG_CONTEXT>": 151667,
|
8 |
+
"<box>": 151672,
|
9 |
+
"<img>": 151665,
|
10 |
+
"<quad>": 151668,
|
11 |
+
"<ref>": 151670,
|
12 |
+
"<tool_call>": 151657,
|
13 |
+
"<|box_end|>": 151649,
|
14 |
+
"<|box_start|>": 151648,
|
15 |
+
"<|endoftext|>": 151643,
|
16 |
+
"<|file_sep|>": 151664,
|
17 |
+
"<|fim_middle|>": 151660,
|
18 |
+
"<|fim_pad|>": 151662,
|
19 |
+
"<|fim_prefix|>": 151659,
|
20 |
+
"<|fim_suffix|>": 151661,
|
21 |
+
"<|im_end|>": 151645,
|
22 |
+
"<|im_start|>": 151644,
|
23 |
+
"<|image_pad|>": 151655,
|
24 |
+
"<|object_ref_end|>": 151647,
|
25 |
+
"<|object_ref_start|>": 151646,
|
26 |
+
"<|quad_end|>": 151651,
|
27 |
+
"<|quad_start|>": 151650,
|
28 |
+
"<|repo_name|>": 151663,
|
29 |
+
"<|video_pad|>": 151656,
|
30 |
+
"<|vision_end|>": 151653,
|
31 |
+
"<|vision_pad|>": 151654,
|
32 |
+
"<|vision_start|>": 151652
|
33 |
+
}
|
internvl3_tokenizer/config.json
ADDED
@@ -0,0 +1,224 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_commit_hash": null,
|
3 |
+
"_name_or_path": "/mnt/petrelfs/wangweiyun/workspace_wwy/open_source/InternVL/internvl_chat/work_dirs/internvl_chat_v3_0/InternVL3_0-1B-MPO-try0-2",
|
4 |
+
"architectures": [
|
5 |
+
"InternVLChatModel"
|
6 |
+
],
|
7 |
+
"auto_map": {
|
8 |
+
"AutoConfig": "configuration_internvl_chat.InternVLChatConfig",
|
9 |
+
"AutoModel": "modeling_internvl_chat.InternVLChatModel",
|
10 |
+
"AutoModelForCausalLM": "modeling_internvl_chat.InternVLChatModel"
|
11 |
+
},
|
12 |
+
"downsample_ratio": 0.5,
|
13 |
+
"dynamic_image_size": true,
|
14 |
+
"force_image_size": 448,
|
15 |
+
"hidden_size": 896,
|
16 |
+
"image_fold": null,
|
17 |
+
"llm_config": {
|
18 |
+
"_attn_implementation_autoset": true,
|
19 |
+
"_name_or_path": "./pretrained/Qwen2.5-32B-Instruct",
|
20 |
+
"add_cross_attention": false,
|
21 |
+
"architectures": [
|
22 |
+
"Qwen2ForCausalLM"
|
23 |
+
],
|
24 |
+
"attention_dropout": 0.0,
|
25 |
+
"bad_words_ids": null,
|
26 |
+
"begin_suppress_tokens": null,
|
27 |
+
"bos_token_id": 151643,
|
28 |
+
"chunk_size_feed_forward": 0,
|
29 |
+
"cross_attention_hidden_size": null,
|
30 |
+
"decoder_start_token_id": null,
|
31 |
+
"diversity_penalty": 0.0,
|
32 |
+
"do_sample": false,
|
33 |
+
"early_stopping": false,
|
34 |
+
"encoder_no_repeat_ngram_size": 0,
|
35 |
+
"eos_token_id": 151643,
|
36 |
+
"exponential_decay_length_penalty": null,
|
37 |
+
"finetuning_task": null,
|
38 |
+
"forced_bos_token_id": null,
|
39 |
+
"forced_eos_token_id": null,
|
40 |
+
"hidden_act": "silu",
|
41 |
+
"hidden_size": 896,
|
42 |
+
"id2label": {
|
43 |
+
"0": "LABEL_0",
|
44 |
+
"1": "LABEL_1"
|
45 |
+
},
|
46 |
+
"initializer_range": 0.02,
|
47 |
+
"intermediate_size": 4864,
|
48 |
+
"is_decoder": false,
|
49 |
+
"is_encoder_decoder": false,
|
50 |
+
"label2id": {
|
51 |
+
"LABEL_0": 0,
|
52 |
+
"LABEL_1": 1
|
53 |
+
},
|
54 |
+
"length_penalty": 1.0,
|
55 |
+
"max_length": 20,
|
56 |
+
"max_position_embeddings": 32768,
|
57 |
+
"max_window_layers": 70,
|
58 |
+
"min_length": 0,
|
59 |
+
"model_type": "qwen2",
|
60 |
+
"moe_config": null,
|
61 |
+
"no_repeat_ngram_size": 0,
|
62 |
+
"num_attention_heads": 14,
|
63 |
+
"num_beam_groups": 1,
|
64 |
+
"num_beams": 1,
|
65 |
+
"num_hidden_layers": 24,
|
66 |
+
"num_key_value_heads": 2,
|
67 |
+
"num_return_sequences": 1,
|
68 |
+
"output_attentions": false,
|
69 |
+
"output_hidden_states": false,
|
70 |
+
"output_scores": false,
|
71 |
+
"pad_token_id": null,
|
72 |
+
"prefix": null,
|
73 |
+
"problem_type": null,
|
74 |
+
"pruned_heads": {},
|
75 |
+
"remove_invalid_values": false,
|
76 |
+
"repetition_penalty": 1.0,
|
77 |
+
"return_dict": true,
|
78 |
+
"return_dict_in_generate": false,
|
79 |
+
"rms_norm_eps": 1e-06,
|
80 |
+
"rope_scaling": {
|
81 |
+
"factor": 2.0,
|
82 |
+
"rope_type": "dynamic",
|
83 |
+
"type": "dynamic"
|
84 |
+
},
|
85 |
+
"rope_theta": 1000000.0,
|
86 |
+
"sep_token_id": null,
|
87 |
+
"sliding_window": null,
|
88 |
+
"suppress_tokens": null,
|
89 |
+
"task_specific_params": null,
|
90 |
+
"temperature": 1.0,
|
91 |
+
"tf_legacy_loss": false,
|
92 |
+
"tie_encoder_decoder": false,
|
93 |
+
"tie_word_embeddings": false,
|
94 |
+
"tokenizer_class": null,
|
95 |
+
"top_k": 50,
|
96 |
+
"top_p": 1.0,
|
97 |
+
"torch_dtype": "bfloat16",
|
98 |
+
"torchscript": false,
|
99 |
+
"transformers_version": "4.48.3",
|
100 |
+
"typical_p": 1.0,
|
101 |
+
"use_bfloat16": true,
|
102 |
+
"use_cache": false,
|
103 |
+
"use_sliding_window": false,
|
104 |
+
"vocab_size": 151674
|
105 |
+
},
|
106 |
+
"max_dynamic_patch": 12,
|
107 |
+
"min_dynamic_patch": 1,
|
108 |
+
"model_type": "internvl_chat",
|
109 |
+
"pad2square": false,
|
110 |
+
"ps_version": "v2",
|
111 |
+
"select_layer": -1,
|
112 |
+
"system_message": null,
|
113 |
+
"template": "internvl2_5",
|
114 |
+
"tie_word_embeddings": false,
|
115 |
+
"torch_dtype": "bfloat16",
|
116 |
+
"transformers_version": null,
|
117 |
+
"use_backbone_lora": 0,
|
118 |
+
"use_llm_lora": 0,
|
119 |
+
"use_thumbnail": true,
|
120 |
+
"vision_config": {
|
121 |
+
"_attn_implementation_autoset": true,
|
122 |
+
"_name_or_path": "OpenGVLab/InternViT-6B-448px-V1-5",
|
123 |
+
"add_cross_attention": false,
|
124 |
+
"architectures": [
|
125 |
+
"InternVisionModel"
|
126 |
+
],
|
127 |
+
"attention_dropout": 0.0,
|
128 |
+
"auto_map": {
|
129 |
+
"AutoConfig": "configuration_intern_vit.InternVisionConfig",
|
130 |
+
"AutoModel": "modeling_intern_vit.InternVisionModel"
|
131 |
+
},
|
132 |
+
"bad_words_ids": null,
|
133 |
+
"begin_suppress_tokens": null,
|
134 |
+
"bos_token_id": null,
|
135 |
+
"capacity_factor": 1.2,
|
136 |
+
"chunk_size_feed_forward": 0,
|
137 |
+
"cross_attention_hidden_size": null,
|
138 |
+
"decoder_start_token_id": null,
|
139 |
+
"diversity_penalty": 0.0,
|
140 |
+
"do_sample": false,
|
141 |
+
"drop_path_rate": 0.1,
|
142 |
+
"dropout": 0.0,
|
143 |
+
"early_stopping": false,
|
144 |
+
"encoder_no_repeat_ngram_size": 0,
|
145 |
+
"eos_token_id": null,
|
146 |
+
"eval_capacity_factor": 1.4,
|
147 |
+
"exponential_decay_length_penalty": null,
|
148 |
+
"finetuning_task": null,
|
149 |
+
"forced_bos_token_id": null,
|
150 |
+
"forced_eos_token_id": null,
|
151 |
+
"hidden_act": "gelu",
|
152 |
+
"hidden_size": 1024,
|
153 |
+
"id2label": {
|
154 |
+
"0": "LABEL_0",
|
155 |
+
"1": "LABEL_1"
|
156 |
+
},
|
157 |
+
"image_size": 448,
|
158 |
+
"initializer_factor": 0.1,
|
159 |
+
"initializer_range": 1e-10,
|
160 |
+
"intermediate_size": 4096,
|
161 |
+
"is_decoder": false,
|
162 |
+
"is_encoder_decoder": false,
|
163 |
+
"label2id": {
|
164 |
+
"LABEL_0": 0,
|
165 |
+
"LABEL_1": 1
|
166 |
+
},
|
167 |
+
"laux_allreduce": "all_nodes",
|
168 |
+
"layer_norm_eps": 1e-06,
|
169 |
+
"length_penalty": 1.0,
|
170 |
+
"max_length": 20,
|
171 |
+
"min_length": 0,
|
172 |
+
"model_type": "intern_vit_6b",
|
173 |
+
"moe_coeff_ratio": 0.5,
|
174 |
+
"moe_intermediate_size": 768,
|
175 |
+
"moe_output_scale": 4.0,
|
176 |
+
"no_repeat_ngram_size": 0,
|
177 |
+
"noisy_gate_policy": "RSample_before",
|
178 |
+
"norm_type": "layer_norm",
|
179 |
+
"num_attention_heads": 16,
|
180 |
+
"num_beam_groups": 1,
|
181 |
+
"num_beams": 1,
|
182 |
+
"num_channels": 3,
|
183 |
+
"num_experts": 8,
|
184 |
+
"num_hidden_layers": 24,
|
185 |
+
"num_return_sequences": 1,
|
186 |
+
"num_routed_experts": 4,
|
187 |
+
"num_shared_experts": 4,
|
188 |
+
"output_attentions": false,
|
189 |
+
"output_hidden_states": false,
|
190 |
+
"output_scores": false,
|
191 |
+
"pad_token_id": null,
|
192 |
+
"patch_size": 14,
|
193 |
+
"prefix": null,
|
194 |
+
"problem_type": null,
|
195 |
+
"pruned_heads": {},
|
196 |
+
"qk_normalization": false,
|
197 |
+
"qkv_bias": true,
|
198 |
+
"remove_invalid_values": false,
|
199 |
+
"repetition_penalty": 1.0,
|
200 |
+
"return_dict": true,
|
201 |
+
"return_dict_in_generate": false,
|
202 |
+
"sep_token_id": null,
|
203 |
+
"shared_expert_intermediate_size": 3072,
|
204 |
+
"suppress_tokens": null,
|
205 |
+
"task_specific_params": null,
|
206 |
+
"temperature": 1.0,
|
207 |
+
"tf_legacy_loss": false,
|
208 |
+
"tie_encoder_decoder": false,
|
209 |
+
"tie_word_embeddings": true,
|
210 |
+
"tokenizer_class": null,
|
211 |
+
"top_k": 50,
|
212 |
+
"top_p": 1.0,
|
213 |
+
"torch_dtype": "bfloat16",
|
214 |
+
"torchscript": false,
|
215 |
+
"transformers_version": "4.48.3",
|
216 |
+
"typical_p": 1.0,
|
217 |
+
"use_bfloat16": true,
|
218 |
+
"use_flash_attn": true,
|
219 |
+
"use_moe": false,
|
220 |
+
"use_residual": true,
|
221 |
+
"use_rts": false,
|
222 |
+
"use_weighted_residual": false
|
223 |
+
}
|
224 |
+
}
|
internvl3_tokenizer/generation_config.json
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_from_model_config": true,
|
3 |
+
"transformers_version": "4.48.3"
|
4 |
+
}
|
internvl3_tokenizer/merges.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
internvl3_tokenizer/preprocessor_config.json
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"crop_size": 448,
|
3 |
+
"do_center_crop": true,
|
4 |
+
"do_normalize": true,
|
5 |
+
"do_resize": true,
|
6 |
+
"feature_extractor_type": "CLIPFeatureExtractor",
|
7 |
+
"image_mean": [
|
8 |
+
0.485,
|
9 |
+
0.456,
|
10 |
+
0.406
|
11 |
+
],
|
12 |
+
"image_std": [
|
13 |
+
0.229,
|
14 |
+
0.224,
|
15 |
+
0.225
|
16 |
+
],
|
17 |
+
"resample": 3,
|
18 |
+
"size": 448
|
19 |
+
}
|
internvl3_tokenizer/special_tokens_map.json
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"additional_special_tokens": [
|
3 |
+
"<|im_start|>",
|
4 |
+
"<|im_end|>",
|
5 |
+
"<|object_ref_start|>",
|
6 |
+
"<|object_ref_end|>",
|
7 |
+
"<|box_start|>",
|
8 |
+
"<|box_end|>",
|
9 |
+
"<|quad_start|>",
|
10 |
+
"<|quad_end|>",
|
11 |
+
"<|vision_start|>",
|
12 |
+
"<|vision_end|>",
|
13 |
+
"<|vision_pad|>",
|
14 |
+
"<|image_pad|>",
|
15 |
+
"<|video_pad|>"
|
16 |
+
],
|
17 |
+
"eos_token": {
|
18 |
+
"content": "<|im_end|>",
|
19 |
+
"lstrip": false,
|
20 |
+
"normalized": false,
|
21 |
+
"rstrip": false,
|
22 |
+
"single_word": false
|
23 |
+
},
|
24 |
+
"pad_token": {
|
25 |
+
"content": "<|endoftext|>",
|
26 |
+
"lstrip": false,
|
27 |
+
"normalized": false,
|
28 |
+
"rstrip": false,
|
29 |
+
"single_word": false
|
30 |
+
}
|
31 |
+
}
|
internvl3_tokenizer/tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
internvl3_tokenizer/tokenizer_config.json
ADDED
@@ -0,0 +1,281 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": false,
|
3 |
+
"add_eos_token": false,
|
4 |
+
"add_prefix_space": false,
|
5 |
+
"added_tokens_decoder": {
|
6 |
+
"151643": {
|
7 |
+
"content": "<|endoftext|>",
|
8 |
+
"lstrip": false,
|
9 |
+
"normalized": false,
|
10 |
+
"rstrip": false,
|
11 |
+
"single_word": false,
|
12 |
+
"special": true
|
13 |
+
},
|
14 |
+
"151644": {
|
15 |
+
"content": "<|im_start|>",
|
16 |
+
"lstrip": false,
|
17 |
+
"normalized": false,
|
18 |
+
"rstrip": false,
|
19 |
+
"single_word": false,
|
20 |
+
"special": true
|
21 |
+
},
|
22 |
+
"151645": {
|
23 |
+
"content": "<|im_end|>",
|
24 |
+
"lstrip": false,
|
25 |
+
"normalized": false,
|
26 |
+
"rstrip": false,
|
27 |
+
"single_word": false,
|
28 |
+
"special": true
|
29 |
+
},
|
30 |
+
"151646": {
|
31 |
+
"content": "<|object_ref_start|>",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": false,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false,
|
36 |
+
"special": true
|
37 |
+
},
|
38 |
+
"151647": {
|
39 |
+
"content": "<|object_ref_end|>",
|
40 |
+
"lstrip": false,
|
41 |
+
"normalized": false,
|
42 |
+
"rstrip": false,
|
43 |
+
"single_word": false,
|
44 |
+
"special": true
|
45 |
+
},
|
46 |
+
"151648": {
|
47 |
+
"content": "<|box_start|>",
|
48 |
+
"lstrip": false,
|
49 |
+
"normalized": false,
|
50 |
+
"rstrip": false,
|
51 |
+
"single_word": false,
|
52 |
+
"special": true
|
53 |
+
},
|
54 |
+
"151649": {
|
55 |
+
"content": "<|box_end|>",
|
56 |
+
"lstrip": false,
|
57 |
+
"normalized": false,
|
58 |
+
"rstrip": false,
|
59 |
+
"single_word": false,
|
60 |
+
"special": true
|
61 |
+
},
|
62 |
+
"151650": {
|
63 |
+
"content": "<|quad_start|>",
|
64 |
+
"lstrip": false,
|
65 |
+
"normalized": false,
|
66 |
+
"rstrip": false,
|
67 |
+
"single_word": false,
|
68 |
+
"special": true
|
69 |
+
},
|
70 |
+
"151651": {
|
71 |
+
"content": "<|quad_end|>",
|
72 |
+
"lstrip": false,
|
73 |
+
"normalized": false,
|
74 |
+
"rstrip": false,
|
75 |
+
"single_word": false,
|
76 |
+
"special": true
|
77 |
+
},
|
78 |
+
"151652": {
|
79 |
+
"content": "<|vision_start|>",
|
80 |
+
"lstrip": false,
|
81 |
+
"normalized": false,
|
82 |
+
"rstrip": false,
|
83 |
+
"single_word": false,
|
84 |
+
"special": true
|
85 |
+
},
|
86 |
+
"151653": {
|
87 |
+
"content": "<|vision_end|>",
|
88 |
+
"lstrip": false,
|
89 |
+
"normalized": false,
|
90 |
+
"rstrip": false,
|
91 |
+
"single_word": false,
|
92 |
+
"special": true
|
93 |
+
},
|
94 |
+
"151654": {
|
95 |
+
"content": "<|vision_pad|>",
|
96 |
+
"lstrip": false,
|
97 |
+
"normalized": false,
|
98 |
+
"rstrip": false,
|
99 |
+
"single_word": false,
|
100 |
+
"special": true
|
101 |
+
},
|
102 |
+
"151655": {
|
103 |
+
"content": "<|image_pad|>",
|
104 |
+
"lstrip": false,
|
105 |
+
"normalized": false,
|
106 |
+
"rstrip": false,
|
107 |
+
"single_word": false,
|
108 |
+
"special": true
|
109 |
+
},
|
110 |
+
"151656": {
|
111 |
+
"content": "<|video_pad|>",
|
112 |
+
"lstrip": false,
|
113 |
+
"normalized": false,
|
114 |
+
"rstrip": false,
|
115 |
+
"single_word": false,
|
116 |
+
"special": true
|
117 |
+
},
|
118 |
+
"151657": {
|
119 |
+
"content": "<tool_call>",
|
120 |
+
"lstrip": false,
|
121 |
+
"normalized": false,
|
122 |
+
"rstrip": false,
|
123 |
+
"single_word": false,
|
124 |
+
"special": false
|
125 |
+
},
|
126 |
+
"151658": {
|
127 |
+
"content": "</tool_call>",
|
128 |
+
"lstrip": false,
|
129 |
+
"normalized": false,
|
130 |
+
"rstrip": false,
|
131 |
+
"single_word": false,
|
132 |
+
"special": false
|
133 |
+
},
|
134 |
+
"151659": {
|
135 |
+
"content": "<|fim_prefix|>",
|
136 |
+
"lstrip": false,
|
137 |
+
"normalized": false,
|
138 |
+
"rstrip": false,
|
139 |
+
"single_word": false,
|
140 |
+
"special": false
|
141 |
+
},
|
142 |
+
"151660": {
|
143 |
+
"content": "<|fim_middle|>",
|
144 |
+
"lstrip": false,
|
145 |
+
"normalized": false,
|
146 |
+
"rstrip": false,
|
147 |
+
"single_word": false,
|
148 |
+
"special": false
|
149 |
+
},
|
150 |
+
"151661": {
|
151 |
+
"content": "<|fim_suffix|>",
|
152 |
+
"lstrip": false,
|
153 |
+
"normalized": false,
|
154 |
+
"rstrip": false,
|
155 |
+
"single_word": false,
|
156 |
+
"special": false
|
157 |
+
},
|
158 |
+
"151662": {
|
159 |
+
"content": "<|fim_pad|>",
|
160 |
+
"lstrip": false,
|
161 |
+
"normalized": false,
|
162 |
+
"rstrip": false,
|
163 |
+
"single_word": false,
|
164 |
+
"special": false
|
165 |
+
},
|
166 |
+
"151663": {
|
167 |
+
"content": "<|repo_name|>",
|
168 |
+
"lstrip": false,
|
169 |
+
"normalized": false,
|
170 |
+
"rstrip": false,
|
171 |
+
"single_word": false,
|
172 |
+
"special": false
|
173 |
+
},
|
174 |
+
"151664": {
|
175 |
+
"content": "<|file_sep|>",
|
176 |
+
"lstrip": false,
|
177 |
+
"normalized": false,
|
178 |
+
"rstrip": false,
|
179 |
+
"single_word": false,
|
180 |
+
"special": false
|
181 |
+
},
|
182 |
+
"151665": {
|
183 |
+
"content": "<img>",
|
184 |
+
"lstrip": false,
|
185 |
+
"normalized": false,
|
186 |
+
"rstrip": false,
|
187 |
+
"single_word": false,
|
188 |
+
"special": true
|
189 |
+
},
|
190 |
+
"151666": {
|
191 |
+
"content": "</img>",
|
192 |
+
"lstrip": false,
|
193 |
+
"normalized": false,
|
194 |
+
"rstrip": false,
|
195 |
+
"single_word": false,
|
196 |
+
"special": true
|
197 |
+
},
|
198 |
+
"151667": {
|
199 |
+
"content": "<IMG_CONTEXT>",
|
200 |
+
"lstrip": false,
|
201 |
+
"normalized": false,
|
202 |
+
"rstrip": false,
|
203 |
+
"single_word": false,
|
204 |
+
"special": true
|
205 |
+
},
|
206 |
+
"151668": {
|
207 |
+
"content": "<quad>",
|
208 |
+
"lstrip": false,
|
209 |
+
"normalized": false,
|
210 |
+
"rstrip": false,
|
211 |
+
"single_word": false,
|
212 |
+
"special": true
|
213 |
+
},
|
214 |
+
"151669": {
|
215 |
+
"content": "</quad>",
|
216 |
+
"lstrip": false,
|
217 |
+
"normalized": false,
|
218 |
+
"rstrip": false,
|
219 |
+
"single_word": false,
|
220 |
+
"special": true
|
221 |
+
},
|
222 |
+
"151670": {
|
223 |
+
"content": "<ref>",
|
224 |
+
"lstrip": false,
|
225 |
+
"normalized": false,
|
226 |
+
"rstrip": false,
|
227 |
+
"single_word": false,
|
228 |
+
"special": true
|
229 |
+
},
|
230 |
+
"151671": {
|
231 |
+
"content": "</ref>",
|
232 |
+
"lstrip": false,
|
233 |
+
"normalized": false,
|
234 |
+
"rstrip": false,
|
235 |
+
"single_word": false,
|
236 |
+
"special": true
|
237 |
+
},
|
238 |
+
"151672": {
|
239 |
+
"content": "<box>",
|
240 |
+
"lstrip": false,
|
241 |
+
"normalized": false,
|
242 |
+
"rstrip": false,
|
243 |
+
"single_word": false,
|
244 |
+
"special": true
|
245 |
+
},
|
246 |
+
"151673": {
|
247 |
+
"content": "</box>",
|
248 |
+
"lstrip": false,
|
249 |
+
"normalized": false,
|
250 |
+
"rstrip": false,
|
251 |
+
"single_word": false,
|
252 |
+
"special": true
|
253 |
+
}
|
254 |
+
},
|
255 |
+
"additional_special_tokens": [
|
256 |
+
"<|im_start|>",
|
257 |
+
"<|im_end|>",
|
258 |
+
"<|object_ref_start|>",
|
259 |
+
"<|object_ref_end|>",
|
260 |
+
"<|box_start|>",
|
261 |
+
"<|box_end|>",
|
262 |
+
"<|quad_start|>",
|
263 |
+
"<|quad_end|>",
|
264 |
+
"<|vision_start|>",
|
265 |
+
"<|vision_end|>",
|
266 |
+
"<|vision_pad|>",
|
267 |
+
"<|image_pad|>",
|
268 |
+
"<|video_pad|>"
|
269 |
+
],
|
270 |
+
"bos_token": null,
|
271 |
+
"chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0]['role'] == 'system' %}\n {{- messages[0]['content'] }}\n {%- else %}\n {{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}\n {%- endif %}\n {{- \"\\n\\n# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n {%- for tool in tools %}\n {{- \"\\n\" }}\n {{- tool | tojson }}\n {%- endfor %}\n {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n {%- if messages[0]['role'] == 'system' %}\n {{- '<|im_start|>system\\n' + messages[0]['content'] + '<|im_end|>\\n' }}\n {%- else %}\n {{- '<|im_start|>system\\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- for message in messages %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) or (message.role == \"assistant\" and not message.tool_calls) %}\n {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {{- '<|im_start|>' + message.role }}\n {%- if message.content %}\n {{- '\\n' + message.content }}\n {%- endif %}\n {%- for tool_call in message.tool_calls %}\n {%- if tool_call.function is defined %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '\\n<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {{- tool_call.arguments | tojson }}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != \"tool\") %}\n {{- '<|im_start|>user' }}\n {%- endif %}\n {{- '\\n<tool_response>\\n' }}\n {{- message.content }}\n {{- '\\n</tool_response>' }}\n {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|im_start|>assistant\\n' }}\n{%- endif %}\n",
|
272 |
+
"clean_up_tokenization_spaces": false,
|
273 |
+
"eos_token": "<|im_end|>",
|
274 |
+
"errors": "replace",
|
275 |
+
"extra_special_tokens": {},
|
276 |
+
"model_max_length": 12288,
|
277 |
+
"pad_token": "<|endoftext|>",
|
278 |
+
"split_special_tokens": false,
|
279 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
280 |
+
"unk_token": null
|
281 |
+
}
|
internvl3_tokenizer/vocab.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
main_api_ax650
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9fc74b9dcf37c2b75d1894fa0f0ecaffe741b11f13a58dda0d30ff1b4ca8d508
|
3 |
+
size 6667616
|
main_api_axcl_x86
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:849231c5e30faa54cf62a55022a45b42ed767a7feb52ea26483e5c70eb9073dd
|
3 |
+
size 1927840
|
main_ax650
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2b0bc06d4c78f592ae546a9f91f260dd62da7bf6b41a8f8a24a7a648d58c7f72
|
3 |
+
size 6568824
|
main_axcl_x86
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:273bac369315774e6e6df5e725647c49e5356f22cc512f1daadfd58066a37ef0
|
3 |
+
size 1840560
|
post_config.json
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"enable_temperature" : true,
|
3 |
+
"temperature" : 0.9,
|
4 |
+
|
5 |
+
"enable_repetition_penalty" : false,
|
6 |
+
"repetition_penalty" : 1.2,
|
7 |
+
"penalty_window" : 20,
|
8 |
+
|
9 |
+
"enable_top_p_sampling" : false,
|
10 |
+
"top_p" : 0.8,
|
11 |
+
|
12 |
+
"enable_top_k_sampling" : true,
|
13 |
+
"top_k" : 10
|
14 |
+
}
|
run_internvl_3_1b_448_api_ax650.sh
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
./main_api_ax650 \
|
2 |
+
--template_filename_axmodel "./internvl3_1b_ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
+
--axmodel_num 24 \
|
4 |
+
--filename_image_encoder_axmodedl "./internvl3_1b_ax650/internvl3_1b_vit.axmodel" \
|
5 |
+
--use_mmap_load_embed 1 \
|
6 |
+
--filename_tokenizer_model "http://0.0.0.0:12345" \
|
7 |
+
--filename_post_axmodel "./internvl3_1b_ax650/qwen2_post.axmodel" \
|
8 |
+
--filename_tokens_embed "./internvl3_1b_ax650/model.embed_tokens.weight.bfloat16.bin" \
|
9 |
+
--tokens_embed_num 151674 \
|
10 |
+
--tokens_embed_size 896 \
|
run_internvl_3_1b_448_api_axcl_x86.sh
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
./main_api_axcl_x86 \
|
2 |
+
--template_filename_axmodel "./internvl3_1b_ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
+
--axmodel_num 24 \
|
4 |
+
--filename_image_encoder_axmodedl "./internvl3_1b_ax650/internvl3_1b_vit.axmodel" \
|
5 |
+
--use_mmap_load_embed 1 \
|
6 |
+
--filename_tokenizer_model "http://0.0.0.0:12345" \
|
7 |
+
--filename_post_axmodel "./internvl3_1b_ax650/qwen2_post.axmodel" \
|
8 |
+
--filename_tokens_embed "./internvl3_1b_ax650/model.embed_tokens.weight.bfloat16.bin" \
|
9 |
+
--tokens_embed_num 151674 \
|
10 |
+
--tokens_embed_size 896 \
|
11 |
+
--devices 0,2,4 \
|
run_internvl_3_1b_448_ax650.sh
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
./main_ax650 \
|
2 |
+
--template_filename_axmodel "./internvl3_1b_ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
+
--axmodel_num 24 \
|
4 |
+
--filename_image_encoder_axmodedl "./internvl3_1b_ax650/internvl3_1b_vit.axmodel" \
|
5 |
+
--use_mmap_load_embed 1 \
|
6 |
+
--filename_tokenizer_model "http://0.0.0.0:12345" \
|
7 |
+
--filename_post_axmodel "./internvl3_1b_ax650/qwen2_post.axmodel" \
|
8 |
+
--filename_tokens_embed "./internvl3_1b_ax650/model.embed_tokens.weight.bfloat16.bin" \
|
9 |
+
--tokens_embed_num 151674 \
|
10 |
+
--tokens_embed_size 896 \
|
11 |
+
--live_print 1
|
run_internvl_3_1b_448_axcl_x86.sh
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
./main_axcl_x86 \
|
2 |
+
--template_filename_axmodel "./internvl3_1b_ax650/qwen2_p128_l%d_together.axmodel" \
|
3 |
+
--axmodel_num 24 \
|
4 |
+
--filename_image_encoder_axmodedl "./internvl3_1b_ax650/internvl3_1b_vit.axmodel" \
|
5 |
+
--use_mmap_load_embed 1 \
|
6 |
+
--filename_tokenizer_model "http://0.0.0.0:12345" \
|
7 |
+
--filename_post_axmodel "./internvl3_1b_ax650/qwen2_post.axmodel" \
|
8 |
+
--filename_tokens_embed "./internvl3_1b_ax650/model.embed_tokens.weight.bfloat16.bin" \
|
9 |
+
--tokens_embed_num 151674 \
|
10 |
+
--tokens_embed_size 896 \
|
11 |
+
--devices 0,2,4 \
|
12 |
+
--live_print 1
|