Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
rchalamala authored Jun 6, 2024
1 parent b06ff08 commit 357c008
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,7 @@ images = [image]

text_prompt = "<|start_header_id|>user<|end_header_id|>\n\nSummarize the visual content of the image.<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
inputs = processor(text=[text_prompt], images=[image], max_length=2048, return_tensors="pt", is_generate=True)
inputs = inputs.to(f"cuda:0")
inputs = inputs.to("cuda:0")
```

Finally, let us generate the responses from the model
Expand Down

0 comments on commit 357c008

Please sign in to comment.