falcon.go 313 B

12345678910111213141516171819202122
  1. package llm
  2. const ModelFamilyFalcon = "falcon"
  3. const (
  4. falconModelType7B = 32
  5. falconModelType40B = 60
  6. falconModelType180B = 80
  7. )
  8. func falconModelType(numLayer uint32) string {
  9. switch numLayer {
  10. case 32:
  11. return "7B"
  12. case 60:
  13. return "40B"
  14. case 80:
  15. return "180B"
  16. default:
  17. return "Unknown"
  18. }
  19. }