Explicit UI control to toggle sampling

It used to be top_k > 1
huggingface · Nov 11, 2023 · 75520a0 · 75520a0
1 parent 6895f4a
commit 75520a0
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/SwiftChat/ControlView.swift b/SwiftChat/ControlView.swift
@@ -51,7 +51,6 @@ struct ControlView: View {
                             CFloat(config.topK)
                         } set: {
                             config.topK = Int($0)
-                            config.doSample = config.topK > 1
                         }, in: 1...50, step: 1) {
                             Text("Top K")
                             Spacer()
@@ -77,10 +76,15 @@ struct ControlView: View {
                            Spacer()
                        }
                    }
-               }
-
+                }.disabled(!config.doSample)
+
+                HStack {
+                    Toggle(isOn: $config.doSample) { Text("Sample") }
+                    Spacer()
+                }
+
                 Divider()
-                    
+
 //                Group {
 //                    DisclosureGroup(isExpanded: $discloseAdvanced) {
 //                        Spacer()