Skip to content

Commit

Permalink
Update Deploy command-r-plus.ipynb (#217)
Browse files Browse the repository at this point in the history
updated cohere_package
  • Loading branch information
jkedrzyckii authored Dec 16, 2024
1 parent b1a7cb9 commit 81cfc2f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions notebooks/sagemaker/Deploy command-r-plus.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit 81cfc2f

Please sign in to comment.