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
ModelZoo
LLaVA_vllm
Commits
d0ec9274
Commit
d0ec9274
authored
Nov 18, 2024
by
laibao
Browse files
Update llava_example.py
parent
54a9b020
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
10 deletions
+10
-10
examples/llava_example.py
examples/llava_example.py
+10
-10
No files found.
examples/llava_example.py
View file @
d0ec9274
import
argparse
import
os
import
subprocess
#
import subprocess
import
torch
from
PIL
import
Image
...
...
@@ -80,19 +80,19 @@ if __name__ == "__main__":
help
=
"image input type"
)
args
=
parser
.
parse_args
()
# Download from s3
s3_bucket_path
=
"s3://air-example-data-2/vllm_opensource_llava/"
#
s3_bucket_path = "s3://air-example-data-2/vllm_opensource_llava/"
local_directory
=
"images"
# Make sure the local directory exists or create it
os
.
makedirs
(
local_directory
,
exist_ok
=
True
)
# Use AWS CLI to sync the directory, assume anonymous access
subprocess
.
check_call
([
"aws"
,
"s3"
,
"sync"
,
s3_bucket_path
,
local_directory
,
"--no-sign-request"
,
])
#
subprocess.check_call([
#
"aws",
#
"s3",
#
"sync",
#
s3_bucket_path,
#
local_directory,
#
"--no-sign-request",
#
])
main
(
args
)
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