revert image processing

Chomolungma · Oct 12, 2023 · 2484462 · 2484462
1 parent 627d92f
commit 2484462
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/llava/serve/cli.py b/llava/serve/cli.py
@@ -52,6 +52,7 @@ def main(args):
         roles = conv.roles
 
     image = load_image(args.image_file)
+    image_tensor = image_processor.preprocess(image, return_tensors='pt')['pixel_values'].half().cuda()
 
     while True:
         try: