We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e70b91a commit 00ccb8dCopy full SHA for 00ccb8d
pkg/grpc/llm/llama/llama.go
@@ -17,9 +17,20 @@ type LLM struct {
17
}
18
19
func (llm *LLM) Load(opts *pb.ModelOptions) error {
20
+
21
+ ropeFreqBase := float32(10000)
22
+ ropeFreqScale := float32(1)
23
24
+ if opts.RopeFreqBase != 0 {
25
+ ropeFreqBase = opts.RopeFreqBase
26
+ }
27
+ if opts.RopeFreqScale != 0 {
28
+ ropeFreqScale = opts.RopeFreqScale
29
30
31
llamaOpts := []llama.ModelOption{
- llama.WithRopeFreqBase(opts.RopeFreqBase),
- llama.WithRopeFreqScale(opts.RopeFreqScale),
32
+ llama.WithRopeFreqBase(ropeFreqBase),
33
+ llama.WithRopeFreqScale(ropeFreqScale),
34
35
36
if opts.ContextSize != 0 {
0 commit comments