Add llama server options struct

This commit is contained in:
2025-07-16 22:09:33 +02:00
parent 84d10b5ccc
commit 5c850c7534

166
server/pkg/options.go Normal file
View File

@@ -0,0 +1,166 @@
package llamactl
type InstanceOptions struct {
// Common params
VerbosePrompt bool `json:"verbose_prompt,omitempty"`
Threads int `json:"threads,omitempty"`
ThreadsBatch int `json:"threads_batch,omitempty"`
CPUMask string `json:"cpu_mask,omitempty"`
CPURange string `json:"cpu_range,omitempty"`
CPUStrict int `json:"cpu_strict,omitempty"`
Priority int `json:"priority,omitempty"`
Poll int `json:"poll,omitempty"`
CPUMaskBatch string `json:"cpu_mask_batch,omitempty"`
CPURangeBatch string `json:"cpu_range_batch,omitempty"`
CPUStrictBatch int `json:"cpu_strict_batch,omitempty"`
PriorityBatch int `json:"priority_batch,omitempty"`
PollBatch int `json:"poll_batch,omitempty"`
CtxSize int `json:"ctx_size,omitempty"`
Predict int `json:"predict,omitempty"`
BatchSize int `json:"batch_size,omitempty"`
UBatchSize int `json:"ubatch_size,omitempty"`
Keep int `json:"keep,omitempty"`
FlashAttn bool `json:"flash_attn,omitempty"`
NoPerf bool `json:"no_perf,omitempty"`
Escape bool `json:"escape,omitempty"`
NoEscape bool `json:"no_escape,omitempty"`
RopeScaling string `json:"rope_scaling,omitempty"`
RopeScale float64 `json:"rope_scale,omitempty"`
RopeFreqBase float64 `json:"rope_freq_base,omitempty"`
RopeFreqScale float64 `json:"rope_freq_scale,omitempty"`
YarnOrigCtx int `json:"yarn_orig_ctx,omitempty"`
YarnExtFactor float64 `json:"yarn_ext_factor,omitempty"`
YarnAttnFactor float64 `json:"yarn_attn_factor,omitempty"`
YarnBetaSlow float64 `json:"yarn_beta_slow,omitempty"`
YarnBetaFast float64 `json:"yarn_beta_fast,omitempty"`
DumpKVCache bool `json:"dump_kv_cache,omitempty"`
NoKVOffload bool `json:"no_kv_offload,omitempty"`
CacheTypeK string `json:"cache_type_k,omitempty"`
CacheTypeV string `json:"cache_type_v,omitempty"`
DefragThold float64 `json:"defrag_thold,omitempty"`
Parallel int `json:"parallel,omitempty"`
Mlock bool `json:"mlock,omitempty"`
NoMmap bool `json:"no_mmap,omitempty"`
Numa string `json:"numa,omitempty"`
Device string `json:"device,omitempty"`
OverrideTensor []string `json:"override_tensor,omitempty"`
GPULayers int `json:"gpu_layers,omitempty"`
SplitMode string `json:"split_mode,omitempty"`
TensorSplit string `json:"tensor_split,omitempty"`
MainGPU int `json:"main_gpu,omitempty"`
CheckTensors bool `json:"check_tensors,omitempty"`
OverrideKV []string `json:"override_kv,omitempty"`
Lora []string `json:"lora,omitempty"`
LoraScaled []string `json:"lora_scaled,omitempty"`
ControlVector []string `json:"control_vector,omitempty"`
ControlVectorScaled []string `json:"control_vector_scaled,omitempty"`
ControlVectorLayerRange string `json:"control_vector_layer_range,omitempty"`
Model string `json:"model,omitempty"`
ModelURL string `json:"model_url,omitempty"`
HFRepo string `json:"hf_repo,omitempty"`
HFRepoDraft string `json:"hf_repo_draft,omitempty"`
HFFile string `json:"hf_file,omitempty"`
HFRepoV string `json:"hf_repo_v,omitempty"`
HFFileV string `json:"hf_file_v,omitempty"`
HFToken string `json:"hf_token,omitempty"`
Verbose bool `json:"verbose,omitempty"`
Verbosity int `json:"verbosity,omitempty"`
// Sampling params
Samplers string `json:"samplers,omitempty"`
Seed int `json:"seed,omitempty"`
SamplingSeq string `json:"sampling_seq,omitempty"`
IgnoreEOS bool `json:"ignore_eos,omitempty"`
Temperature float64 `json:"temperature,omitempty"`
TopK int `json:"top_k,omitempty"`
TopP float64 `json:"top_p,omitempty"`
MinP float64 `json:"min_p,omitempty"`
XTCProbability float64 `json:"xtc_probability,omitempty"`
XTCThreshold float64 `json:"xtc_threshold,omitempty"`
Typical float64 `json:"typical,omitempty"`
RepeatLastN int `json:"repeat_last_n,omitempty"`
RepeatPenalty float64 `json:"repeat_penalty,omitempty"`
PresencePenalty float64 `json:"presence_penalty,omitempty"`
FrequencyPenalty float64 `json:"frequency_penalty,omitempty"`
DryMultiplier float64 `json:"dry_multiplier,omitempty"`
DryBase float64 `json:"dry_base,omitempty"`
DryAllowedLength int `json:"dry_allowed_length,omitempty"`
DryPenaltyLastN int `json:"dry_penalty_last_n,omitempty"`
DrySequenceBreaker []string `json:"dry_sequence_breaker,omitempty"`
DynatempRange float64 `json:"dynatemp_range,omitempty"`
DynatempExp float64 `json:"dynatemp_exp,omitempty"`
Mirostat int `json:"mirostat,omitempty"`
MirostatLR float64 `json:"mirostat_lr,omitempty"`
MirostatEnt float64 `json:"mirostat_ent,omitempty"`
LogitBias []string `json:"logit_bias,omitempty"`
Grammar string `json:"grammar,omitempty"`
GrammarFile string `json:"grammar_file,omitempty"`
JSONSchema string `json:"json_schema,omitempty"`
JSONSchemaFile string `json:"json_schema_file,omitempty"`
// Server/Example-specific params
NoContextShift bool `json:"no_context_shift,omitempty"`
Special bool `json:"special,omitempty"`
NoWarmup bool `json:"no_warmup,omitempty"`
SPMInfill bool `json:"spm_infill,omitempty"`
Pooling string `json:"pooling,omitempty"`
ContBatching bool `json:"cont_batching,omitempty"`
NoContBatching bool `json:"no_cont_batching,omitempty"`
MMProj string `json:"mmproj,omitempty"`
MMProjURL string `json:"mmproj_url,omitempty"`
NoMMProj bool `json:"no_mmproj,omitempty"`
NoMMProjOffload bool `json:"no_mmproj_offload,omitempty"`
Alias string `json:"alias,omitempty"`
Host string `json:"host,omitempty"`
Port int `json:"port,omitempty"`
Path string `json:"path,omitempty"`
NoWebUI bool `json:"no_webui,omitempty"`
Embedding bool `json:"embedding,omitempty"`
Reranking bool `json:"reranking,omitempty"`
APIKey string `json:"api_key,omitempty"`
APIKeyFile string `json:"api_key_file,omitempty"`
SSLKeyFile string `json:"ssl_key_file,omitempty"`
SSLCertFile string `json:"ssl_cert_file,omitempty"`
ChatTemplateKwargs string `json:"chat_template_kwargs,omitempty"`
Timeout int `json:"timeout,omitempty"`
ThreadsHTTP int `json:"threads_http,omitempty"`
CacheReuse int `json:"cache_reuse,omitempty"`
Metrics bool `json:"metrics,omitempty"`
Slots bool `json:"slots,omitempty"`
Props bool `json:"props,omitempty"`
NoSlots bool `json:"no_slots,omitempty"`
SlotSavePath string `json:"slot_save_path,omitempty"`
Jinja bool `json:"jinja,omitempty"`
ReasoningFormat string `json:"reasoning_format,omitempty"`
ReasoningBudget int `json:"reasoning_budget,omitempty"`
ChatTemplate string `json:"chat_template,omitempty"`
ChatTemplateFile string `json:"chat_template_file,omitempty"`
NoPrefillAssistant bool `json:"no_prefill_assistant,omitempty"`
SlotPromptSimilarity float64 `json:"slot_prompt_similarity,omitempty"`
LoraInitWithoutApply bool `json:"lora_init_without_apply,omitempty"`
// Speculative decoding params
DraftMax int `json:"draft_max,omitempty"`
DraftMin int `json:"draft_min,omitempty"`
DraftPMin float64 `json:"draft_p_min,omitempty"`
CtxSizeDraft int `json:"ctx_size_draft,omitempty"`
DeviceDraft string `json:"device_draft,omitempty"`
GPULayersDraft int `json:"gpu_layers_draft,omitempty"`
ModelDraft string `json:"model_draft,omitempty"`
CacheTypeKDraft string `json:"cache_type_k_draft,omitempty"`
CacheTypeVDraft string `json:"cache_type_v_draft,omitempty"`
// Audio/TTS params
ModelVocoder string `json:"model_vocoder,omitempty"`
TTSUseGuideTokens bool `json:"tts_use_guide_tokens,omitempty"`
// Default model params
EmbdBGESmallEnDefault bool `json:"embd_bge_small_en_default,omitempty"`
EmbdE5SmallEnDefault bool `json:"embd_e5_small_en_default,omitempty"`
EmbdGTESmallDefault bool `json:"embd_gte_small_default,omitempty"`
FIMQwen1_5BDefault bool `json:"fim_qwen_1_5b_default,omitempty"`
FIMQwen3BDefault bool `json:"fim_qwen_3b_default,omitempty"`
FIMQwen7BDefault bool `json:"fim_qwen_7b_default,omitempty"`
FIMQwen7BSpec bool `json:"fim_qwen_7b_spec,omitempty"`
FIMQwen14BSpec bool `json:"fim_qwen_14b_spec,omitempty"`
}