in ezsmdeploy/__init__.py [0:0]
def create_model(self):
if not self.multimodel:
self.sagemakermodel = Model(
name="model-" + self.name,
model_data=self.modelpath[0],
image_uri=self.image,
role=sagemaker.get_execution_role(),
# sagemaker_session=self.session,
predictor_cls=sagemaker.predictor.Predictor,
)
else:
self.sagemakermodel = MultiDataModel(
name="model-" + self.name,
model_data_prefix="/".join(self.modelpath[0].split("/")[:-1]) + "/",
image_uri=self.image,
role=sagemaker.get_execution_role(),
# sagemaker_session=self.session,
predictor_cls=sagemaker.predictor.Predictor,
)
for path in self.modelpath:
self.add_model(path, "serving/")
self.ei = False