From 469d7f6a90b4020aede3d2d453157f640f9b46c4 Mon Sep 17 00:00:00 2001
From: Michael Gold <goldmichael@gmail.com>
Date: Mon, 18 Sep 2023 08:00:57 -0400
Subject: [PATCH] fix: CUDA out of memory error on Windows

---
 README.md | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 09caf8e..b259f0b 100644
--- a/README.md
+++ b/README.md
@@ -72,10 +72,18 @@ python rerender.py --cfg config/real2sculpture.json
 <details>
 <summary>Installation on Windows</summary>
 
-  Before running the above 1-4 steps, you need prepare:
+  Before running the above 1-4 steps, you need to prepare:
 1. Install [CUDA](https://developer.nvidia.com/cuda-toolkit-archive)
 2. Install [git](https://git-scm.com/download/win)
 3. Install [VS](https://visualstudio.microsoft.com/) with Windows 10/11 SDK (for building deps/ebsynth/bin/ebsynth.exe)
+
+Additionally, to avoid CUDA out of memory errors, install xformers:
+
+```shell
+pip install xformers==0.0.16
+# install compatible version of pytorch
+pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
+```  
 </details>
 
 ## (1) Inference