@@ -136,18 +136,18 @@ def _generate_ts_config_properties(handler_service):
136
136
ts_env = ts_environment .TorchServeEnvironment ()
137
137
138
138
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
151
151
}}'''
152
152
user_defined_configuration ["models" ] = models_string
153
153
logger .warn ("Sagemaker TS environment variables have been set and will be used "
0 commit comments