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