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
call .\tortoise-venv\Scripts\activate.bat
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 setup.py install
deactivate

View File

@ -2,7 +2,9 @@ python -m venv tortoise-venv
source ./tortoise-venv/bin/activate
python -m pip install --upgrade pip
# 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 setup.py install
deactivate

View File

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