Instructions to use OFA-Sys/ofa-tiny with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use OFA-Sys/ofa-tiny with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("OFA-Sys/ofa-tiny", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
31aa84f
1
Parent(s): 74391db
Update README.md
Browse files
README.md
CHANGED
|
@@ -30,5 +30,5 @@ To use it in Transformers, please refer to https://github.com/OFA-Sys/OFA/tree/f
|
|
| 30 |
>>> patch_img = patch_resize_transform(img).unsqueeze(0)
|
| 31 |
|
| 32 |
>>> gen = model.generate(inputs, patch_images=patch_img, num_beams=4)
|
| 33 |
-
>>> print(tokenizer.
|
| 34 |
```
|
|
|
|
| 30 |
>>> patch_img = patch_resize_transform(img).unsqueeze(0)
|
| 31 |
|
| 32 |
>>> gen = model.generate(inputs, patch_images=patch_img, num_beams=4)
|
| 33 |
+
>>> print(tokenizer.batch_decode(gen, skip_special_tokens=True))
|
| 34 |
```
|