Skip to content

Commit d4aafa3

Browse files
committed
support server test(camb)
1 parent 8395c91 commit d4aafa3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lmdeploy/cli/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,7 @@ def calib_search_scale(parser):
367367
@staticmethod
368368
def device(parser,
369369
default: str = 'cuda',
370-
choices: List[str] = ['cuda', 'ascend']):
370+
choices: List[str] = ['cuda', 'ascend','camb']):
371371
"""Add argument device to parser."""
372372

373373
return parser.add_argument('--device',

0 commit comments

Comments
 (0)