dstars commited on
Commit
f136d79
·
verified ·
1 Parent(s): b586a9e

Update demo/agent.py

Browse files
Files changed (1) hide show
  1. demo/agent.py +1 -1
demo/agent.py CHANGED
@@ -9,7 +9,7 @@ from lmdeploy import pipeline, TurbomindEngineConfig, GenerationConfig, ChatTemp
9
  from lmdeploy.vl import load_image
10
 
11
  class ConversationalAgent:
12
- def __init__(self, model_path, outputs_dir, device='cpu', backend_config=None):
13
  # 传入 device 参数,并设置 pipeline 时指定设备
14
  self.device = device
15
  self.pipe = pipeline(
 
9
  from lmdeploy.vl import load_image
10
 
11
  class ConversationalAgent:
12
+ def __init__(self, model_path, outputs_dir, device='cpu'):
13
  # 传入 device 参数,并设置 pipeline 时指定设备
14
  self.device = device
15
  self.pipe = pipeline(