From f221292dfab8472fbe511f0ceed4e9fba42e6281 Mon Sep 17 00:00:00 2001 From: smthemex <138738845+smthemex@users.noreply.github.com> Date: Thu, 12 Sep 2024 22:48:52 +0800 Subject: [PATCH] Update DeepFakeDefender_node.py tensor not using copy() --- DeepFakeDefender_node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/DeepFakeDefender_node.py b/DeepFakeDefender_node.py index 09a3fc3..44ab245 100644 --- a/DeepFakeDefender_node.py +++ b/DeepFakeDefender_node.py @@ -140,7 +140,7 @@ class DeepFakeDefender_Sampler: empty_img = Image.new('RGB', (512, 512), (255, 255, 255)) B, _, _, _ = image.shape if B==1: - origin_img_list =[tensor2pil(image.copy())] + origin_img_list =[tensor2pil(image.clone())] img_list=[nomarl_upscale_topil(image, crop_width, crop_height)] else: img_list = list(torch.chunk(image, chunks=B))