From 81cfc2fc013338dd82e5b55441e2895ea4e82276 Mon Sep 17 00:00:00 2001 From: jkedrzyckii <140448212+jkedrzyckii@users.noreply.github.com> Date: Mon, 16 Dec 2024 14:56:19 -0500 Subject: [PATCH] Update Deploy command-r-plus.ipynb (#217) updated cohere_package --- notebooks/sagemaker/Deploy command-r-plus.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/notebooks/sagemaker/Deploy command-r-plus.ipynb b/notebooks/sagemaker/Deploy command-r-plus.ipynb index ca01b4b98..ad1fc8bfc 100644 --- a/notebooks/sagemaker/Deploy command-r-plus.ipynb +++ b/notebooks/sagemaker/Deploy command-r-plus.ipynb @@ -100,8 +100,8 @@ "outputs": [], "source": [ "cohere_packages = {\n", - " 'ml.p4de.24xlarge': \"cohere-command-r-plus-a100-v1--a2db668865cf31ff841f921d27d2257c\",\n", - " 'ml.p5.48xlarge': \"cohere-command-r-plus-h100-v1--312cd543f0e33e06b7cf47fe16bcd91c\",\n", + " 'ml.p4de.24xlarge': \"cohere-command-r-plus-a100-v1--5ec16e7130d63afba508433ea2899632\",\n", + " 'ml.p5.48xlarge': \"cohere-command-r-plus-h100-v1--6e599b46bf1131f5b367faf287f4bcbd\",\n", "}\n", "\n", "cohere_package = cohere_packages[instance_type]\n",