EveryDream2trainer/torch2_install.md

33 lines
1.2 KiB
Markdown

Torch 2 support is extremely pre-alpha. It is not recommended for use.
There is a bunch of triton error spam in logs, I've tried to suppress the errors but it's not working yet.
Clone a new copy, the VENV is not compatible and git will not manage the environment
`git clone https://github.com/victorchall/EveryDream2trainer ed2torch2`
`cd ed2torch2`
`git checkout torch2`
run normal install (`windows_setup.cmd` or build the docker container)
Try running training.
If your hyperparametr/grad scaler goes to a tiny or massive value its not working. I sugget using validation to make sure its actually working.
If you have problems, might try to install the latest xformers wheel from the github actions:
Download the Xformers 3.10 cu118 wheel for your system
https://github.com/facebookresearch/xformers/actions/runs/4501451442
linux: ubuntu-22.04-py3.10-torch2.0.0+cu118
https://github.com/facebookresearch/xformers/suites/11760994158/artifacts/613483176
win: windows-2019-py3.10-torch2.0.0+cu118
https://github.com/facebookresearch/xformers/suites/11760994158/artifacts/613483194
Save the .whl file to your everydream2trainer folder
activate your venv and pip install the wheel file
`pip install xformers-0.0.17.dev484-cp310-cp310-win_amd64.whl`