Add note about training from adapter
Browse files
README.md
CHANGED
@@ -123,6 +123,18 @@ if __name__ == "__main__":
|
|
123 |
|
124 |
```
|
125 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
126 |
## Limitations
|
127 |
|
128 |
- **Focus**: The model primarily focuses on PDF-type documents and high-ressources languages, potentially limiting its generalization to other document types or less represented languages.
|
|
|
123 |
|
124 |
```
|
125 |
|
126 |
+
**Note:** If you need to further train ColPali from this adapter, you should run:
|
127 |
+
|
128 |
+
```python
|
129 |
+
lora_config = LoraConfig.from_pretrained("vidore/colpali-v1.1")
|
130 |
+
lora_config.inference_mode = False # force training mode for fine-tuning
|
131 |
+
|
132 |
+
model = get_peft_model(model, lora_config)
|
133 |
+
|
134 |
+
print("after")
|
135 |
+
model.print_trainable_parameters()
|
136 |
+
```
|
137 |
+
|
138 |
## Limitations
|
139 |
|
140 |
- **Focus**: The model primarily focuses on PDF-type documents and high-ressources languages, potentially limiting its generalization to other document types or less represented languages.
|