Skip to content

Commit 1689ab6

Browse files
authored
Don't attempt to download locals from S3 bucket (#1669)
1 parent 6fa460f commit 1689ab6

File tree

1 file changed

+14
-13
lines changed

1 file changed

+14
-13
lines changed

Diff for: pkg/workloads/cortex/lib/client/tensorflow.py

+14-13
Original file line numberDiff line numberDiff line change
@@ -277,19 +277,20 @@ def _run_inference(self, model_input: Any, model_name: str, model_version: str)
277277
self._models.remove_model(model_name, model_version)
278278

279279
# download model
280-
logger().info(
281-
f"downloading model {model_name} of version {model_version} from the {upstream_model['provider']} upstream"
282-
)
283-
date = self._models.download_model(
284-
upstream_model["provider"],
285-
upstream_model["bucket"],
286-
model_name,
287-
model_version,
288-
upstream_model["path"],
289-
)
290-
if not date:
291-
raise WithBreak
292-
current_upstream_ts = int(date.timestamp())
280+
if model_name not in self._spec_models.get_local_model_names():
281+
logger().info(
282+
f"downloading model {model_name} of version {model_version} from the {upstream_model['provider']} upstream"
283+
)
284+
date = self._models.download_model(
285+
upstream_model["provider"],
286+
upstream_model["bucket"],
287+
model_name,
288+
model_version,
289+
upstream_model["path"],
290+
)
291+
if not date:
292+
raise WithBreak
293+
current_upstream_ts = int(date.timestamp())
293294

294295
# load model
295296
try:

0 commit comments

Comments
 (0)