diff --git a/README.md b/README.md index 55b5a29..8fdd6dc 100644 --- a/README.md +++ b/README.md @@ -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: