Update models

1 files changed, 22 insertions, 12 deletions

diff --git a/PKGBUILD b/PKGBUILD
index 3ad884e1d373..f56ba1e174b1 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -5,7 +5,7 @@
 pkgname=katago-cuda
 _pkgname=katago
 pkgver=1.14.1
-pkgrel=1
+pkgrel=2
 pkgdesc='Baduk engine with no human-provided knowledge'
 arch=(x86_64)
 provides=($_pkgname)
@@ -13,26 +13,36 @@ conflicts=($_pkgname)
 url='https://github.com/lightvector/KataGo'
 license=(MIT)
 depends=(libzip)
-makedepends=(cmake cuda cudnn git)
-source=("$pkgname"::"git+$url#tag=v${pkgver}"
-        "https://media.katagotraining.org/uploaded/networks/models/kata1/kata1-b18c384nbt-s9131461376-d4087399203.bin.gz")
-sha256sums=('SKIP'
-            'd266f19b4497ed19a72285a422813d8ee97e91d8a0b56f27585d0c02e61c9d38')
-noextract=("${source[1]##*/}")
+makedepends=(cmake cuda cudnn ninja)
+source=("${pkgname}-${pkgver}.tar.gz"::"${url}/archive/refs/tags/v${pkgver}.tar.gz"
+        # Strongest confidently-rated network
+        "https://media.katagotraining.org/uploaded/networks/models/kata1/kata1-b18c384nbt-s9131461376-d4087399203.bin.gz"
+        # Latest network
+        "https://media.katagotraining.org/uploaded/networks/models/kata1/kata1-b18c384nbt-s9555257600-d4198296303.bin.gz")
+sha256sums=('1a80d7fbd2b3a2684049afe61407d2276f6faf1dd1ca3f886cdb07c170c08e65'
+            'd266f19b4497ed19a72285a422813d8ee97e91d8a0b56f27585d0c02e61c9d38'
+            '30ff9a4f20f3f3cc40ba174187d87541e8becdd0dff54b12db7dfa24a3837d1e')
+_model_source=("${source[@]:1}")
+_model=("${_model_source[@]##*/}")
+noextract=("${_model[@]}")
 
 build() {
   cmake \
+    -S "KataGo-${pkgver}/cpp" \
     -B build \
     -D CMAKE_BUILD_TYPE=Release \
     -D CMAKE_INSTALL_PREFIX=/usr \
     -D USE_BACKEND=CUDA \
-    -S "${srcdir}/${pkgname}/cpp"
-  cmake --build build -j
+    -D NO_GIT_REVISION=1 \
+    -G Ninja
+  cmake --build build
 }
 
 package() {
-  cd "${srcdir}"
   install -Dm755 build/katago "${pkgdir}/usr/bin/katago"
-  install -Dm644 "${pkgname}/LICENSE" "${pkgdir}/usr/share/licenses/${_pkgname}/LICENSE"
-  install -Dm644 kata1-b18c384nbt-s9131461376-d4087399203.bin.gz "${pkgdir}/usr/share/${_pkgname}/kata1-b18c384nbt-s9131461376-d4087399203.bin.gz"
+  install -Dm644 "KataGo-${pkgver}/LICENSE" "${pkgdir}/usr/share/licenses/${_pkgname}/LICENSE"
+  # install models
+  for mi in "${_model[@]}"; do
+      install -Dm644 "${mi}" "${pkgdir}/usr/share/${_pkgname}/${mi}"
+  done
 }