diff --git a/cmd/gpu-turnstile/main.go b/cmd/gpu-turnstile/main.go index c38b5b3..848054a 100644 --- a/cmd/gpu-turnstile/main.go +++ b/cmd/gpu-turnstile/main.go @@ -68,9 +68,9 @@ func parseFlags(args []string) (configPath string, install, remove, noCopy, help i++ case strings.HasPrefix(args[i], "-config="): configPath = strings.TrimPrefix(args[i], "-config=") - case args[i] == "--install-service" || args[i] == "-install-service": + case args[i] == "--install-service" || args[i] == "-install-service" || args[i] == "-i": install = true - case args[i] == "--remove-service" || args[i] == "-remove-service": + case args[i] == "--remove-service" || args[i] == "-remove-service" || args[i] == "-r": remove = true case args[i] == "--no-copy" || args[i] == "-no-copy": noCopy = true @@ -82,7 +82,7 @@ func parseFlags(args []string) (configPath string, install, remove, noCopy, help forceUpdate = true case args[i] == "--update-now" || args[i] == "-update-now": updateNow = true - case args[i] == "--monitor" || args[i] == "-monitor": + case args[i] == "--monitor" || args[i] == "-monitor" || args[i] == "-m": monitor = true case args[i] == "--elevated-child": elevatedChild = true @@ -100,15 +100,16 @@ const usageText = `GPU arbitration proxy for Ollama + ComfyUI Usage: gpu-turnstile -config run the proxy - gpu-turnstile --install-service [--no-copy] [-config path] install + start as a service - gpu-turnstile --remove-service stop + uninstall the service + gpu-turnstile -i | --install-service [--no-copy] [-config path] + install + start as a service + gpu-turnstile -r | --remove-service stop + uninstall the service gpu-turnstile -v | --version print just the version gpu-turnstile --force-update check for a signed update now, apply it and restart the service (no admin needed when the service runs) gpu-turnstile --update-now like --force-update, but only through the running service - gpu-turnstile --monitor live status view (downstreams, + gpu-turnstile -m | --monitor live status view (downstreams, GPU lock, queue); Ctrl+C quits gpu-turnstile -h | --help this help