From 9e2081483157587e243eda24938b5e523a0b06e3 Mon Sep 17 00:00:00 2001 From: songkey-win Date: Sat, 9 Nov 2024 11:57:17 +0800 Subject: [PATCH] lc --- meme.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/meme.py b/meme.py index cb0de19..99eb6d2 100644 --- a/meme.py +++ b/meme.py @@ -343,9 +343,9 @@ class GetVideoDriveParams: save_size=512, trans_ratio=trans_ratio) drive_params = dict( - face_parts=face_parts_embedding.unsqueeze(0).to(dtype=dtype), - drive_coeff=drive_coeff.unsqueeze(0).to(dtype=dtype), - condition=control_heatmaps.unsqueeze(0).to(dtype=dtype), + face_parts=face_parts_embedding.unsqueeze(0).to(dtype=dtype, device='cpu'), + drive_coeff=drive_coeff.unsqueeze(0).to(dtype=dtype, device='cpu'), + condition=control_heatmaps.unsqueeze(0).to(dtype=dtype, device='cpu'), ) return (drive_params, )