eltorio
/

IDEFICS3_ROCO

Image-Text-to-Text

Model card Files Files and versions Community

æLtorio commited on Nov 13, 2024

Commit

5cbfb5c

·

unverified ·

1 Parent(s): 6ff184e

update tl;dr

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -30,8 +30,10 @@ This repository contains a fine-tuned version of the Hugging Face [Idefics3-8B-L
 For immediate use, you can load the model directly from Hugging Face:
 ```python
 from transformers import AutoModelForImageTextToText
-model =AutoModelForImageTextToText.from_pretrained("eltorio/IDEFICS3_ROCO")
 ```
 ### Model Information

 For immediate use, you can load the model directly from Hugging Face:
 ```python
+import torch
 from transformers import AutoModelForImageTextToText
+device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
+model = AutoModelForImageTextToText.from_pretrained("eltorio/IDEFICS3_ROCO").to(device)
 ```
 ### Model Information