From 56261bf550f09758ada8f515a1ff221724a84e05 Mon Sep 17 00:00:00 2001 From: Songkey Date: Wed, 13 Nov 2024 18:31:01 +0800 Subject: [PATCH] Update meme.py --- meme.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/meme.py b/meme.py index fd9593c..3d6d181 100644 --- a/meme.py +++ b/meme.py @@ -206,7 +206,7 @@ class HMFaceToolkitsLoader: dtype = torch.float16 device = torch.device(f'cuda:{gpu_id}') - face_aligner = HelloCameraDemo(face_alignment_module=HelloFaceAlignment(gpu_id=gpu_id), reset=True) + face_aligner = HelloCameraDemo(face_alignment_module=HelloFaceAlignment(gpu_id=gpu_id), reset=False) harkit_bs = HelloARKitBSPred(gpu_id=gpu_id) h3dmm = Hello3DMMPred(gpu_id=gpu_id) clip_image_encoder = CLIPVisionModelWithProjection.from_pretrained(