diff --git a/scripts/0_setup.sh b/scripts/0_setup.sh
index aa02e77..ee766fa 100755
--- a/scripts/0_setup.sh
+++ b/scripts/0_setup.sh
@@ -6,9 +6,18 @@ set -e
 mkdir -p .dfl
 mkdir -p workspace
 
+architecture=''
+if [[ ! -z "$(uname -a | grep -oE ' arm64$')" ]]; then
+  architecture='arm64'
+fi
+
 if [ ! -d .dfl/DeepFaceLab ]; then
   echo "Cloning DeepFaceLab"
-  git clone --depth 1 "https://github.com/iperov/DeepFaceLab.git" .dfl/DeepFaceLab
+  git clone --depth 1 "https://github.com/chychkan/DeepFaceLab.git" .dfl/DeepFaceLab
+
+  if [ "$architecture" == "arm64" ]; then
+    git checkout support-arm64
+  fi
 fi
 
 if [ ! -d .dfl/env ]; then
@@ -25,13 +34,8 @@ if [[ ! -z "$version" && -f "requirements_$version.txt" ]]; then
   version_suffix="_$version"
 fi
 
-architecture=''
-if [[ ! -z "$(uname -a | grep -oE ' arm64$')" && -f "requirements${version_suffix}_arm64.txt" ]]; then
-  architecture='arm64'
-fi
-
 architecture_suffix=''
-if [[ ! -z "$architecture" ]]; then
+if [ "$architecture" == "arm64" && -f "requirements${version_suffix}_arm64.txt" ]; then
   architecture_suffix="_arm64"
 fi