From 67d7a7ca8187bce082f3b27f768fe49959debb1e Mon Sep 17 00:00:00 2001 From: Xingang Pan Date: Thu, 29 Jun 2023 09:37:13 +0800 Subject: [PATCH] Fix README --- README.md | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) 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: