Spaces:
Runtime error
Runtime error
Update demo/agent.py
Browse files- 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'
|
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(
|