From 0d5943af915c39b9d53c8ba26a9519f917284e45 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 28 Nov 2023 12:52:55 -0700 Subject: [PATCH] Update readme for torch requirements --- README.md | 4 ++-- requirements.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 84e7b180..8763d856 100644 --- a/README.md +++ b/README.md @@ -35,8 +35,8 @@ git submodule update --init --recursive python3 -m venv venv source venv/bin/activate # .\venv\Scripts\activate on windows -# windows install pytorch first with -# pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu117 +# install torch first +pip3 install torch pip3 install -r requirements.txt ``` diff --git a/requirements.txt b/requirements.txt index 47cdecdf..5d569109 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,3 +23,4 @@ prodigyopt controlnet_aux==0.0.7 python-dotenv bitsandbytes +xformers \ No newline at end of file