diff --git a/notebooks/kubeflow_pipelines/pipelines/labs/pipeline_vertex/tuning_lightweight_component.py b/notebooks/kubeflow_pipelines/pipelines/labs/pipeline_vertex/tuning_lightweight_component.py
index ba061591..76fc490a 100644
--- a/notebooks/kubeflow_pipelines/pipelines/labs/pipeline_vertex/tuning_lightweight_component.py
+++ b/notebooks/kubeflow_pipelines/pipelines/labs/pipeline_vertex/tuning_lightweight_component.py
@@ -49,8 +49,9 @@ def tune_hyperparameters(
         {
             "machine_spec": {
                 "machine_type": "n1-standard-4",
-                "accelerator_type": "NVIDIA_TESLA_K80",
-                "accelerator_count": 1,
+                # Enable if you want to use GPU.
+                # "accelerator_type": "NVIDIA_TESLA_K80",
+                # "accelerator_count": 1,
             },
             "replica_count": 1,
             "container_spec": {
diff --git a/notebooks/kubeflow_pipelines/pipelines/solutions/pipeline_vertex/tuning_lightweight_component.py b/notebooks/kubeflow_pipelines/pipelines/solutions/pipeline_vertex/tuning_lightweight_component.py
index 98c23093..1c4764c4 100644
--- a/notebooks/kubeflow_pipelines/pipelines/solutions/pipeline_vertex/tuning_lightweight_component.py
+++ b/notebooks/kubeflow_pipelines/pipelines/solutions/pipeline_vertex/tuning_lightweight_component.py
@@ -46,8 +46,9 @@ def tune_hyperparameters(
         {
             "machine_spec": {
                 "machine_type": "n1-standard-4",
-                "accelerator_type": "NVIDIA_TESLA_K80",
-                "accelerator_count": 1,
+                # Enable if you want to use GPU.
+                # "accelerator_type": "NVIDIA_TESLA_K80",
+                # "accelerator_count": 1,
             },
             "replica_count": 1,
             "container_spec": {