Update README.md
Browse files
README.md
CHANGED
|
@@ -51,3 +51,12 @@ The following hyperparameters were used during training:
|
|
| 51 |
- Pytorch 2.5.1+cu124
|
| 52 |
- Datasets 3.1.0
|
| 53 |
- Tokenizers 0.20.3
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
- Pytorch 2.5.1+cu124
|
| 52 |
- Datasets 3.1.0
|
| 53 |
- Tokenizers 0.20.3
|
| 54 |
+
|
| 55 |
+
### USAGE
|
| 56 |
+
#Load your fine-tuned ColPali
|
| 57 |
+
|
| 58 |
+
model = ColQwen2.from_pretrained(
|
| 59 |
+
'malya7/toxic_pandas',
|
| 60 |
+
torch_dtype=torch.bfloat16,
|
| 61 |
+
device_map=device,
|
| 62 |
+
)
|