Skip to content

Commit 1920ef7

Browse files
committed
Changes the location for the gpu_shapes_index file.
1 parent 68a529f commit 1920ef7

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

ads/aqua/common/utils.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -1288,7 +1288,9 @@ def load_gpu_shapes_index(
12881288
try:
12891289
auth = auth or authutil.default_signer()
12901290
# Construct the object storage path. Adjust bucket name and path as needed.
1291-
storage_path = f"oci://{CONDA_BUCKET_NAME}@{CONDA_BUCKET_NS}/{file_name}/1"
1291+
storage_path = (
1292+
f"oci://{CONDA_BUCKET_NAME}@{CONDA_BUCKET_NS}/service_pack/{file_name}"
1293+
)
12921294
logger.debug("Loading GPU shapes index from Object Storage")
12931295
with fsspec.open(storage_path, mode="r", **auth) as file_obj:
12941296
data = json.load(file_obj)

ads/aqua/modeldeployment/deployment.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1310,7 +1310,8 @@ def list_shapes(self, **kwargs) -> List[ComputeShapeSummary]:
13101310
memory_in_gbs=oci_shape.memory_in_gbs,
13111311
shape_series=oci_shape.shape_series,
13121312
name=oci_shape.name,
1313-
gpu_specs=gpu_specs.shapes.get(oci_shape.name),
1313+
gpu_specs=gpu_specs.shapes.get(oci_shape.name.upper())
1314+
or gpu_specs.shapes.get(oci_shape.name.upper()),
13141315
)
13151316
for oci_shape in oci_shapes
13161317
]

0 commit comments

Comments
 (0)