Temporarily change branch to fork

microsoft · Nov 2, 2023 · f427e42 · f427e42
1 parent 2e0fa09
commit f427e42
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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 .