Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.

shirayu/tsukaima

Repository files navigation

tsukaima

License CI CodeQL Typos

Tsukaima is a tool to call local large language models (LLMs) using the existing OpenAI ChatGPT clients. Currently, the following models are supported.

How to use

$ python3 -m venv myvenv
$ source myvenv/bin/activate
(myvenv) $ pip install -U git+https://github.com/shirayu/tsukaima.git
(myvenv) $ wget https://raw.githubusercontent.com/shirayu/tsukaima/main/examples_config/rinna.json -O rinna.json
(myvenv) $ tsukaima --host 0.0.0.0 --port 6006 --config ./rinna.json

Set API endpoint to the address (Eg: http://0.0.0.0:6006/v1/chat/completions) to use ChatGPT clients such as BetterChatGPT.

Screenshot

Check other config examples.

Specification

  • Messages whose role is system will be ignored

Setting file format

Please read tsukaima.schema.schema

Tips

  • You may need install NCCL
    • sudo apt install libnccl-dev libnccl2

Reference

About

LLM via OpenAI ChatGPT API

Resources

License

Stars

Watchers

Forks

Sponsor this project

 

Packages

 
 
 

Contributors