From 9a68108bdffb338ab4c804b6fd2344102c639f6b Mon Sep 17 00:00:00 2001 From: oksanabaza Date: Tue, 7 May 2024 10:18:15 +0100 Subject: [PATCH] prep --- demo-notebooks/additional-demos/hf_interactive.ipynb | 1 + demo-notebooks/additional-demos/local_interactive.ipynb | 1 + 2 files changed, 2 insertions(+) diff --git a/demo-notebooks/additional-demos/hf_interactive.ipynb b/demo-notebooks/additional-demos/hf_interactive.ipynb index 9181f3aac..492c70c81 100644 --- 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 index 36adfb500..f11906772 100644 --- 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",