Update ml/backend.go
This commit is contained in:
parent
cbeb2aab4f
commit
159821594c
@ -137,7 +137,7 @@ type YarnConfig struct {
|
|||||||
YarnBetaSlow float32 // Slow decay parameter for YaRN
|
YarnBetaSlow float32 // Slow decay parameter for YaRN
|
||||||
}
|
}
|
||||||
|
|
||||||
// DefaultYarnConfig returns a default configuration for YaRN (Yet Another Recurrent Network)
|
// DefaultYarnConfig returns a default configuration for YaRN (Yet Another Rope Extension)
|
||||||
func DefaultYarnConfig(nCtx int32) *YarnConfig {
|
func DefaultYarnConfig(nCtx int32) *YarnConfig {
|
||||||
return &YarnConfig{
|
return &YarnConfig{
|
||||||
YarnCtxTrain: int(nCtx),
|
YarnCtxTrain: int(nCtx),
|
||||||
|
Loading…
x
Reference in New Issue
Block a user