wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-07-12 00:40:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

corrected random seed for torch

Browse Source
This commit is contained in:
Yannic Kilcher
2023-02-09 09:20:00 +01:00
parent c53d8e9bce
commit 7c4ff73241
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oasst-shared/oasst_shared/schemas/inference.py
+1 -1
View File
@@ -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