Ettore Di Giacinto
c7119a2882
|
2 years ago | |
---|---|---|
.github/workflows | 2 years ago | |
client | 2 years ago | |
kubernetes | 2 years ago | |
.goreleaser.yaml | 2 years ago | |
Earthfile | 2 years ago | |
LICENSE | 2 years ago | |
README.md | 2 years ago | |
api.go | 2 years ago | |
go.mod | 2 years ago | |
go.sum | 2 years ago | |
interactive.go | 2 years ago | |
main.go | 2 years ago |
README.md
π« llama-cli
llama-cli is a straightforward golang CLI interface for llama.cpp, providing a simple API and a command line interface that allows text generation using a GPT-based model like llama directly from the terminal.
Container images
The llama-cli
container images come preloaded with the alpaca.cpp model, enabling you to start making predictions immediately! To begin, run:
docker run -ti --rm quay.io/go-skynet/llama-cli:v0.1 --instruction "What's an alpaca?" --topk 10000
You will receive a response like the following:
An alpaca is a member of the South American Camelid family, which includes the llama, guanaco and vicuΓ±a. It is a domesticated species that originates from the Andes mountain range in South America. Alpacas are used in the textile industry for their fleece, which is much softer than wool. Alpacas are also used for meat, milk, and fiber.
Basic usage
To use llama-cli, specify a pre-trained GPT-based model, an input text, and an instruction for text generation. llama-cli takes the following arguments when running from the CLI:
llama-cli --model <model_path> --instruction <instruction> [--input <input>] [--template <template_path>] [--tokens <num_tokens>] [--threads <num_threads>] [--temperature <temperature>] [--topp <top_p>] [--topk <top_k>]
Parameter | Environment Variable | Default Value | Description |
---|---|---|---|
template | TEMPLATE | A file containing a template for output formatting (optional). | |
instruction | INSTRUCTION | Input prompt text or instruction. "-" for STDIN. | |
input | INPUT | - | Path to text or "-" for STDIN. |
model | MODEL_PATH | The path to the pre-trained GPT-based model. | |
tokens | TOKENS | 128 | The maximum number of tokens to generate. |
threads | THREADS | NumCPU() | The number of threads to use for text generation. |
temperature | TEMPERATURE | 0.95 | Sampling temperature for model output. |
top_p | TOP_P | 0.85 | The cumulative probability for top-p sampling. |
top_k | TOP_K | 20 | The number of top-k tokens to consider for text generation. |
Here's an example of using llama-cli
:
llama-cli --model ~/ggml-alpaca-7b-q4.bin --instruction "What's an alpaca?"
This will generate text based on the given model and instruction.
Advanced usage
llama-cli
also provides an API for running text generation as a service.
Example of starting the API with docker
:
docker run -p 8080:8080 -ti --rm quay.io/go-skynet/llama-cli:v0.1 api
And you'll see:
βββββββββββββββββββββββββββββββββββββββββββββββββββββ
β Fiber v2.42.0 β
β http://127.0.0.1:8080 β
β (bound on host 0.0.0.0 and port 8080) β
β β
β Handlers ............. 1 Processes ........... 1 β
β Prefork ....... Disabled PID ................. 1 β
βββββββββββββββββββββββββββββββββββββββββββββββββββββ
You can control the API server options with command line arguments:
llama-cli api --model <model_path> [--address <address>] [--threads <num_threads>]
The API takes takes the following:
Parameter | Environment Variable | Default Value | Description |
---|---|---|---|
model | MODEL_PATH | The path to the pre-trained GPT-based model. | |
threads | THREADS | CPU cores | The number of threads to use for text generation. |
address | ADDRESS | :8080 | The address and port to listen on. |
Once the server is running, you can make requests to it using HTTP. For example, to generate text based on an instruction, you can send a POST request to the /predict
endpoint with the instruction as the request body:
curl --location --request POST 'http://localhost:8080/predict' --header 'Content-Type: application/json' --data-raw '{
"text": "What is an alpaca?",
"topP": 0.8,
"topK": 50,
"temperature": 0.7,
"tokens": 100
}'
Golang client API
The llama-cli
codebase has also a small client in go that can be used alongside with the api:
package main
import (
"fmt"
client "github.com/go-skynet/llama-cli/client"
)
func main() {
cli := client.NewClient("http://ip:30007")
out, err := cli.Predict("What's an alpaca?")
if err != nil {
panic(err)
}
fmt.Println(out)
}
Kubernetes
You can run the API directly in Kubernetes:
kubectl apply -f https://raw.githubusercontent.com/go-skynet/llama-cli/master/kubernetes/deployment.yaml