Fix README

PeacefulHorizon · Jun 29, 2023 · 67d7a7c · 67d7a7c
1 parent 15e0565
commit 67d7a7c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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: