Fix CLAP processor: audios -> audio kwarg
transformers 5.x renamed the parameter. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -40,7 +40,7 @@ def embed_audio(audio: np.ndarray, sample_rate: int = 48000) -> np.ndarray:
|
|||||||
if _model is None or _processor is None:
|
if _model is None or _processor is None:
|
||||||
raise RuntimeError("CLAP model not loaded. Call load_model() first.")
|
raise RuntimeError("CLAP model not loaded. Call load_model() first.")
|
||||||
|
|
||||||
inputs = _processor(audios=audio, sampling_rate=sample_rate, return_tensors="pt")
|
inputs = _processor(audio=audio, sampling_rate=sample_rate, return_tensors="pt")
|
||||||
with torch.no_grad():
|
with torch.no_grad():
|
||||||
embeddings = _model.get_audio_features(**inputs)
|
embeddings = _model.get_audio_features(**inputs)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user