Use maxTemperature in session creation sample code

See also #53.
webmachinelearning · Oct 29, 2024 · 944f131 · 944f131
1 parent fa45f9d
commit 944f131
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -185,10 +185,14 @@ const customSession = await ai.languageModel.create({
 
 const capabilities = await ai.languageModel.capabilities();
 const slightlyHighTemperatureSession = await ai.languageModel.create({
-  temperature: Math.max(capabilities.defaultTemperature * 1.2, 1.0),
+  temperature: Math.max(
+    capabilities.defaultTemperature * 1.2,
+    capabilities.maxTemperature
+  ),
+  topK: 10
 });
 
-// capabilities also contains maxTemperature, defaultTopK, and maxTopK.
+// capabilities also contains defaultTopK and maxTopK.
 ```
 
 ### Session persistence and cloning