search

derrian-distro / LoRA_Easy_Training_Scripts

A UI made in Pyside6 to make training LoRA/LoCon and other LoRA type models in sd-scripts easy
GNU General Public License v3.0
1.06k stars 103 forks source link

what version of torch is being used for the flux training #233

Closed zethfoxster closed 1 month ago

zethfoxster commented 1 month ago

its been found that this version below increases training speed and reduces vram usage are we currently on this?

torch==2.4.0 torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu124

Jelosus2 commented 1 month ago

It uses the latest (aka 2.4.0 and 0.19.0). You can always check a package version by going to (Trainer)/backend/sd_scripts, opening a cmd in that directory and executing this commands in order:

# Windows
.\venv\Scripts\activate
# Linux
./venv/bin/activate

# Windows
pip list -v | findstr package-name
# Linux
pip list -v | grep package-name