falcon.go 277 B

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