diff --git a/README.md b/README.md index a2668e70b..5e946f43d 100644 --- a/README.md +++ b/README.md @@ -93,7 +93,7 @@ You can reduce the completion to a choice between multiple possibilities: ``` python import outlines -model = outlines.models.transformers("mistralai/Mistral-7B") +model = outlines.models.transformers("mistralai/Mistral-7B-v0.1") prompt = """You are a sentiment-labelling assistant. Is the following review positive or negative? @@ -111,7 +111,7 @@ You can instruct the model to only return integers or floats: ``` python import outlines -model = outlines.models.transformers("mistralai/Mistral-7B") +model = outlines.models.transformers("mistralai/Mistral-7B-v0.1") prompt = "1+1=" answer = outlines.generate.format(model, int)(prompt) @@ -129,7 +129,7 @@ hood: ``` python import outlines -model = outlines.models.transformers("mistralai/Mistral-7B") +model = outlines.models.transformers("mistralai/Mistral-7B-v0.1") prompt = "Is 1+1=2? " unguided = outlines.generate.continuation(model, max_tokens=30)(prompt) @@ -151,7 +151,7 @@ print(guided) ``` python import outlines -model = outlines.models.transformers("mistralai/Mistral-7B") +model = outlines.models.transformers("mistralai/Mistral-7B-v0.1") prompt = "What is the IP address of the Google DNS servers? " unguided = generate.continuation(model, max_tokens=30)(prompt) @@ -211,7 +211,7 @@ class Character(BaseModel): strength: int -model = outlines.models.transformers("mistralai/Mistral-7B", device="cuda") +model = outlines.models.transformers("mistralai/Mistral-7B-v0.1", device="cuda") # Construct guided sequence generator generator = outlines.generate.json(model, Character, max_tokens=100) @@ -254,7 +254,7 @@ import outlines def add(a: int, b: int): return a + b -model = outlines.models.transformers("mistralai/Mistral-7B") +model = outlines.models.transformers("mistralai/Mistral-7B-v0.1") generator = outlines.generate.json(model, add) result = generator("Return two integers named a and b respectively. a is odd and b even.") @@ -298,7 +298,7 @@ def labelling(to_label, examples): {{ to_label }} // """ -model = outlines.models.transformers("mistralai/Mistral-7B") +model = outlines.models.transformers("mistralai/Mistral-7B-v0.1") prompt = labelling("Just awesome", examples) answer = outlines.generate.continuation(model, max_tokens=100)(prompt) ```