You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

ModelOptions.cs 3.3 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. using System.Text;
  2. using LLama.Abstractions;
  3. namespace LLama.Web.Common
  4. {
  5. public class ModelOptions
  6. : ILLamaParams
  7. {
  8. public string Name { get; set; }
  9. public int MaxInstances { get; set; }
  10. /// <summary>
  11. /// Model context size (n_ctx)
  12. /// </summary>
  13. public uint ContextSize { get; set; } = 512;
  14. /// <summary>
  15. /// the GPU that is used for scratch and small tensors
  16. /// </summary>
  17. public int MainGpu { get; set; } = 0;
  18. /// <summary>
  19. /// if true, reduce VRAM usage at the cost of performance
  20. /// </summary>
  21. public bool LowVram { get; set; } = false;
  22. /// <summary>
  23. /// Number of layers to run in VRAM / GPU memory (n_gpu_layers)
  24. /// </summary>
  25. public int GpuLayerCount { get; set; } = 20;
  26. /// <summary>
  27. /// Seed for the random number generator (seed)
  28. /// </summary>
  29. public uint Seed { get; set; } = 1686349486;
  30. /// <summary>
  31. /// Use f16 instead of f32 for memory kv (memory_f16)
  32. /// </summary>
  33. public bool UseFp16Memory { get; set; } = true;
  34. /// <summary>
  35. /// Use mmap for faster loads (use_mmap)
  36. /// </summary>
  37. public bool UseMemorymap { get; set; } = true;
  38. /// <summary>
  39. /// Use mlock to keep model in memory (use_mlock)
  40. /// </summary>
  41. public bool UseMemoryLock { get; set; } = false;
  42. /// <summary>
  43. /// Compute perplexity over the prompt (perplexity)
  44. /// </summary>
  45. public bool Perplexity { get; set; } = false;
  46. /// <summary>
  47. /// Model path (model)
  48. /// </summary>
  49. public string ModelPath { get; set; }
  50. /// <summary>
  51. /// List of LoRAs to apply
  52. /// </summary>
  53. public AdapterCollection LoraAdapters { get; set; } = new();
  54. /// <summary>
  55. /// base model path for the lora adapter (lora_base)
  56. /// </summary>
  57. public string LoraBase { get; set; } = string.Empty;
  58. /// <summary>
  59. /// Number of threads (-1 = autodetect) (n_threads)
  60. /// </summary>
  61. public int Threads { get; set; } = Math.Max(Environment.ProcessorCount / 2, 1);
  62. /// <summary>
  63. /// batch size for prompt processing (must be >=32 to use BLAS) (n_batch)
  64. /// </summary>
  65. public uint BatchSize { get; set; } = 512;
  66. /// <summary>
  67. /// Whether to convert eos to newline during the inference.
  68. /// </summary>
  69. public bool ConvertEosToNewLine { get; set; } = false;
  70. /// <summary>
  71. /// Whether to use embedding mode. (embedding) Note that if this is set to true,
  72. /// The LLamaModel won't produce text response anymore.
  73. /// </summary>
  74. public bool EmbeddingMode { get; set; } = false;
  75. /// <summary>
  76. /// how split tensors should be distributed across GPUs
  77. /// </summary>
  78. public float[] TensorSplits { get; set; }
  79. /// <summary>
  80. /// RoPE base frequency
  81. /// </summary>
  82. public float RopeFrequencyBase { get; set; } = 10000.0f;
  83. /// <summary>
  84. /// RoPE frequency scaling factor
  85. /// </summary>
  86. public float RopeFrequencyScale { get; set; } = 1.0f;
  87. /// <summary>
  88. /// Use experimental mul_mat_q kernels
  89. /// </summary>
  90. public bool MulMatQ { get; set; }
  91. /// <summary>
  92. /// The encoding to use for models
  93. /// </summary>
  94. public Encoding Encoding { get; set; } = Encoding.UTF8;
  95. /// <summary>
  96. /// Load vocab only (no weights)
  97. /// </summary>
  98. public bool VocabOnly { get; set; }
  99. }
  100. }