Upload CELESTIAL training notebooks v2.0

CELESTIAL_Domain_Adapters_Training.ipynb

@@ -435,13 +435,18 @@
         "    trainer.save_model(adapter_path)\n",
         "    \n",
         "    # Save metadata\n",
+        "    lora_config_dict = lora_config.to_dict()\n",
+        "    # Convert any sets to lists for JSON serialization\n",
+        "    if 'target_modules' in lora_config_dict and isinstance(lora_config_dict['target_modules'], set):\n",
+        "        lora_config_dict['target_modules'] = list(lora_config_dict['target_modules'])\n",
+        "    \n",
         "    metadata = {\n",
         "        \"name\": f\"{domain_name}_adapter\",\n",
         "        \"domain\": domain_name,\n",
         "        \"training_examples\": len(domain_data),\n",
         "        \"base_model\": model_name,\n",
         "        \"created_at\": datetime.now().isoformat(),\n",
-        "        \"lora_config\": lora_config.to_dict()\n",
+        "        \"lora_config\": lora_config_dict\n",
         "    }\n",
         "    \n",
         "    with open(f\"{adapter_path}/adapter_metadata.json\", \"w\") as f:\n",

CELESTIAL_Mistral_7B_Training.ipynb

@@ -621,13 +621,18 @@
         "    os.makedirs(adapter_dir, exist_ok=True)\n",
         "    \n",
         "    # Save adapter metadata\n",
+        "    adapter_config_dict = adapter_config.to_dict()\n",
+        "    # Convert any sets to lists for JSON serialization\n",
+        "    if 'target_modules' in adapter_config_dict and isinstance(adapter_config_dict['target_modules'], set):\n",
+        "        adapter_config_dict['target_modules'] = list(adapter_config_dict['target_modules'])\n",
+        "    \n",
         "    metadata = {\n",
         "        \"name\": f\"{adapter_name}_adapter\",\n",
         "        \"description\": description,\n",
         "        \"domain\": adapter_name,\n",
         "        \"base_model\": model_name,\n",
         "        \"created_at\": datetime.now().isoformat(),\n",
-        "        \"config\": adapter_config.to_dict()\n",
+        "        \"config\": adapter_config_dict\n",
         "    }\n",
         "    \n",
         "    with open(f\"{adapter_dir}/adapter_metadata.json\", \"w\") as f:\n",