From 286d3cd51cc327381737844d330348ec97577e60 Mon Sep 17 00:00:00 2001
From: Tamar Rott Shaham <54222333+tamarott@users.noreply.github.com>
Date: Mon, 16 Aug 2021 11:50:22 +0300
Subject: [PATCH] Update README.md

---
 README.md | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index f546f569..371d746f 100644
--- a/README.md
+++ b/README.md
@@ -37,7 +37,12 @@ If you use this code for your research, please cite our paper:
 python -m pip install -r requirements.txt
 ```
 
-This code was tested with python 3.6  
+This code was tested with python 3.6, torch 1.4
+
+Please note: the code currently only supports torch 1.4 or earlier because of the optimization scheme.
+
+For later torch versions, you may try this repository: https://github.com/kligvasser/SinGAN (results won't necessarily be identical to the official implementation).
+
 
 ###  Train
 To train SinGAN model on your own image, put the desired training image under Input/Images, and run