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