leaderboard / README.md
Jae-Won Chung
Add Dockerfile and fix requirements.txt typo
36fdd36
|
raw
history blame
1 kB
metadata
title: ML.ENERGY Leaderboard
python_version: 3.9
app_file: app.py
sdk: gradio
sdk_version: 3.23.0
pinned: false

ML.ENERGY Leaderboard

Devs

Currently setup in ampere02:

  1. Find model weights in /data/leaderboard/weights/, e.g. subdirectory llama and vicuna.

  2. Let's share the Huggingface Transformer cache:

export TRANSFORMERS_CACHE=/data/leaderboard/hfcache

Run benchmarks like this:

$ docker build -t leaderboard:latest .
$ docker run -it --name jw-leaderboard --gpus all --cap-add SYS_ADMIN -v /data/leaderboard:/data/leaderboard -v $HOME/workspace/leaderboard:/workspace/leaderboard leaderboard:latest bash

# cd leaderboard
# python benchmark.py --model-path /data/leaderboard/weights/lmsys/vicuna-7B --input-file /data/leaderboard/sharegpt/sg_90k_part1_html_cleaned_lang_first_sampled.json
# python benchmark.py --model-path databricks/dolly-v2-12b --input-file /data/leaderboard/sharegpt/sg_90k_part1_html_cleaned_lang_first_sampled.json