From fd10cef3282da8521a174923a168ab378cd2c517 Mon Sep 17 00:00:00 2001
From: Mark Saroufim <marksaroufim@meta.com>
Date: Mon, 11 Nov 2024 18:16:53 -0800
Subject: [PATCH] Update train_workflow.yml

---
 .github/workflows/train_workflow.yml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/train_workflow.yml b/.github/workflows/train_workflow.yml
index 8e79ccc..be54465 100644
--- a/.github/workflows/train_workflow.yml
+++ b/.github/workflows/train_workflow.yml
@@ -39,4 +39,6 @@ jobs:
       if: always()
       with:
         name: training-logs
-        path: training.log
\ No newline at end of file
+        path: training.log
+      env:
+        CUDA_VISIBLE_DEVICES: 0 # Make sure only one GPU is used for testing