From f427e4289d9c085979adb149f5305f4f67a29a7e Mon Sep 17 00:00:00 2001
From: Lev Kurilenko <lekurile@microsoft.com>
Date: Thu, 2 Nov 2023 20:27:27 +0000
Subject: [PATCH] Temporarily change branch to fork

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

diff --git a/.github/workflows/nv-ds-chat.yml b/.github/workflows/nv-ds-chat.yml
index b53fac36315b..cdefc0eeaf57 100644
--- a/.github/workflows/nv-ds-chat.yml
+++ b/.github/workflows/nv-ds-chat.yml
@@ -43,7 +43,9 @@ jobs:
               BRANCH="${{ github.event.inputs.dse_branch }}"
           fi
           echo "DeepSpeedExamples Branch: $BRANCH"
-          git clone -b $BRANCH https://github.com/microsoft/DeepSpeedExamples.git
+          #git clone -b $BRANCH https://github.com/microsoft/DeepSpeedExamples.git
+          # TODO: Revert back to main branch once fork is merged into DSE
+          git clone -b $BRANCH https://github.com/santacml/DeepSpeedExamples.git
           cd DeepSpeedExamples/applications/DeepSpeed-Chat
           pip install -r requirements.txt
           pip install -e .