mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 223 Pull Requests 5 Packages Projects Releases Wiki Activity

explicitly use python3 because some OSs will not have python alias to python3, allow batch size 1

Browse Source
This commit is contained in:
mrq 2023-02-21 20:20:52 +00:00
parent 8a1a48f31e
commit f119993fb5
4 changed files with 26 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
setup-cuda.sh
View File

@ -2,12 +2,13 @@
git submodule init git submodule init
git submodule update git submodule update
python -m venv venv python3 -m venv venv
source ./venv/bin/activate source ./venv/bin/activate
python -m pip install --upgrade pip python3 -m pip install --upgrade pip
# CUDA
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
python -m pip install -r ./dlas/requirements.txt python3 -m pip install -r ./dlas/requirements.txt
python -m pip install -r ./tortoise-tts/requirements.txt python3 -m pip install -r ./tortoise-tts/requirements.txt
python -m pip install -r ./requirements.txt python3 -m pip install -r ./requirements.txt
python -m pip install -e ./tortoise-tts/ python3 -m pip install -e ./tortoise-tts/
deactivate deactivate

12
setup-rocm.sh
View File

@ -2,13 +2,13 @@
git submodule init git submodule init
git submodule update git submodule update
python -m venv venv python3 -m venv venv
source ./venv/bin/activate source ./venv/bin/activate
python -m pip install --upgrade pip python3 -m pip install --upgrade pip
# ROCM # ROCM
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 # 5.2 does not work for me desu pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 # 5.2 does not work for me desu
python -m pip install -r ./dlas/requirements.txt python3 -m pip install -r ./dlas/requirements.txt
python -m pip install -r ./tortoise-tts/requirements.txt python3 -m pip install -r ./tortoise-tts/requirements.txt
python -m pip install -r ./requirements.txt python3 -m pip install -r ./requirements.txt
python -m pip install -e ./tortoise-tts/ python3 -m pip install -e ./tortoise-tts/
deactivate deactivate

11
src/utils.py
View File

@ -575,15 +575,18 @@ def optimize_training_settings( epochs, batch_size, learning_rate, learning_rate
batch_size = lines batch_size = lines
messages.append(f"Batch size is larger than your dataset, clamping batch size to: {batch_size}") messages.append(f"Batch size is larger than your dataset, clamping batch size to: {batch_size}")
if batch_size / mega_batch_factor < 2:
mega_batch_factor = int(batch_size / 2)
messages.append(f"Mega batch factor is too large for the given batch size, clamping mega batch factor to: {mega_batch_factor}")
if batch_size % lines != 0: if batch_size % lines != 0:
nearest_slice = int(lines / batch_size) + 1 nearest_slice = int(lines / batch_size) + 1
batch_size = int(lines / nearest_slice) batch_size = int(lines / nearest_slice)
messages.append(f"Batch size not neatly divisible by dataset size, adjusting batch size to: {batch_size} ({nearest_slice} steps per epoch)") messages.append(f"Batch size not neatly divisible by dataset size, adjusting batch size to: {batch_size} ({nearest_slice} steps per epoch)")
if batch_size == 1 and mega_batch_factor != 1:
mega_batch_factor = 1
messages.append(f"Mega batch factor is too large for the given batch size, clamping mega batch factor to: {mega_batch_factor}")
elif batch_size / mega_batch_factor < 2:
mega_batch_factor = int(batch_size / 2)
messages.append(f"Mega batch factor is too large for the given batch size, clamping mega batch factor to: {mega_batch_factor}")
iterations = calc_iterations(epochs=epochs, lines=lines, batch_size=batch_size) iterations = calc_iterations(epochs=epochs, lines=lines, batch_size=batch_size)
if epochs < print_rate: if epochs < print_rate:

12
update.sh
View File

@ -2,14 +2,14 @@
git pull git pull
git submodule update git submodule update
python -m venv venv python3 -m venv venv
source ./venv/bin/activate source ./venv/bin/activate
python -m pip install --upgrade pip python3 -m pip install --upgrade pip
python -m pip install -r ./dlas/requirements.txt python3 -m pip install -r ./dlas/requirements.txt
python -m pip install -r ./tortoise-tts/requirements.txt python3 -m pip install -r ./tortoise-tts/requirements.txt
python -m pip install -e ./tortoise-tts python3 -m pip install -e ./tortoise-tts
python -m pip install -r ./requirements.txt python3 -m pip install -r ./requirements.txt
deactivate deactivate