fix(ollama): classify 'rocm' as its own cpu + update classifier

webinstall · Sep 15, 2024 · d5489d1 · d5489d1
1 parent efe3df6
commit d5489d1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/_webi/build-classifier b/_webi/build-classifier
diff --git a/ollama/releases.js b/ollama/releases.js
@@ -22,6 +22,12 @@ module.exports = async function (request) {
 
       rel.arch = 'aarch64';
     }
+
+    let isROCm = rel.name.includes('-rocm');
+    if (isROCm) {
+      Object.assign(rel, { arch: 'x86_64_rocm' });
+    }
+
     releases.push(rel);
   }
   all.releases = releases;
+3 −1		README.md
+23 −16		host-targets.js
+2 −2		package-lock.json
+1 −1		package.json
+5 −0		triplet.js
+1 −1		types.js