cicdatopea commited on
Commit
e35c422
1 Parent(s): 0b5077e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -51,7 +51,7 @@ prepare_inputs = vl_chat_processor(
51
  images=[pil_images],
52
  force_batchify=True,
53
  system_prompt=""
54
- )
55
  prepare_inputs = prepare_inputs.to(vl_gpt.device)
56
 
57
  # run image encoder to get the image embeddings
 
51
  images=[pil_images],
52
  force_batchify=True,
53
  system_prompt=""
54
+ ).to(vl_gpt.device)
55
  prepare_inputs = prepare_inputs.to(vl_gpt.device)
56
 
57
  # run image encoder to get the image embeddings