diff --git a/CHANGELOG.md b/CHANGELOG.md index 868c9568c..d59c8f3ec 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ### Changed +- `MoondreamService` argument `model_id` is now `model`. + - `VADAnalyzer` arguments have been renamed for more clarity. ## [0.0.12] - 2024-05-14 diff --git a/src/pipecat/services/moondream.py b/src/pipecat/services/moondream.py index 01084a108..444dea1ed 100644 --- a/src/pipecat/services/moondream.py +++ b/src/pipecat/services/moondream.py @@ -46,7 +46,7 @@ def detect_device(): class MoondreamService(VisionService): def __init__( self, - model_id="vikhyatk/moondream2", + model="vikhyatk/moondream2", revision="2024-04-02", use_cpu=False ): @@ -58,12 +58,12 @@ class MoondreamService(VisionService): device = torch.device("cpu") dtype = torch.float32 - self._tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision) + self._tokenizer = AutoTokenizer.from_pretrained(model, revision=revision) logger.debug("Loading Moondream model...") self._model = AutoModelForCausalLM.from_pretrained( - model_id, trust_remote_code=True, revision=revision + model, trust_remote_code=True, revision=revision ).to(device=device, dtype=dtype) self._model.eval()