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

webinstall · Sep 15, 2024 · 0aa2b04 · 0aa2b04
1 parent c4a6d74
commit 0aa2b04
Showing 1 changed file with 6 additions and 0 deletions.
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;