update

.github/workflows/train_workflow.yml

@@ -9,7 +9,7 @@ on:
 
 jobs:
   train:
-    runs-on: [gpumode-nvidia-arc, amd-mi250] # Add more GPUs as we get them here
+    runs-on: [gpumode-nvidia-arc, amdgpu-mi250-x86-64] # Add more GPUs as we get them here
     steps:
       - name: Install dependencies
         run: |