prep

oksanabaza · May 7, 2024 · 9a68108 · 9a68108
1 parent c7ff66b
commit 9a68108
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/demo-notebooks/additional-demos/hf_interactive.ipynb b/demo-notebooks/additional-demos/hf_interactive.ipynb
@@ -92,6 +92,7 @@
     "# Create our cluster and submit\n",
     "# The SDK will try to find the name of your default local queue based on the annotation \"kueue.x-k8s.io/default-queue\": \"true\" unless you specify the local queue manually below\n",
     "cluster = Cluster(ClusterConfiguration(name='hfgputest', \n",
+    "                                       head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "                                       namespace=\"default\", # Update to your namespace\n",
     "                                       num_workers=1,\n",
     "                                       min_cpus=8, \n",

diff --git a/demo-notebooks/additional-demos/local_interactive.ipynb b/demo-notebooks/additional-demos/local_interactive.ipynb
@@ -55,6 +55,7 @@
     "\n",
     "cluster = Cluster(ClusterConfiguration(namespace=namespace,\n",
     "                                       name=cluster_name,\n",
+    "                                       head_gpus=1, # For GPU enabled workloads set the head_gpus and num_gpus\n",
     "                                       num_workers=1,\n",
     "                                       min_cpus=1,\n",
     "                                       max_cpus=1,\n",