diff --git a/.DS_Store b/.DS_Store deleted file mode 100644 index c88a062..0000000 Binary files a/.DS_Store and /dev/null differ diff --git a/.gitignore b/.gitignore index d99d3f0..9b5eea2 100644 --- a/.gitignore +++ b/.gitignore @@ -2,3 +2,4 @@ __pycache__/ .venv workspace.ipynb dist +.DS_Store diff --git a/readme.md b/readme.md index dd98f5b..dd0d151 100644 --- a/readme.md +++ b/readme.md @@ -1,6 +1,6 @@ # Jsonformer: A Bulletproof Way to Generate Structured JSON from Language Models. -### Problem: Getting models to output structed JSON is hard +### Problem: Getting models to output structured JSON is hard ### Solution: Only generate the content tokens and fill in the fixed tokens @@ -31,7 +31,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer model = AutoModelForCausalLM.from_pretrained("databricks/dolly-v2-12b") tokenizer = AutoTokenizer.from_pretrained("databricks/dolly-v2-12b") -schema = { +json_schema = { "type": "object", "properties": { "name": {"type": "string"},