Commit d0ec9274 authored by laibao's avatar laibao
Browse files

Update llava_example.py

parent 54a9b020
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)
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