From 7c4ff7324122009070ab3320ccf81bc354e62029 Mon Sep 17 00:00:00 2001 From: Yannic Kilcher Date: Thu, 9 Feb 2023 09:20:00 +0100 Subject: [PATCH] corrected random seed for torch --- oasst-shared/oasst_shared/schemas/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/oasst-shared/oasst_shared/schemas/inference.py b/oasst-shared/oasst_shared/schemas/inference.py index a638e55c..96b05c7b 100644 --- a/oasst-shared/oasst_shared/schemas/inference.py +++ b/oasst-shared/oasst_shared/schemas/inference.py @@ -13,7 +13,7 @@ class WorkRequest(pydantic.BaseModel): conversation: protocol.Conversation = pydantic.Field(..., repr=False) model_name: str = "distilgpt2" max_new_tokens: int = 100 - seed: int = pydantic.Field(default_factory=lambda: random.randint(0, 2**31 - 1)) + seed: int = pydantic.Field(default_factory=lambda: random.randint(0, 0xFFFF_FFFF_FFFF_FFFF - 1)) do_sample: bool = True top_k: int = 50 top_p: float = 0.9