Hugging Face's logo

microsoft
/
kosmos-2-patch14-224

Image-to-Text
Transformers
PyTorch
Safetensors
kosmos-2
image-text-to-text
image-captioning
Inference Endpoints
Model card Files Community
19

To run on the GPU

#16
by jshenoy - opened
Discussion
jshenoy

If you want it to run on the GPU, you can just do

inputs.to('cuda:0')
model.to('cuda:0')
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment