From 641ac94b4239c7ec38ed7469fb88d1949ec6f27c Mon Sep 17 00:00:00 2001 From: Alish Dipani Date: Mon, 6 May 2024 20:36:30 +0530 Subject: [PATCH] Removed warning about torch version --- torch_dreams/utils.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/torch_dreams/utils.py b/torch_dreams/utils.py index a03e0ec..d820c13 100644 --- 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: