Chelsea / llm /config.py
CineAI's picture
Update llm/config.py
6daa3b2 verified
raw
history blame
136 Bytes
config = {"model": "mistralai/Mixtral-8x7B-Instruct-v0.1", "temperature": 0.1, "max_new_tokens": 1024, "top_k": 5, "load_in_8bit": True}