Skip to content

Latest commit

 

History

History
54 lines (31 loc) · 1.45 KB

README.md

File metadata and controls

54 lines (31 loc) · 1.45 KB
Raw

PyGPT4All

Official Python CPU inference for GPT4All language models based on llama.cpp and ggml

License: MIT

NB: Under active development

Installation

pip install pygpt4all

Usage

GPT4All model

Download a GPT4All model from http://gpt4all.io/models/ggml-gpt4all-l13b-snoozy.bin

from pygpt4all.models.gpt4all import GPT4All

def new_text_callback(text):
    print(text, end="")

model = GPT4All('./models/ggml-gpt4all-l13b-snoozy.bin')
model.generate("Once upon a time, ", n_predict=55, new_text_callback=new_text_callback)

GPT4All-J model

Download the GPT4All-J model from https://gpt4all.io/models/ggml-gpt4all-j-v1.3-groovy.bin

from pygpt4all.models.gpt4all_j import GPT4All_J

def new_text_callback(text):
    print(text, end="")

model = GPT4All_J('./models/ggml-gpt4all-j-v1.3-groovy.bin')
model.generate("Once upon a time, ", n_predict=55, new_text_callback=new_text_callback)

License

This project is licensed under the MIT License.