Skip to content
This repository has been archived by the owner on Apr 3, 2024. It is now read-only.

Latest commit

 

History

History
28 lines (18 loc) · 645 Bytes

README.md

File metadata and controls

28 lines (18 loc) · 645 Bytes

Open In Colab

A headless Oobabooga wrapper.


pip install ooba
import ooba

path = ooba.download("https://huggingface.co/TheBloke/Mistral-7B-Instruct-v0.1-GGUF")

llm = ooba.llm(path)
messages = [{"role": "user", "content": "Hi Mistral."}]

for token in llm.chat(messages):
    print(token)

ooba.product.demo.mp4