Update README.md

This commit is contained in:
Qing 2023-04-10 09:16:35 +08:00 committed by GitHub
parent 2c25cff9f7
commit d283595681
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -61,7 +61,7 @@ Lama Cleaner make it easy to use SOTA AI model in just two commands:
```bash
# In order to use the GPU, install cuda version of pytorch first.
# pip install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
# pip install torch==1.13.1+cu117 torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cu117
pip install lama-cleaner
lama-cleaner --model=lama --device=cpu --port=8080
```