ArcFace uses GPU by default, no need for 'CPU' value.

This commit is contained in:
baudneo 2025-08-01 17:04:29 -06:00
parent 6ac2f7f91e
commit 81525e984e
No known key found for this signature in database
GPG Key ID: 51445F2ED08EBC7F

View File

@ -149,11 +149,7 @@ class ArcfaceEmbedding(BaseEmbedding):
self.runner = ONNXModelRunner(
os.path.join(self.download_path, self.model_file),
device=self.config.face_recognition.device
or (
"GPU"
if self.config.face_recognition.model_size == "large"
else "CPU"
),
or "GPU",
)
def _preprocess_inputs(self, raw_inputs):