A package to make an LLM talk with itself forever.
from thendisnever.thend import isnever
isnever()
model_name
: The model to generate text with.- This is the default model.
- This must be a model compatible with AutoModelForCausalLM.
prompt
: The initial prompt for the model.- This is the inspiration for the default prompt.
max_memory_ratio
: The % of past tokens to remember.- This must be a real number between or equal to 0 and 1.
- When running
isnever()
for the first time, it will download the model and tokenizer from HuggingFace. This will take a while, but it only needs to be done once. - If you want to use the CPU (not recommended because it's slow, but it works), make sure you have PyTorch for CPU installed before installing this package.
Check out this guide that describes how to create a package with Typer.
Before testing, make sure to update the version:
pyproject.toml
->tool.poetry
src/thendisnever/__init__.py
->__version__
And the dependencies:
pyproject.toml
->tool.poetry.dependencies