Add GPU availability check in worker process for @spaces.GPU decorator

app.py

@@ -577,6 +577,17 @@ def run_graio_demo(args):
     # 参考: https://huggingface.co/spaces/KlingTeam/LivePortrait/blob/main/app.py
     @spaces.GPU(duration=360)
     def gpu_wrapped_generate_video(*args, **kwargs):
+        # 在 worker 进程中检查 GPU 可用性
+        try:
+            if torch.cuda.is_available():
+                # 尝试访问 GPU 以确保它已准备好
+                _ = torch.cuda.current_device()
+                logging.info(f"GPU ready in worker process: {torch.cuda.get_device_name(0)}")
+            else:
+                logging.warning("GPU not available in worker process, but continuing...")
+        except RuntimeError as e:
+            logging.warning(f"GPU initialization error in worker process: {e}. Continuing anyway...")
+        
         return generate_video(*args, **kwargs)
 
     def toggle_audio_inputs(person_num):