From 4abe54f33ece6b90a33f5dd7e5e52f6066c97e08 Mon Sep 17 00:00:00 2001 From: lipku Date: Mon, 16 Dec 2024 12:58:58 +0800 Subject: [PATCH] compliance with old pytoch --- lipreal.py | 2 +- musereal.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lipreal.py b/lipreal.py index fea8700..b7a6d52 100644 --- a/lipreal.py +++ b/lipreal.py @@ -48,7 +48,7 @@ print('Using {} for inference.'.format(device)) def _load(checkpoint_path): if device == 'cuda': - checkpoint = torch.load(checkpoint_path,weights_only=True) + checkpoint = torch.load(checkpoint_path) #,weights_only=True else: checkpoint = torch.load(checkpoint_path, map_location=lambda storage, loc: storage) diff --git a/musereal.py b/musereal.py index a05e5fc..ebb25a7 100644 --- a/musereal.py +++ b/musereal.py @@ -76,7 +76,7 @@ def load_avatar(avatar_id): # "bbox_shift":self.bbox_shift # } - input_latent_list_cycle = torch.load(latents_out_path,weights_only=True) + input_latent_list_cycle = torch.load(latents_out_path) #,weights_only=True with open(coords_path, 'rb') as f: coord_list_cycle = pickle.load(f) input_img_list = glob.glob(os.path.join(full_imgs_path, '*.[jpJP][pnPN]*[gG]'))