Skip to content

Commit 398ee81

Browse files
author
hhsecond
committed
bug fix on str.lower
1 parent 1e07e43 commit 398ee81

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlflow_redisai/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def target_help():
6060

6161

6262
def run_local(name, model_uri, flavor=None, config=None):
63-
device = config.get('device')
63+
device = config.get('device', 'cpu')
6464
if 'gpu' in device.lower():
6565
commands = ['docker', 'run', '-p', '6379:6379', '--gpus', 'all', '--rm', 'redisai/redisai:latest']
6666
else:

0 commit comments

Comments
 (0)