Updated setup scripts to use cuda 11.8 and torch 2.0.0 to fix RTX 4090 compatibility

Added API to the generate function, so it can be called from other scripts
This commit is contained in:
Bikkies 2023-02-11 19:46:26 +11:00
parent 9bf1ea5b0a
commit 64ae4bb563
3 changed files with 7 additions and 2 deletions

View File

@ -1,7 +1,9 @@
python -m venv tortoise-venv python -m venv tortoise-venv
call .\tortoise-venv\Scripts\activate.bat call .\tortoise-venv\Scripts\activate.bat
python -m pip install --upgrade pip python -m pip install --upgrade pip
python -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu118
python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cu118
python -m pip install --pre torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118
python -m pip install -r ./requirements.txt python -m pip install -r ./requirements.txt
python setup.py install python setup.py install
deactivate deactivate

View File

@ -2,7 +2,9 @@ python -m venv tortoise-venv
source ./tortoise-venv/bin/activate source ./tortoise-venv/bin/activate
python -m pip install --upgrade pip python -m pip install --upgrade pip
# CUDA # CUDA
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu118
pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cu118
pip install --pre torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118
python -m pip install -r ./requirements.txt python -m pip install -r ./requirements.txt
python setup.py install python setup.py install
deactivate deactivate

View File

@ -779,6 +779,7 @@ def setup_gradio():
submit_event = submit.click(generate, submit_event = submit.click(generate,
inputs=input_settings, inputs=input_settings,
outputs=[selected_voice, output_audio, generation_results], outputs=[selected_voice, output_audio, generation_results],
api_name='generate'
) )
copy_button.click(import_generate_settings, copy_button.click(import_generate_settings,