Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
change
sglang
Commits
03e04b23
Commit
03e04b23
authored
Feb 01, 2024
by
Lianmin Zheng
Browse files
update docs for Yi-VL
parent
86442530
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
7 deletions
+6
-7
README.md
README.md
+3
-1
examples/quick_start/srt_example_yi_vl.py
examples/quick_start/srt_example_yi_vl.py
+3
-6
No files found.
README.md
View file @
03e04b23
...
...
@@ -357,9 +357,11 @@ python -m sglang.launch_server --model-path meta-llama/Llama-2-7b-chat-hf --port
-
Llama
-
Mistral
-
Mixtral
-
Qwen / Qwen 2
-
LLaVA
-
`python3 -m sglang.launch_server --model-path liuhaotian/llava-v1.5-7b --tokenizer-path llava-hf/llava-1.5-7b-hf --chat-template vicuna_v1.1 --port 30000`
-
Qwen / Qwen 2
-
Yi-VL
-
see
[
srt_example_yi_vl.py
](
examples/quick_start/srt_example_yi_vl.py
)
.
-
AWQ quantization
## Benchmark And Performance
...
...
examples/quick_start/srt_example_yi_vl.py
View file @
03e04b23
...
...
@@ -46,12 +46,9 @@ def batch():
if
__name__
==
"__main__"
:
runtime
=
sgl
.
Runtime
(
model_path
=
"BabyChou/Yi-VL-6B"
,
tokenizer
_path
=
"BabyChou/Yi-VL-
6
B"
)
runtime
=
sgl
.
Runtime
(
model_path
=
"BabyChou/Yi-VL-6B"
)
# runtime = sgl.Runtime(model
_path="BabyChou/Yi-VL-
34
B")
sgl
.
set_default_backend
(
runtime
)
# Or you can use API models
# sgl.set_default_backend(sgl.OpenAI("gpt-4-vision-preview"))
# sgl.set_default_backend(sgl.VertexAI("gemini-pro-vision"))
# Run a single request
print
(
"
\n
========== single ==========
\n
"
)
...
...
@@ -65,4 +62,4 @@ if __name__ == "__main__":
print
(
"
\n
========== batch ==========
\n
"
)
batch
()
runtime
.
shutdown
()
\ No newline at end of file
runtime
.
shutdown
()
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment