|
--- |
|
{} |
|
--- |
|
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") |
