Update handler.py

handler.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 class EndpointHandler:
     def __init__(self, path=""):
         # 初始化对话生成管道
-        self.pipeline = pipeline("text-generation", model=path)
+        self.pipeline = pipeline("text-generation", model=path, device=0)
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         """