| from colpali_engine.models import ColQwen2, ColQwen2Processor | |
| from peft import PeftModel | |
| merged_model = ColQwen2.from_pretrained( | |
| "michaelfeil/colqwen2-v0.1", | |
| torch_dtype="bfloat16", | |
| ) | |
| merged_model = PeftModel.from_pretrained(model, "vidore/colqwen2-v0.1") | |
| merged_model: ColQwen2 = merged_model.merge_and_unload() | |
| processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v0.1") | |
| processor.push_to_hub("michaelfeil/colqwen2-v0.1") | |
| # merged_model.save_pretrained("mf-colqwen2-v0.1",max_shard_size = "2GB",) | |
| merged_model.push_to_hub("michaelfeil/colqwen2-v0.1", max_shard_size="2GB") |