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_swaper_ffmpeg_stream.py |    2 --
 1 files changed, 0 insertions(+), 2 deletions(-)

diff --git a/auto_media_publisher/utils/face_swap/face_swaper_ffmpeg_stream.py b/auto_media_publisher/modifiers/face_swap/face_swaper_ffmpeg_stream.py
similarity index 99%
rename from auto_media_publisher/utils/face_swap/face_swaper_ffmpeg_stream.py
rename to auto_media_publisher/modifiers/face_swap/face_swaper_ffmpeg_stream.py
index 5966fe3..0f2452d 100644
--- a/auto_media_publisher/utils/face_swap/face_swaper_ffmpeg_stream.py
+++ b/auto_media_publisher/modifiers/face_swap/face_swaper_ffmpeg_stream.py
@@ -8,7 +8,6 @@
 import torch
 import json
 import insightface
-from insightface.app.common import Face
 from .face_nsfw_predictor import NSFWPredictor
 from .face_enhancer import FaceEnhancer
 from .face_analyser import FaceAnalyser
@@ -41,7 +40,6 @@
         self.stop_event = threading.Event()
 
     def _select_device(self):
-        return 'cpu'
         if torch.cuda.is_available():
             return 'cuda'
         elif torch.backends.mps.is_available():

--
Gitblit v1.9.1