INNER CODE UNIT · Go

main

kaito-project/kaito · cmd/preset-generator/main.go:28

func main() {
	token := flag.String("token", "", "Hugging Face API token")
	flag.Parse()

	if flag.NArg() < 1 {
		fmt.Println("Usage: preset-generator <model_repo>")
		os.Exit(1)
	}

	modelRepo := flag.Arg(0)

	param, err := generator.GeneratePreset(modelRepo, *token)
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
		os.Exit(1)
	}

	// Use yaml.MapSlice to force exact ordering to match Python output

View source record →