From 3eba3210359ec4e36bebf4696623659f12253827 Mon Sep 17 00:00:00 2001
From: Alexander Kryuchkov <17554646+KernelA@users.noreply.github.com>
Date: Tue, 8 Aug 2023 22:45:47 +0300
Subject: [PATCH] Update README.md

---
 README.md | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index c4c19e9..328c57d 100644
--- a/README.md
+++ b/README.md
@@ -33,14 +33,14 @@ You can download a trained model from the [model registry](https://docs.wandb.ai
 
 ### CPU (training will be slow)
 ```
-pip install -r ./requirements.torch.cpu.txt
+pip install -r ./requirements.torch.gpu.txt
 pip install -r ./requirements.cpu.txt -r ./requirements.base.txt
 ```
 
 ### GPU:
 ```
-pip install -r ./requirements.torch.cpu.txt
-pip install -r ./requirements.cpu.txt -r ./requirements.base.txt
+pip install -r ./requirements.torch.gpu.txt
+pip install -r ./requirements.gpu.txt -r ./requirements.base.txt
 ```
 
 Only for `presentation.ipynb`