bump version and deps

viam-labs · Dec 4, 2024 · 8badd3a · 8badd3a
1 parent 2c83133
commit 8badd3a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -37,13 +37,13 @@ The following attributes are available for `viam-labs:vision:yolov8` model:
 
 | Name | Type | Inclusion | Description |
 | ---- | ---- | --------- | ----------- |
-| `revision` | string | **Required** |  Moondream model revision, defaults to "2024-03-06" |
+| `revision` | string | **Required** |  Moondream model revision, defaults to "2024-08-26" |
 
 ### Example Configurations
 
 ```json
 {
-  "revision": "2024-03-06"
+  "revision": "2024-08-26"
 }
 ```
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,7 @@
-viam-sdk~=0.21.0rc1
+viam-sdk~=0.34.0
 transformers
 timm
 einops
-pillow
+pillow
+torch
+torchvision
diff --git a/src/moondream.py b/src/moondream.py
@@ -56,7 +56,7 @@ def validate(cls, config: ComponentConfig):
     def reconfigure(self, config: ComponentConfig, dependencies: Mapping[ResourceName, ResourceBase]):
         self.DEPS = dependencies
         model_id = "vikhyatk/moondream2"
-        revision = config.attributes.fields["revision"].string_value or "2024-03-06"
+        revision = config.attributes.fields["revision"].string_value or "2024-08-26"
         self.model = AutoModelForCausalLM.from_pretrained(
             model_id, trust_remote_code=True, revision=revision
         )