diff --git a/README.md b/README.md index 3833910..e805b61 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,7 @@ Standalone Python package for the MLX port of IBM Granite Speech 4.1-2b-plus. The default model is -[`mlx-community/granite-speech-4.1-2b-plus-mlx`](https://huggingface.co/mlx-community/granite-speech-4.1-2b-plus-mlx). +[`olivius/granite-speech-4.1-2b-plus-mlx`](https://huggingface.co/olivius/granite-speech-4.1-2b-plus-mlx). ## Quickstart diff --git a/scripts/upload_to_hf.py b/scripts/upload_to_hf.py index 21dcedc..b69855d 100755 --- a/scripts/upload_to_hf.py +++ b/scripts/upload_to_hf.py @@ -11,7 +11,7 @@ SOURCE_CACHE = ( Path.home() / ".cache/huggingface/hub/models--ibm-granite--granite-speech-4.1-2b-plus" ) -DEST_REPO = "mlx-community/granite-speech-4.1-2b-plus-mlx" +DEST_REPO = "olivius/granite-speech-4.1-2b-plus-mlx" def find_weights_dir(root: Path) -> Path | None: diff --git a/src/granite_speech_plus_mlx/pipeline.py b/src/granite_speech_plus_mlx/pipeline.py index 819df41..8ff778a 100644 --- a/src/granite_speech_plus_mlx/pipeline.py +++ b/src/granite_speech_plus_mlx/pipeline.py @@ -9,7 +9,7 @@ from typing import Any from .chunking import chunk_audio, prefix_text from .prompts import GRANITE_SYSTEM_PROMPT, PROMPT_MODES, build_prompt -DEFAULT_MODEL = "mlx-community/granite-speech-4.1-2b-plus-mlx" +DEFAULT_MODEL = "olivius/granite-speech-4.1-2b-plus-mlx" @dataclass