Update demo/agent.py

demo/agent.py

@@ -7,7 +7,6 @@ from PIL import Image
 
 # from lmdeploy import pipeline, TurbomindEngineConfig, GenerationConfig, ChatTemplateConfig
 from lmdeploy import pipeline, GenerationConfig, ChatTemplateConfig
-from lmdeploy.pytorch import PytorchEngineConfig
 from lmdeploy.vl import load_image
 
 class ConversationalAgent:
@@ -19,7 +18,7 @@ class ConversationalAgent:
             chat_template_config=ChatTemplateConfig(model_name='internvl2-internlm2'),
             # backend_config=TurbomindEngineConfig(session_len=8192),
             
-            backend_config=PytorchEngineConfig(max_length=8192),
+            # backend_config=PytorchEngineConfig(max_length=8192),
             device=self.device
         )