def tf1_host(tf1_svc, enable_microbatch, clean_context): with export_service_bundle(tf1_svc) as saved_path: server_image = clean_context.enter_context( build_api_server_docker_image(saved_path)) with run_api_server_docker_container( server_image, enable_microbatch=enable_microbatch, timeout=500) as host: yield host
def transformers_host(transformers_image): with run_api_server_docker_container(transformers_image, timeout=500) as host: yield host
def tf2_host(tf2_image): with run_api_server_docker_container(tf2_image, timeout=500) as host: yield host
def h2o_docker_host(h2o_image): with run_api_server_docker_container(h2o_image, timeout=500) as host: yield host