Skip to content

GitLab

Commit d0ec9274 authored by laibao's avatar laibao
Browse files

Update llava_example.py

parent 54a9b020
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 10 deletions
examples/llava_example.py
View file @ d0ec9274
import argparse import argparse
import os import os
import subprocess # import subprocess
import torch import torch
from PIL import Image from PIL import Image
...@@ -80,19 +80,19 @@ if __name__ == "__main__": ...@@ -80,19 +80,19 @@ if __name__ == "__main__":
help="image input type") help="image input type")
args = parser.parse_args() args = parser.parse_args()
# Download from s3 # 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" local_directory = "images"
# Make sure the local directory exists or create it # Make sure the local directory exists or create it
os.makedirs(local_directory, exist_ok=True) os.makedirs(local_directory, exist_ok=True)
# Use AWS CLI to sync the directory, assume anonymous access # Use AWS CLI to sync the directory, assume anonymous access
subprocess.check_call([ # subprocess.check_call([
"aws", # "aws",
"s3", # "s3",
"sync", # "sync",
s3_bucket_path, # s3_bucket_path,
local_directory, # local_directory,
"--no-sign-request", # "--no-sign-request",
]) # ])
main(args) main(args)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment