Benchmarking Models

With the Container Launch config, we can spin up a secure VM, and start a container on it automatically. Let’s start with a very simple example - running evals on a new model from Huggingface.

We have an image that does this with the LM Eval Harness library using HF Transformers. The code for it can be found and forked here. We also have a notebook that automatically launches this image on the most affordable A6000’s on the market to run the Mistral-7b-Instruct-v0.2 model though a benchmark. You can configure the model, gpu type, number of fewshot examples, and the benchmark eval.


The requirements are simple, so in your preferred python environment:

git clone
cd examples/

Then in deploy_container.ipynb you will need to input two things: 1. Your Shadeform API Key from here, and 2. A huggingface_token if the model you need is gated (like Llama2 or Gemma).

The notebook kicks off a VM request and runs the container specifically on a 1xA6000 Machine which are quite afforable on our platform. The container will download the model and run the benchmark for us.

The notebook leverages a lot of the principles found in this guide for using the platform to find affordable GPU’s.

Deploying a container

Once we have an available instance, we will deploy a container that runs lm-eval-harness on the model. Here is where we pass in environment variables, and select the docker image via our API.

payload = {
  "cloud": best_instance["cloud"],
  "region": region,
  "shade_instance_type": shade_instance_type,
  "shade_cloud": True,
  "name": "cool_gpu_server",
  "launch_configuration": {
    "type": "docker",
     #This selects the image to launch, and sets environment variables "tasks" and "num_fewshot"
    "docker_configuration": {
      "image": "shadeform/lm-eval-harness",
      "envs": [
	      	"name": "model",
	      	"value": "mistralai/Mistral-7B-Instruct-v0.2"
	      	"name": "tasks",
	      	"value": "hellaswag"
          "name": "num_fewshot",
          "value": "10"

Checking Results via the Shadeform Platform

The first way to check the results is by watching the logs under “Running Instances”, clicking our instance, and then the Logs(beta) tab. Here we see a stream of the logs from the docker container. When it is done, it will look something like this FinishedBenchmark

From the results of the evaluation, we can see that the Mistral model achieved a normalized accuracy score of 84.6% with 10 few-shot examples on the Hellaswag benchmark.

Checking Results inside your IDE

At the bottom of the notebook, there is a cell that prints out the IP address of your instance.

instance_response = requests.request("GET", base_url, headers=headers)

instance = json.loads(instance_response.text)["instances"][0]
instance_status = instance['status']
if instance_status == 'active':
    print(f"Instance is active with IP: {instance['ip']}")
    print(f"Instance isn't yet active: {instance}" )

We will ssh into the instance by leveraging this ip address and our shadeform .pem file.

ssh -i <location-of-your-pem-file> shadeform@<ip-address-copied-from-cell-output>

Once in, we can check the logs via the container id. The -a flag is if it has already finished.

docker ps -a
docker container logs <container-id>

And just like that, we can run a GPU-accelerated computing job in the cloud for affordable pricing without leaving our IDE!