fix(model): fix missing package in test models (#552)

Because - Some test models are missing required packages This commit - add missing packages in model config
instill-ai · Apr 17, 2024 · a28a21b · a28a21b
1 parent 69f223a
commit a28a21b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/assets/model-dummy-image-to-image/instill.yaml b/assets/model-dummy-image-to-image/instill.yaml
@@ -2,5 +2,6 @@ build:
   gpu: false
   python_version: "3.11"  # support only 3.11
   python_packages:
+    - torch==2.2.1
 repo: instill/dummy-image-to-image
 tag: test
diff --git a/assets/model-dummy-text-to-image/instill.yaml b/assets/model-dummy-text-to-image/instill.yaml
@@ -2,5 +2,6 @@ build:
   gpu: false
   python_version: "3.11"  # support only 3.11
   python_packages:
+    - torch==2.2.1
 repo: instill/dummy-text-to-image
 tag: test