Merge pull request #63 from alishdipani/patch-1

Removed warning about torch version
Mayukhdeb · May 6, 2024 · 91904b2 · 91904b2
2 parents bfb1696 + 641ac94
commit 91904b2
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/torch_dreams/utils.py b/torch_dreams/utils.py
@@ -64,9 +64,6 @@ def get_fft_scale(h, w, decay_power=0.75, device="cuda"):
 def fft_to_rgb(height, width, image_parameter, device="cuda"):
     """convert image param to NCHW
 
-    WARNING: torch v1.7.0 works differently from torch v1.8.0 on fft.
-    torch-dreams supports ONLY 1.8.x
-
     Latest docs: https://pytorch.org/docs/stable/fft.html
 
     Also refer: