mirror of
https://github.com/mudler/LocalAI.git
synced 2026-10-23 22:59:00 -04:00
Feat: rwkv improvements: (#937)
This commit is contained in:
7 files changed
+208
-150
No files matched your search
@@ -60,6 +60,8 @@ func gRPCModelOpts(c config.Config) *pb.ModelOptions {
|
||||
Device: c.AutoGPTQ.Device,
|
||||
UseTriton: c.AutoGPTQ.Triton,
|
||||
UseFastTokenizer: c.AutoGPTQ.UseFastTokenizer,
|
||||
// RWKV
|
||||
Tokenizer: c.Tokenizer,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -44,4 +44,7 @@ type PredictionOptions struct {
|
||||
|
||||
// Diffusers
|
||||
ClipSkip int `json:"clip_skip" yaml:"clip_skip"`
|
||||
|
||||
// RWKV (?)
|
||||
Tokenizer string `json:"tokenizer" yaml:"tokenizer"`
|
||||
}
|
||||
Reference in new issue
Block a user