Add a parallel sampling case (#34)

2024-01-18 06:29:43 +00:00
parent 501f944445
commit b240f75100
3 changed files with 42 additions and 2 deletions
@@ -50,7 +50,7 @@ def gen(
    regex: Optional[str] = None,
 ):
    if choices:
-        return SglSelect(name, choices, temperature)
+        return SglSelect(name, choices, 0.0 if temperature is None else temperature)

    # check regex is valid
    if regex is not None: