The model card for Kosmos 2.5 has been updated to address an issue where the dtype parameter in the from_pretrained method was not functioning as intended. The code has been revised to use torch_dtype instead, aligning with the standard PyTorch API convention in Hugging Face's Transformers library. This change ensures proper data type specification during model loading and resolves issues with the previous dtype parameter.
The updated code is as follows:
```
model = Kosmos2_5ForConditionalGeneration.from_pretrained(repo, device_map=device, torch_dtype=dtype)
```
This update does not alter the model's functionality but ensures compatibility with the Hugging Face Transformers API.
Add code snippets, library name and pipeline tag (#18)
- Add code snippets, library name and pipeline tag (21ab46e9261f3c32615e400e3c3f7e67cd227ddf)
Co-authored-by: Niels Rogge <nielsr@users.noreply.huggingface.co>
