Fix readme to reflect routing through enum

dottxt-ai · Jan 17, 2025 · 063291d · 063291d
1 parent 8b173a3
commit 063291d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -343,7 +343,7 @@ class Operation(Enum):
     mul = partial(mul)
 
 model = outlines.models.transformers("WizardLM/WizardMath-7B-V1.1")
-generator = outlines.generate.json(model, add)
+generator = outlines.generate.json(model, Operation)
 result = generator("Return json with two float named c and d respectively. c is negative and d greater than 1.0.")
 
 print(result)