We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ca26cd2 commit 4800eb1Copy full SHA for 4800eb1
src/sagemaker/serve/builder/transformers_builder.py
@@ -353,7 +353,6 @@ def _build_for_transformers(self):
353
if not os.path.exists(self.model_path):
354
os.makedirs(self.model_path)
355
356
-
357
code_path = Path(self.model_path).joinpath("code")
358
359
save_pkl(code_path, (self.inference_spec, self.schema_builder))
@@ -363,13 +362,13 @@ def _build_for_transformers(self):
363
362
364
self.secret_key = prepare_for_mms(
365
model_path=self.model_path,
366
- shared_libs=self.shared_libs,
367
- dependencies=self.dependencies,
368
- session=self.sagemaker_session,
369
- image_uri=self.image_uri,
370
- inference_spec=self.inference_spec,
+ shared_libs=self.shared_libs,
+ dependencies=self.dependencies,
+ session=self.sagemaker_session,
+ image_uri=self.image_uri,
+ inference_spec=self.inference_spec,
371
)
372
373
- self._build_transformers_env()
+ self._build_transformers_env()
374
375
return self.pysdk_model
0 commit comments