Use forked DeepFaceLab with minor fixes

This commit is contained in:
Oleksandr Chychkan
2022-02-04 10:29:06 -08:00
parent 3f0958a141
commit 07869626bf
+11 -7
View File
@@ -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