Update src/litbench_pipeline.py

src/litbench_pipeline.py

@@ -35,6 +35,8 @@ from datasets import load_dataset
 from huggingface_hub import login
 
 
+os.environ["CUDA_VISIBLE_DEVICES"]="0"
+
 # Load token from Hugging Face Secrets
 HF_TOKEN = os.getenv("access_token")
 # Login using the token