From 203e571aba6b93b58801857df04e90a785ce365c Mon Sep 17 00:00:00 2001
From: wangzhibo <wangzhibo@shsening.com>
Date: 星期三, 30 七月 2025 22:11:26 +0800
Subject: [PATCH] add face swapper
---
auto_media_publisher/modifiers/face_swap/face_enhancer.py | 1 -
1 files changed, 0 insertions(+), 1 deletions(-)
diff --git a/auto_media_publisher/utils/face_swap/face_enhancer.py b/auto_media_publisher/modifiers/face_swap/face_enhancer.py
similarity index 98%
rename from auto_media_publisher/utils/face_swap/face_enhancer.py
rename to auto_media_publisher/modifiers/face_swap/face_enhancer.py
index d4b6beb..1b600fe 100644
--- a/auto_media_publisher/utils/face_swap/face_enhancer.py
+++ b/auto_media_publisher/modifiers/face_swap/face_enhancer.py
@@ -16,7 +16,6 @@
self._enhancer = GFPGANer(model_path=FACE_ENHANCER_MODEL_PATH, upscale=1, device=self.device)
def _select_device(self):
- return 'cpu'
if torch.cuda.is_available():
return 'cuda'
elif torch.backends.mps.is_available():
--
Gitblit v1.9.1