From 276840a3bca7e80f33ccb55840e7e881e9e458e5 Mon Sep 17 00:00:00 2001 From: Alex Lew Date: Tue, 12 Sep 2023 02:37:41 -0400 Subject: [PATCH 1/2] Update README.md --- README.md | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/README.md b/README.md index 5452954..1cd9551 100644 --- a/README.md +++ b/README.md @@ -1,8 +1,6 @@ # LLaMPPL + HuggingFace -
[![docs](https://github.com/probcomp/hfppl/actions/workflows/docs.yml/badge.svg)](https://probcomp.github.io/hfppl) -
LLaMPPL is a research prototype for language model probabilistic programming: specifying language generation tasks by writing probabilistic programs that combine calls to LLMs, symbolic program logic, and probabilistic conditioning. To solve these tasks, LLaMPPL uses a specialized sequential Monte Carlo inference algorithm. This technique, SMC steering, is described in [our recent workshop abstract](https://arxiv.org/abs/2306.03081). @@ -96,4 +94,4 @@ Further documentation can be found at https://probcomp.github.io/hfppl. ## Installation To get started, clone this repository and run `pip install .` -The example `examples/hard_constraints.py` uses LLaMA2, which requires authorization from Meta. To run it, create a file `examples/hf_api_token.py`, with a constant `HF_AUTH_TOKEN` set to your HuggingFace API token. Alternatively, you can switch to a different (open) model by changing the first argument of `CachedCausalLM.from_pretrained` to another HuggingFace model identifier. \ No newline at end of file +The example `examples/hard_constraints.py` uses LLaMA2, which requires authorization from Meta. To run it, create a file `examples/hf_api_token.py`, with a constant `HF_AUTH_TOKEN` set to your HuggingFace API token. Alternatively, you can switch to a different (open) model by changing the first argument of `CachedCausalLM.from_pretrained` to another HuggingFace model identifier. From 279ce5c6f03062b3ac35216e58df4f356cd175a6 Mon Sep 17 00:00:00 2001 From: Alex Lew Date: Tue, 12 Sep 2023 03:44:25 -0400 Subject: [PATCH 2/2] Update mkdocs.yml --- mkdocs.yml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mkdocs.yml b/mkdocs.yml index 7ee3d27..9aef663 100644 --- a/mkdocs.yml +++ b/mkdocs.yml @@ -2,6 +2,8 @@ site_name: LLaMPPL Docs theme: name: "material" + palette: + scheme: slate markdown_extensions: - pymdownx.highlight: @@ -33,4 +35,4 @@ nav: - batching.md - caching.md - immutability.md - - Code Reference: reference/ \ No newline at end of file + - Code Reference: reference/