mirror of
https://github.com/coaidev/coai.git
synced 2025-05-20 05:20:15 +09:00
27 lines
655 B
Go
27 lines
655 B
Go
package oneapi
|
|
|
|
import (
|
|
"chat/globals"
|
|
)
|
|
|
|
var HitModels = []string{
|
|
globals.Claude1, globals.Claude1100k,
|
|
globals.Claude2, globals.Claude2100k,
|
|
globals.StableDiffusion,
|
|
globals.LLaMa270B, globals.LLaMa213B, globals.LLaMa27B,
|
|
globals.CodeLLaMa34B, globals.CodeLLaMa13B, globals.CodeLLaMa7B,
|
|
}
|
|
|
|
func (c *ChatInstance) GetToken(model string) int {
|
|
switch model {
|
|
case globals.Claude1, globals.Claude2:
|
|
return 5000
|
|
case globals.Claude2100k, globals.Claude1100k:
|
|
return 50000
|
|
case globals.LLaMa270B, globals.LLaMa213B, globals.LLaMa27B, globals.CodeLLaMa34B, globals.CodeLLaMa13B, globals.CodeLLaMa7B:
|
|
return 3000
|
|
default:
|
|
return 2500
|
|
}
|
|
}
|