pytorch update

This commit is contained in:
vegu-ai-tools 2025-05-04 12:37:04 +03:00
parent a94b49bc6a
commit dbdbd64355
6 changed files with 422 additions and 429 deletions

View file

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

View file

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

View file

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

File diff suppressed because it is too large Load diff

View file

@ -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!

View file

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