Adding missing download in tensor parallelism

Richard8610 · Mar 7, 2024 · e87b00a · e87b00a
1 parent d3adef0
commit e87b00a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/docs/tutorial_notebooks/scaling/JAX/tensor_parallel_async.ipynb b/docs/tutorial_notebooks/scaling/JAX/tensor_parallel_async.ipynb
@@ -45,7 +45,7 @@
     "# Github URL where python scripts are stored.\n",
     "base_url = \"https://raw.githubusercontent.com/phlippe/uvadlc_notebooks/master/docs/tutorial_notebooks/scaling/JAX/\"\n",
     "# Files to download.\n",
-    "python_files = [\"single_gpu.py\", \"data_parallel.py\", \"tensor_parallel.py\", \"utils.py\"]\n",
+    "python_files = [\"single_gpu.py\", \"data_parallel.py\", \"pipeline_parallel.py\", \"tensor_parallel.py\", \"utils.py\"]\n",
     "# For each file, check whether it already exists. If not, try downloading it.\n",
     "for file_name in python_files:\n",
     "    if not os.path.isfile(file_name):\n",

diff --git a/docs/tutorial_notebooks/scaling/JAX/tensor_parallel_simple.ipynb b/docs/tutorial_notebooks/scaling/JAX/tensor_parallel_simple.ipynb
@@ -51,7 +51,7 @@
     "# Github URL where python scripts are stored.\n",
     "base_url = \"https://raw.githubusercontent.com/phlippe/uvadlc_notebooks/master/docs/tutorial_notebooks/scaling/JAX/\"\n",
     "# Files to download.\n",
-    "python_files = [\"single_gpu.py\", \"data_parallel.py\", \"utils.py\"]\n",
+    "python_files = [\"single_gpu.py\", \"data_parallel.py\", \"pipeline_parallel.py\", \"utils.py\"]\n",
     "# For each file, check whether it already exists. If not, try downloading it.\n",
     "for file_name in python_files:\n",
     "    if not os.path.isfile(file_name):\n",

diff --git a/docs/tutorial_notebooks/scaling/JAX/tensor_parallel_transformer.ipynb b/docs/tutorial_notebooks/scaling/JAX/tensor_parallel_transformer.ipynb
@@ -46,6 +46,7 @@
     "python_files = [\n",
     "    \"single_gpu.py\",\n",
     "    \"data_parallel.py\",\n",
+    "    \"pipeline_parallel.py\",\n",
     "    \"tensor_parallel.py\",\n",
     "    \"tensor_parallel_async.py\",\n",
     "    \"utils.py\",\n",