notebooks3

Author: singh-git10

introduction_to_amazon_algorithms/jumpstart-foundation-models/llama-2-chat-completion.ipynb

@@ -111,7 +111,7 @@
    "source": [
     "from sagemaker.jumpstart.model import JumpStartModel\n",
     "\n",
-    "model = JumpStartModel(model_id=model_id)\n",
+    "model = JumpStartModel(model_id=model_id, model_version=model_version)\n",
     "predictor = model.deploy()"
    ]
   },

introduction_to_amazon_algorithms/jumpstart-foundation-models/llama-2-finetuning.ipynb

@@ -119,7 +119,7 @@
    },
    "outputs": [],
    "source": [
-    "model_id, model_version = \"meta-textgeneration-llama-2-7b\", \"*\""
+    "model_id, model_version = \"meta-textgeneration-llama-2-7b\", \"2.*\""
    ]
   },
   {
@@ -133,7 +133,7 @@
    "source": [
     "from sagemaker.jumpstart.model import JumpStartModel\n",
     "\n",
-    "pretrained_model = JumpStartModel(model_id=model_id)\n",
+    "pretrained_model = JumpStartModel(model_id=model_id, model_version=model_version)\n",
     "pretrained_predictor = pretrained_model.deploy()"
    ]
   },
@@ -358,6 +358,7 @@
     "\n",
     "estimator = JumpStartEstimator(\n",
     "    model_id=model_id,\n",
+    "    model_version=model_version,\n",
     "    environment={\"accept_eula\": \"true\"},\n",
     "    disable_output_compression=True,  # For Llama-2-70b, add instance_type = \"ml.g5.48xlarge\"\n",
     ")\n",

introduction_to_amazon_algorithms/jumpstart-foundation-models/llama-2-text-completion.ipynb

@@ -84,12 +84,9 @@
    },
    "outputs": [],
    "source": [
-    "(\n",
-    "    model_id,\n",
-    "    model_version,\n",
-    ") = (\n",
+    "(model_id, model_version,) = (\n",
     "    \"meta-textgeneration-llama-2-7b\",\n",
-    "    \"*\",\n",
+    "    \"2.*\",\n",
     ")"
    ]
   },
@@ -116,7 +113,7 @@
    "source": [
     "from sagemaker.jumpstart.model import JumpStartModel\n",
     "\n",
-    "model = JumpStartModel(model_id=model_id)\n",
+    "model = JumpStartModel(model_id=model_id, model_version=model_version)\n",
     "predictor = model.deploy()"
    ]
   },