Update demo/agent.py

demo/agent.py

@@ -12,21 +12,9 @@ class ConversationalAgent:
     def __init__(self, 
                  model_path,
                  outputs_dir) -> None:
-
-        # 自动检测设备
-        # self.device = device or torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        # print(f"Using device: {self.device}")
-        
         self.pipe = pipeline(model_path,
                         chat_template_config=ChatTemplateConfig(model_name='internvl2-internlm2'),
                         backend_config=TurbomindEngineConfig(session_len=8192))
-        # 初始化 pipeline，并显式指定设备
-        # self.pipe = pipeline(
-        #     model_path,
-        #     chat_template_config=ChatTemplateConfig(model_name='internvl2-internlm2'),
-        #     backend_config=TurbomindEngineConfig(session_len=8192, device=self.device)  # 在 Turbomind 中添加设备
-        # )
-
         self.uploaded_images_storage = os.path.join(outputs_dir, "uploaded")
         self.uploaded_images_storage = os.path.abspath(self.uploaded_images_storage)
         os.makedirs(self.uploaded_images_storage, exist_ok=True)