wehub-resource-sync
142 lines
5.9 KiB
Swift
142 lines
5.9 KiB
Swift
//
|
|
// MixedSamplersView.swift
|
|
// MNNLLMiOS
|
|
//
|
|
// Created by 游薪渝(揽清) on 2025/3/24.
|
|
//
|
|
|
|
import SwiftUI
|
|
|
|
struct MixedSamplersView: View {
|
|
@Binding var selectedSamplers: Set<String>
|
|
@Binding var samplersOrder: [String]
|
|
let onUpdate: () -> Void
|
|
|
|
@Binding var temperature: Double
|
|
@Binding var topK: Double
|
|
@Binding var topP: Double
|
|
@Binding var minP: Double
|
|
@Binding var tfsZ: Double
|
|
@Binding var typical: Double
|
|
@Binding var penalty: Double
|
|
|
|
let onUpdateTemperature: (Double) -> Void
|
|
let onUpdateTopK: (Int) -> Void
|
|
let onUpdateTopP: (Double) -> Void
|
|
let onUpdateMinP: (Double) -> Void
|
|
let onUpdateTfsZ: (Double) -> Void
|
|
let onUpdateTypical: (Double) -> Void
|
|
let onUpdatePenalty: (Double) -> Void
|
|
|
|
var body: some View {
|
|
VStack(alignment: .leading, spacing: 10) {
|
|
Text("Configure mixed samplers:")
|
|
.font(.subheadline)
|
|
.foregroundColor(.secondary)
|
|
|
|
List {
|
|
ForEach(samplersOrder, id: \.self) { sampler in
|
|
VStack {
|
|
HStack {
|
|
switch sampler {
|
|
case "temperature":
|
|
ParameterSliderView(
|
|
title: "Temperature",
|
|
value: $temperature,
|
|
range: 0.0 ... 2.0,
|
|
format: "%.2f",
|
|
intValue: false,
|
|
onChanged: onUpdateTemperature
|
|
)
|
|
.disabled(!selectedSamplers.contains(sampler))
|
|
case "topK":
|
|
ParameterSliderView(
|
|
title: "Top K",
|
|
value: $topK,
|
|
range: 1 ... 100,
|
|
format: "%.0f",
|
|
intValue: true,
|
|
onChanged: { onUpdateTopK(Int($0)) }
|
|
)
|
|
.disabled(!selectedSamplers.contains(sampler))
|
|
case "topP":
|
|
ParameterSliderView(
|
|
title: "Top P",
|
|
value: $topP,
|
|
range: 0.0 ... 1.0,
|
|
format: "%.2f",
|
|
intValue: false,
|
|
onChanged: onUpdateTopP
|
|
)
|
|
.disabled(!selectedSamplers.contains(sampler))
|
|
case "minP":
|
|
ParameterSliderView(
|
|
title: "Min P",
|
|
value: $minP,
|
|
range: 0.05 ... 0.3,
|
|
format: "%.2f",
|
|
intValue: false,
|
|
onChanged: onUpdateMinP
|
|
)
|
|
.disabled(!selectedSamplers.contains(sampler))
|
|
case "tfs":
|
|
ParameterSliderView(
|
|
title: "TFS-Z",
|
|
value: $tfsZ,
|
|
range: 0.9 ... 0.99,
|
|
format: "%.2f",
|
|
intValue: false,
|
|
onChanged: onUpdateTfsZ
|
|
)
|
|
.disabled(!selectedSamplers.contains(sampler))
|
|
case "typical":
|
|
ParameterSliderView(
|
|
title: "Typical",
|
|
value: $typical,
|
|
range: 0.8 ... 0.95,
|
|
format: "%.2f",
|
|
intValue: false,
|
|
onChanged: onUpdateTypical
|
|
)
|
|
.disabled(!selectedSamplers.contains(sampler))
|
|
case "penalty":
|
|
ParameterSliderView(
|
|
title: "Penalty",
|
|
value: $penalty,
|
|
range: 0.0 ... 0.5,
|
|
format: "%.2f",
|
|
intValue: false,
|
|
onChanged: onUpdatePenalty
|
|
)
|
|
.disabled(!selectedSamplers.contains(sampler))
|
|
default:
|
|
EmptyView()
|
|
}
|
|
|
|
// Toggle switch
|
|
Toggle("", isOn: Binding(
|
|
get: { selectedSamplers.contains(sampler) },
|
|
set: { isSelected in
|
|
if isSelected {
|
|
selectedSamplers.insert(sampler)
|
|
} else {
|
|
selectedSamplers.remove(sampler)
|
|
}
|
|
onUpdate()
|
|
}
|
|
))
|
|
.labelsHidden()
|
|
}
|
|
.padding(.vertical, 4)
|
|
}
|
|
}
|
|
.onMove { from, to in
|
|
samplersOrder.move(fromOffsets: from, toOffset: to)
|
|
onUpdate()
|
|
}
|
|
}
|
|
.environment(\.editMode, .constant(.active))
|
|
}
|
|
}
|
|
}
|