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.

IInferenceParams.cs 2.7 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. using System.Collections.Generic;
  2. using LLama.Common;
  3. using LLama.Native;
  4. namespace LLama.Abstractions
  5. {
  6. /// <summary>
  7. /// The paramters used for inference.
  8. /// </summary>
  9. public interface IInferenceParams
  10. {
  11. /// <summary>
  12. /// number of tokens to keep from initial prompt
  13. /// </summary>
  14. public int TokensKeep { get; set; }
  15. /// <summary>
  16. /// how many new tokens to predict (n_predict), set to -1 to inifinitely generate response
  17. /// until it complete.
  18. /// </summary>
  19. public int MaxTokens { get; set; }
  20. /// <summary>
  21. /// logit bias for specific tokens
  22. /// </summary>
  23. public Dictionary<int, float>? LogitBias { get; set; }
  24. /// <summary>
  25. /// Sequences where the model will stop generating further tokens.
  26. /// </summary>
  27. public IReadOnlyList<string> AntiPrompts { get; set; }
  28. /// <summary>
  29. /// 0 or lower to use vocab size
  30. /// </summary>
  31. public int TopK { get; set; }
  32. /// <summary>llama_eval
  33. /// 1.0 = disabled
  34. /// </summary>
  35. public float TopP { get; set; }
  36. /// <summary>llama_eval
  37. /// 0.0 = disabled
  38. /// </summary>
  39. public float MinP { get; set; }
  40. /// <summary>
  41. /// 1.0 = disabled
  42. /// </summary>
  43. public float TfsZ { get; set; }
  44. /// <summary>
  45. /// 1.0 = disabled
  46. /// </summary>
  47. public float TypicalP { get; set; }
  48. /// <summary>
  49. /// 1.0 = disabled
  50. /// </summary>
  51. public float Temperature { get; set; }
  52. /// <summary>
  53. /// 1.0 = disabled
  54. /// </summary>
  55. public float RepeatPenalty { get; set; }
  56. /// <summary>
  57. /// last n tokens to penalize (0 = disable penalty, -1 = context size) (repeat_last_n)
  58. /// </summary>
  59. public int RepeatLastTokensCount { get; set; }
  60. /// <summary>
  61. /// frequency penalty coefficient
  62. /// 0.0 = disabled
  63. /// </summary>
  64. public float FrequencyPenalty { get; set; }
  65. /// <summary>
  66. /// presence penalty coefficient
  67. /// 0.0 = disabled
  68. /// </summary>
  69. public float PresencePenalty { get; set; }
  70. /// <summary>
  71. /// Mirostat uses tokens instead of words.
  72. /// algorithm described in the paper https://arxiv.org/abs/2007.14966.
  73. /// 0 = disabled, 1 = mirostat, 2 = mirostat 2.0
  74. /// </summary>
  75. public MirostatType Mirostat { get; set; }
  76. /// <summary>
  77. /// target entropy
  78. /// </summary>
  79. public float MirostatTau { get; set; }
  80. /// <summary>
  81. /// learning rate
  82. /// </summary>
  83. public float MirostatEta { get; set; }
  84. /// <summary>
  85. /// consider newlines as a repeatable token (penalize_nl)
  86. /// </summary>
  87. public bool PenalizeNL { get; set; }
  88. /// <summary>
  89. /// Grammar to constrain possible tokens
  90. /// </summary>
  91. SafeLLamaGrammarHandle? Grammar { get; set; }
  92. }
  93. }