mirror of
https://github.com/vegu-ai/talemate.git
synced 2025-09-01 09:59:08 +00:00
pytorch update
This commit is contained in:
parent
a94b49bc6a
commit
dbdbd64355
6 changed files with 422 additions and 429 deletions
|
@ -17,7 +17,6 @@ Roleplay with AI with a focus on strong narration and consistent world and game
|
|||
- Context management for character details, world information, past events, and pinned information
|
||||
- Customizable templates for all prompts using Jinja2
|
||||
- Modern, responsive UI
|
||||
- Node editor
|
||||
|
||||
## Documentation
|
||||
|
||||
|
|
|
@ -62,7 +62,7 @@ IF ERRORLEVEL 1 (
|
|||
echo CUDA found. Installing PyTorch with CUDA support...
|
||||
REM uninstalling existing torch, torchvision, torchaudio
|
||||
python -m pip uninstall torch torchaudio -y
|
||||
python -m pip install torch~=2.4.1 torchaudio~=2.4.1 --index-url https://download.pytorch.org/whl/cu121
|
||||
python -m pip install torch~=2.7.0 torchaudio~=2.7.0 --index-url https://download.pytorch.org/whl/cu128
|
||||
)
|
||||
|
||||
REM copy config.example.yaml to config.yaml only if config.yaml doesn't exist
|
||||
|
|
|
@ -26,7 +26,7 @@ if [ $cuda == "y" ]; then
|
|||
echo "Installing PyTorch with CUDA support..."
|
||||
# uninstall torch and torchaudio
|
||||
pip uninstall torch torchaudio -y
|
||||
pip install torch~=2.4.1 torchaudio~=2.4.1 --index-url https://download.pytorch.org/whl/cu121
|
||||
pip install torch~=2.7.0 torchaudio~=2.7.0 --index-url https://download.pytorch.org/whl/cu128
|
||||
fi
|
||||
|
||||
# copy config.example.yaml to config.yaml only if config.yaml doesn't exist
|
||||
|
|
842
poetry.lock
generated
842
poetry.lock
generated
File diff suppressed because it is too large
Load diff
|
@ -27,7 +27,7 @@ IF ERRORLEVEL 1 (
|
|||
echo CUDA found. Installing PyTorch with CUDA support...
|
||||
REM uninstalling existing torch, torchvision, torchaudio
|
||||
python -m pip uninstall torch torchaudio -y
|
||||
python -m pip install torch~=2.4.1 torchaudio~=2.4.1 --index-url https://download.pytorch.org/whl/cu121
|
||||
python -m pip install torch~=2.7.0 torchaudio~=2.7.0 --index-url https://download.pytorch.org/whl/cu128
|
||||
)
|
||||
|
||||
echo Virtual environment updated!
|
||||
|
|
|
@ -21,7 +21,7 @@ python3 -m poetry install
|
|||
if command -v nvcc >/dev/null 2>&1; then
|
||||
echo "CUDA found. Installing PyTorch with CUDA support..."
|
||||
python3 -m pip uninstall torch torchaudio -y
|
||||
python3 -m pip install torch~=2.4.1 torchaudio~=2.4.1 --index-url https://download.pytorch.org/whl/cu121
|
||||
python3 -m pip install torch~=2.7.0 torchaudio~=2.7.0 --index-url https://download.pytorch.org/whl/cu128
|
||||
else
|
||||
echo "CUDA not found. Keeping PyTorch installation without CUDA support..."
|
||||
fi
|
||||
|
|
Loading…
Add table
Reference in a new issue