diff --git a/klite.embd b/klite.embd
index f7b6d8fe4..4b5f2e4d8 100644
--- a/klite.embd
+++ b/klite.embd
@@ -8652,7 +8652,7 @@ pre code,td,th{padding:0}pre code,table{background-color:transparent}.table,inpu
 	function togglepalmmodel()
 	{
 		let mdlname = document.getElementById("custom_palm_model").value;
-		if(mdlname.includes("gemini-1.5-") || mdlname.includes("gemini-exp-"))
+		if(mdlname.includes("gemini-2.") || mdlname.includes("gemini-1.5-") || mdlname.includes("gemini-exp-"))
 		{
 			document.getElementById("gemini_system_instruction").classList.remove("hidden");
 			document.getElementById("gemini_role_options").classList.remove("hidden");
@@ -13309,7 +13309,7 @@ pre code,td,th{padding:0}pre code,table{background-color:transparent}.table,inpu
 					oai_payload.logit_bias = submit_payload.params.logit_bias;
 				}
 
-				let is_using_o1 = custom_oai_model.toLowerCase().startsWith("o1-mini") || custom_oai_model.toLowerCase().startsWith("o1-preview");
+				let is_using_o1 = custom_oai_model.toLowerCase().startsWith("o1-") || custom_oai_model.toLowerCase()=="o1";
 				if(is_using_o1)
 				{
 					//o1 does not support ANY customization