wip: loraConfig

Author: mahdikhashan

examples/v1beta1/sdk/llm-hp-optimization.ipynb

@@ -338,7 +338,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 26,
+   "execution_count": 27,
    "id": "45c5a2476e1bffb7",
    "metadata": {
     "ExecuteTime": {
@@ -369,8 +369,8 @@
     "        num_train_epochs=1,\n",
     "    ),\n",
     "    lora_config=LoraConfig(\n",
-    "        r = 8,\n",
-    "        lora_alpha=8,\n",
+    "        r=katib.search.int(min=8, max=32),\n",
+    "        lora_alpha=16,\n",
     "        lora_dropout=0.1,\n",
     "        bias=\"none\",\n",
     "    )\n",