glm53 1.3 KB

12345678910111213141516171819202122232425262728293031323334
  1. #!/bin/bash
  2. # docker compose with the model and every overlay in experimental/compose, in
  3. # the order the README's numbers were measured with (later files win).
  4. #
  5. # ./glm53 up -d start
  6. # ./glm53 down --timeout 60 stop
  7. # ./glm53 logs -f | ps | config anything else docker compose takes
  8. # ./glm53 -f compose/ext.yaml up -d your own files, after the overlays
  9. # ./glm53 --stock up -d compose/glm53.yaml alone (stock kernels;
  10. # the entrypoint also wants ALLOW_STOCK=1)
  11. #
  12. # On a host running dispramd (kindling-spark-os) it also adds
  13. # experimental/compose/dispram.yaml, which puts the tail of the KV pool in the
  14. # GB10's display carveout. DISPRAM=0 leaves it out.
  15. set -e
  16. cd "$(dirname "$0")"
  17. overlays=(arx snapshot adaptive-k fp8 megamoe fixes sp recoverssm)
  18. files=(-f compose/glm53.yaml)
  19. extra=()
  20. stock=0
  21. while [ $# -gt 0 ]; do
  22. case $1 in
  23. --stock) stock=1; shift ;;
  24. -f) extra+=(-f "$2"); shift 2 ;;
  25. *) break ;;
  26. esac
  27. done
  28. if [ $stock = 0 ]; then
  29. for o in "${overlays[@]}"; do files+=(-f "experimental/compose/$o.yaml"); done
  30. if [ "${DISPRAM:-1}" != 0 ] && [ -S /run/dispram/dispram.sock ] && [ -d /opt/kindling/dispram/python ]; then
  31. files+=(-f experimental/compose/dispram.yaml)
  32. fi
  33. fi
  34. exec docker compose "${files[@]}" "${extra[@]}" "$@"