nagolinc / aistoryteller Goto Github PK

This repo allows you to have your own choose-your-own-adventure stories using llama and stable-difufsion

usage: flaskApp.py [-h] [--model_id MODEL_ID] [--prompt_suffix PROMPT_SUFFIX] [--image_sizes IMAGE_SIZES [IMAGE_SIZES ...]] [--llm_model LLM_MODEL] [--failure_threshold FAILURE_THRESHOLD]

Generate a story

options: -h, --help show this help message and exit --model_id MODEL_ID model id --prompt_suffix PROMPT_SUFFIX prompt suffix --image_sizes IMAGE_SIZES [IMAGE_SIZES ...] image sizes --llm_model LLM_MODEL llm model --failure_threshold FAILURE_THRESHOLD this is the number that must be rolled on a 20 sided die to succeed

IMPORTANT! You will need to download a gguf lama model (for example from https://huggingface.co/TheBloke/Mistral-7B-OpenOrca-GGUF/) and pass it to flaskApp.py

for example

python flaskApp.py --llm_model mistral-7b-openorca.Q5_K_M.gguf