Spaces:

mlabonne
/

llm-auto-eval

Running

mlabonne commited on Jan 18, 2024

Commit

79284c5

verified ·

1 Parent(s): 7580a4a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ For further details, see the project on 💻 [GitHub](https://github.com/mlabonn
 """
 def autoeval(BENCHMARK, MODEL, GPU, NUMBER_OF_GPUS, CONTAINER_DISK, CLOUD_TYPE, REPO, TRUST_REMOTE_CODE, DEBUG, RUNPOD_TOKEN, GITHUB_API_TOKEN):
-    # Create a pod
     pod = runpod.create_pod(
         name=f"Eval {MODEL.split('/')[-1]} on {BENCHMARK.capitalize()}",
         image_name="runpod/pytorch:2.0.1-py3.10-cuda11.8.0-devel-ubuntu22.04",

 """
 def autoeval(BENCHMARK, MODEL, GPU, NUMBER_OF_GPUS, CONTAINER_DISK, CLOUD_TYPE, REPO, TRUST_REMOTE_CODE, DEBUG, RUNPOD_TOKEN, GITHUB_API_TOKEN):
+    runpod.api_key = RUNPOD_TOKEN
     pod = runpod.create_pod(
         name=f"Eval {MODEL.split('/')[-1]} on {BENCHMARK.capitalize()}",
         image_name="runpod/pytorch:2.0.1-py3.10-cuda11.8.0-devel-ubuntu22.04",