We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1e07e43 commit 398ee81Copy full SHA for 398ee81
mlflow_redisai/__init__.py
@@ -60,7 +60,7 @@ def target_help():
60
61
62
def run_local(name, model_uri, flavor=None, config=None):
63
- device = config.get('device')
+ device = config.get('device', 'cpu')
64
if 'gpu' in device.lower():
65
commands = ['docker', 'run', '-p', '6379:6379', '--gpus', 'all', '--rm', 'redisai/redisai:latest']
66
else:
0 commit comments