Compare commits

...

1 Commits

Author SHA1 Message Date
64ae4bb563 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
2023-02-11 19:46:26 +11:00
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,