aiden
/
DragGAN
mirror of https://github.com/XingangPan/DragGAN
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'XingangPan:main' into main

Browse Source
pull/142/head
Tianhao Xie 3 years ago committed by GitHub
parent 5dc6156229 e77f69d665
commit caeb80f324
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File

6
README.md
Unescape Escape View File

@ -74,8 +74,12 @@ export PYTORCH_ENABLE_MPS_FALLBACK=1
Provided docker image is based on NGC PyTorch repository. To quickly try out visualizer in Docker, run the following: Provided docker image is based on NGC PyTorch repository. To quickly try out visualizer in Docker, run the following:
```sh ```sh
# before you build the docker container, make sure you have cloned this repo, and downloaded the pretrained model by `python scripts/download_model.py`.
docker build . -t draggan:latest docker build . -t draggan:latest
docker run -p 7860: 7860 -v "$PWD":/workspace/src -it draggan:latest bash docker run -p 7860:7860 -v "$PWD":/workspace/src -it draggan:latest bash
# (Use GPU)if you want to utilize your Nvidia gpu to accelerate in docker, please add command tag `--gpus all`, like:
# docker run --gpus all -p 7860:7860 -v "$PWD":/workspace/src -it draggan:latest bash
cd src && python visualizer_drag_gradio.py --listen cd src && python visualizer_drag_gradio.py --listen
``` ```
Now you can open a shared link from Gradio (printed in the terminal console). Now you can open a shared link from Gradio (printed in the terminal console).

Write Preview
Loading…
Cancel
Save