Fix README

pull/129/head
Xingang Pan 2 years ago
parent 15e0565347
commit 67d7a7ca81

@ -64,6 +64,9 @@ cat environment.yml | \
grep -v -E 'nvidia|cuda' > environment-no-nvidia.yml && \
conda env create -f environment-no-nvidia.yml
conda activate stylegan3
# On MacOS
export PYTORCH_ENABLE_MPS_FALLBACK=1
```
## Run Gradio visualizer in Docker
@ -78,10 +81,6 @@ cd src && python visualizer_drag_gradio.py
Now you can open a shared link from Gradio (printed in the terminal console).
Beware the Docker image takes about 25GB of disk space!
# On MacOS
export PYTORCH_ENABLE_MPS_FALLBACK=1
```
## Download pre-trained StyleGAN2 weights
To download pre-trained weights, simply run:

Loading…
Cancel
Save