From d5489d1f235b33c1f9e76767b136edbe73294087 Mon Sep 17 00:00:00 2001 From: AJ ONeal Date: Sat, 14 Sep 2024 19:18:27 +0000 Subject: [PATCH] fix(ollama): classify 'rocm' as its own cpu + update classifier --- _webi/build-classifier | 2 +- ollama/releases.js | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/_webi/build-classifier b/_webi/build-classifier index 28536b0aa..20e001829 160000 --- a/_webi/build-classifier +++ b/_webi/build-classifier @@ -1 +1 @@ -Subproject commit 28536b0aa2985ff81abc378807ce90863be882e8 +Subproject commit 20e001829a43d0c273517cd3f2a0abb9e127838b diff --git a/ollama/releases.js b/ollama/releases.js index 9ca5861ac..b7b24f471 100644 --- 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;