Feat: rwkv improvements: (#937)

This commit is contained in:
Dave authored and GitHub committed 2023-08-22 18:48:06 +02:00
1 parent 0d6165e481
commit 901f0709c5
7 files changed
+208 -150

No files matched your search

+2
View File
@@ -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,
}
}
+3
View File
@@ -44,4 +44,7 @@ type PredictionOptions struct {
// Diffusers
ClipSkip int `json:"clip_skip" yaml:"clip_skip"`
// RWKV (?)
Tokenizer string `json:"tokenizer" yaml:"tokenizer"`
}