ColPali
Safetensors
English
vidore
vidore-experimental
tonywu71 commited on
Commit
147f75a
1 Parent(s): b9631dd

Add note about training from adapter

Browse files
Files changed (1) hide show
  1. README.md +12 -0
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.