From 5161fefe5f578ed04bd5ea2b03c9492989ac6c35 Mon Sep 17 00:00:00 2001 From: Gloria Park <138884103+gloriacohere@users.noreply.github.com> Date: Mon, 18 Nov 2024 10:14:54 -0500 Subject: [PATCH] Update Deploy command-r.ipynb bug fixes --- notebooks/sagemaker/Deploy command-r.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/notebooks/sagemaker/Deploy command-r.ipynb b/notebooks/sagemaker/Deploy command-r.ipynb index d1dcea9..ec4ec01 100644 --- a/notebooks/sagemaker/Deploy command-r.ipynb +++ b/notebooks/sagemaker/Deploy command-r.ipynb @@ -100,8 +100,8 @@ "outputs": [], "source": [ "cohere_packages = {\n", - " 'ml.p4de.24xlarge': \"cohere-command-r-a100-v1-6-d9a1f9283fe433edabbc8b9bf98f3519\",\n", - " 'ml.p5.48xlarge': \"cohere-command-r-h100-v1-6-d90e64821f283e0abba2ee874d6c8a31\",\n", + " 'ml.p4de.24xlarge': \"cohere-command-r-a100-v1-7-110-509d71c226593e7db36991f0c06fd80e\",\n", + " 'ml.p5.48xlarge': \"cohere-command-r-h100-v1-7-110-9a8425efcaf1353aa34041d49e8c51ee\",\n", "}\n", "\n", "cohere_package = cohere_packages[instance_type]\n",