Skip to content

Commit

Permalink
env: n_ctx must be an int
Browse files Browse the repository at this point in the history
  • Loading branch information
saghul committed May 16, 2024
1 parent 18d7b96 commit 6d82f01
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion skynet/env.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ def tobool(val: str | None):
# Some formats are auto-detected: https://github.com/abetlen/llama-cpp-python/blob/c50d3300d2a09c98765be7f2c05b7e4fd0b4232e/llama_cpp/llama_chat_format.py#L724
model_chat_format = os.environ.get('MODEL_CHAT_FORMAT')
llama_path = os.environ.get('LLAMA_PATH')
llama_n_ctx = os.environ.get('LLAMA_N_CTX', 8192)
llama_n_ctx = int(os.environ.get('LLAMA_N_CTX', 8192))
llama_n_gpu_layers = int(os.environ.get('LLAMA_N_GPU_LAYERS', -1 if is_mac else 40))
llama_n_batch = int(os.environ.get('LLAMA_N_BATCH', 512))

Expand Down

0 comments on commit 6d82f01

Please sign in to comment.