New GPU docker image, update Cuda and pytorch version, add xformers. With --enable-xformers
, SD model get 30%~40% performance improvement
- cuda: 11.7
- pytorch: 1.13.1
- minimum nvidia driver: 515.43.04+
- xformers: 0.0.16rc425
docker run --gpus all -p 8080:8080 \
-v /path/to/cache_dir:/root/.cache/ \
--rm cwq1913/lama-cleaner:gpu-0.33.0 \
lama-cleaner --device=cuda --port=8080 --host=0.0.0.0 --enable-xformers