TobyYang7 commited on
Commit
bbbdbee
1 Parent(s): 7dc477a

Update llava_llama3/serve/cli.py

Browse files
Files changed (1) hide show
  1. llava_llama3/serve/cli.py +3 -0
llava_llama3/serve/cli.py CHANGED
@@ -33,7 +33,10 @@ def chat_llava(args, image_file, text, tokenizer, model, image_processor, contex
33
  conv = conv_templates[args.conv_mode].copy()
34
  roles = conv.roles
35
 
 
36
  image = load_image(image_file)
 
 
37
  image_size = image.size
38
  image_tensor = process_images([image], image_processor, model.config)
39
  if type(image_tensor) is list:
 
33
  conv = conv_templates[args.conv_mode].copy()
34
  roles = conv.roles
35
 
36
+ print(image_file,type(image_file))
37
  image = load_image(image_file)
38
+ print(image,type(image))
39
+
40
  image_size = image.size
41
  image_tensor = process_images([image], image_processor, model.config)
42
  if type(image_tensor) is list: