msnowchanj/one-peace-al-retrieval

ONE-PEACE Audio-Language Retrieval Checkpoint

Original repository: OFA-Sys/ONE-PEACE

This repository hosts the official ONE-PEACE finetuned checkpoint for Audio-Language Retrieval on AudioCaps & Clotho.

Usage

from huggingface_hub import hf_hub_download
from one_peace.models import from_pretrained

ckpt = hf_hub_download(
    repo_id="yourname/one-peace-al-retrieval",
    filename="finetune_al_retrieval.pt"
)

model = from_pretrained(
    ckpt,
    model_type="one_peace_retrieval",
    device="cuda",
    dtype="float32"
)