INNER CODE UNIT · Go

catalogFields

kaito-project/kaito · presets/workspace/generator/update_model_catalog/main.go:97

func catalogFields(e *generator.CatalogEntry) map[string]string {
	m := map[string]string{
		"modelFileSize":     e.ModelFileSize,
		"architectures":     fmt.Sprintf("%v", e.Architectures),
		"modelTokenLimit":   fmt.Sprintf("%d", e.ModelTokenLimit),
		"hiddenSize":        fmt.Sprintf("%d", e.HiddenSize),
		"numHiddenLayers":   fmt.Sprintf("%d", e.NumHiddenLayers),
		"numAttentionHeads": fmt.Sprintf("%d", e.NumAttentionHeads),
		"numKeyValueHeads":  fmt.Sprintf("%d", e.NumKeyValueHeads),
	}
	if e.Description != "" {
		m["description"] = e.Description
	}
	if e.License != "" {
		m["license"] = e.License
	}
	if e.PipelineTag != "" {
		m["pipelineTag"] = e.PipelineTag

View source record →