mirror of
https://github.com/nomic-ai/gpt4all.git
synced 2025-10-23 00:50:47 +00:00
Adapt code
This commit is contained in:
@@ -20,24 +20,14 @@ var DefaultOptions PredictOptions = PredictOptions{
|
||||
}
|
||||
|
||||
var DefaultModelOptions ModelOptions = ModelOptions{
|
||||
Threads: 4,
|
||||
ModelType: GPTJType,
|
||||
Threads: 4,
|
||||
}
|
||||
|
||||
type ModelOptions struct {
|
||||
Threads int
|
||||
ModelType ModelType
|
||||
Threads int
|
||||
}
|
||||
type ModelOption func(p *ModelOptions)
|
||||
|
||||
type ModelType int
|
||||
|
||||
const (
|
||||
LLaMAType ModelType = 0
|
||||
GPTJType ModelType = iota
|
||||
MPTType ModelType = iota
|
||||
)
|
||||
|
||||
// SetTokens sets the number of tokens to generate.
|
||||
func SetTokens(tokens int) PredictOption {
|
||||
return func(p *PredictOptions) {
|
||||
@@ -110,13 +100,6 @@ func SetThreads(c int) ModelOption {
|
||||
}
|
||||
}
|
||||
|
||||
// SetModelType sets the model type.
|
||||
func SetModelType(c ModelType) ModelOption {
|
||||
return func(p *ModelOptions) {
|
||||
p.ModelType = c
|
||||
}
|
||||
}
|
||||
|
||||
// Create a new PredictOptions object with the given options.
|
||||
func NewModelOptions(opts ...ModelOption) ModelOptions {
|
||||
p := DefaultModelOptions
|
||||
|
Reference in New Issue
Block a user