prep

Author: oksanabaza

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",

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",