Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 16:17:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix random seeds to actually randomize

Browse Source 
Without this fix, manual seeds get locked in.
This commit is contained in:
Alex "mcmonkey" Goodwin 2023-04-10 06:29:10 -07:00
parent 1911504f82
commit 30befe492a
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/text_generation.py
View File

@ -1,3 +1,4 @@
import random
import re import re
import time import time
import traceback import traceback
@ -97,7 +98,8 @@ def formatted_outputs(reply, model_name):
def set_manual_seed(seed): def set_manual_seed(seed):
if seed != -1: if seed == -1:
seed = random.randint(1, 2**31)
torch.manual_seed(seed) torch.manual_seed(seed)
if torch.cuda.is_available(): if torch.cuda.is_available():
torch.cuda.manual_seed_all(seed) torch.cuda.manual_seed_all(seed)