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 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,