Fix lint

GoogleCloudPlatform · Apr 22, 2024 · 92a0289 · 92a0289
1 parent 68f467c
commit 92a0289
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/notebooks/community/model_garden/model_garden_pytorch_mistral_peft_tuning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_mistral_peft_tuning.ipynb
@@ -280,9 +280,7 @@
         "finetuning_precision_mode = \"float16\"\n",
         "\n",
         "# Worker pool spec for 4bit finetuning.\n",
-        "accelerator_type = (\n",
-        "    \"NVIDIA_L4\"  # @param[\"NVIDIA_TESLA_V100\", \"NVIDIA_L4\", \"NVIDIA_TESLA_A100\"]\n",
-        ")\n",
+        "accelerator_type = \"NVIDIA_L4\"  # @param[\"NVIDIA_TESLA_V100\", \"NVIDIA_L4\", \"NVIDIA_TESLA_A100\"]\n",
         "\n",
         "if accelerator_type == \"NVIDIA_TESLA_V100\":\n",
         "    machine_type = \"n1-highmem-16\"\n",