Skip to content

Commit 1476743

Browse files
authored
Update torchserve.py
1 parent 2ea3361 commit 1476743

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

src/sagemaker_pytorch_serving_container/torchserve.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -136,18 +136,18 @@ def _generate_ts_config_properties(handler_service):
136136
ts_env = ts_environment.TorchServeEnvironment()
137137

138138
if ts_env.is_env_set() and not ENABLE_MULTI_MODEL:
139-
models_string = f'''{{\\
140-
"{DEFAULT_TS_MODEL_NAME}": {{\\
141-
"1.0": {{\\
142-
"defaultVersion": true,\\
143-
"marName": "{DEFAULT_TS_MODEL_NAME}.mar",\\
144-
"minWorkers": {ts_env._min_workers},\\
145-
"maxWorkers": {ts_env._max_workers},\\
146-
"batchSize": {ts_env._batch_size},\\
147-
"maxBatchDelay": {ts_env._max_batch_delay},\\
148-
"responseTimeout": {ts_env._response_timeout}\\
149-
}}\\
150-
}}\\
139+
models_string = f'''{{\\\n
140+
"{DEFAULT_TS_MODEL_NAME}": {{\\\n
141+
"1.0": {{\\\n
142+
"defaultVersion": true,\\\n
143+
"marName": "{DEFAULT_TS_MODEL_NAME}.mar",\\\n
144+
"minWorkers": {ts_env._min_workers},\\\n
145+
"maxWorkers": {ts_env._max_workers},\\\n
146+
"batchSize": {ts_env._batch_size},\\\n
147+
"maxBatchDelay": {ts_env._max_batch_delay},\\\n
148+
"responseTimeout": {ts_env._response_timeout}\\\n
149+
}}\\\n
150+
}}\\\n
151151
}}'''
152152
user_defined_configuration["models"] = models_string
153153
logger.warn("Sagemaker TS environment variables have been set and will be used "

0 commit comments

Comments
 (0)