mirror of
https://github.com/TabbyML/tabby
synced 2024-11-22 00:08:06 +00:00
chore: add experimental eval setup
This commit is contained in:
parent
06ac9f594e
commit
848aff7d1f
19
experimental/eval/docker-compose.yaml
Normal file
19
experimental/eval/docker-compose.yaml
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
version: '3.5'
|
||||||
|
|
||||||
|
services:
|
||||||
|
scheduler:
|
||||||
|
image: tabbyml/tabby
|
||||||
|
platform: linux/amd64
|
||||||
|
command: scheduler --now
|
||||||
|
volumes:
|
||||||
|
- "$HOME/.tabby:/data"
|
||||||
|
|
||||||
|
tabby:
|
||||||
|
depends_on:
|
||||||
|
- scheduler
|
||||||
|
restart: always
|
||||||
|
image: tabbyml/tabby
|
||||||
|
platform: linux/amd64
|
||||||
|
command: serve --model TabbyML/T5P-220M
|
||||||
|
volumes:
|
||||||
|
- "$HOME/.tabby:/data"
|
24
experimental/eval/skypilot.yaml
Normal file
24
experimental/eval/skypilot.yaml
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
resources:
|
||||||
|
accelerators: T4:1
|
||||||
|
|
||||||
|
setup: |
|
||||||
|
set -ex
|
||||||
|
|
||||||
|
# On some cloud providers, docker-compose is not installed by default.
|
||||||
|
sudo curl -L https://github.com/docker/compose/releases/download/v2.17.2/docker-compose-linux-x86_64 -o /usr/local/bin/docker-compose
|
||||||
|
sudo chmod a+x /usr/local/bin/docker-compose
|
||||||
|
|
||||||
|
# Pull tabby images.
|
||||||
|
git clone https://github.com/TabbyML/tabby.git || true
|
||||||
|
cd tabby/experimental
|
||||||
|
|
||||||
|
# On certain cloud providers (e.g lambda cloud), the default user is not added to docker group, so we need sudo here
|
||||||
|
sudo docker-compose pull
|
||||||
|
|
||||||
|
# Add current user to docker group, it won't take effect immediately as skypilot job is started by a long-running daemon.
|
||||||
|
sudo usermod -aG docker $USER
|
||||||
|
|
||||||
|
|
||||||
|
run: |
|
||||||
|
cd tabby/experimental
|
||||||
|
sudo docker-compose up
|
Loading…
Reference in New Issue
Block a user