From 237975903832296fcf35fdcd431800328692bda8 Mon Sep 17 00:00:00 2001
From: taylorZhang <448205106@qq.com>
Date: Tue, 17 Jun 2025 17:38:25 +0800
Subject: [PATCH] fix: head_pose adapt device="gpu"

---
 personfromvid/models/head_pose_estimator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/personfromvid/models/head_pose_estimator.py b/personfromvid/models/head_pose_estimator.py
index 818cf86..435bc38 100644
--- a/personfromvid/models/head_pose_estimator.py
+++ b/personfromvid/models/head_pose_estimator.py
@@ -140,7 +140,7 @@ def __init__(
 
     def _resolve_device(self, device: str) -> str:
         """Resolve device string to actual device."""
-        if device == "auto":
+        if device in ["auto", "gpu"]:
             # Check for CUDA availability
             try:
                 import torch