zhengjl/text-generation-webui
Public Access
1
0
Fork 0
Code Issues Pull Requests 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
Download Patch File Download Diff File Expand all files Collapse all files

10
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,10 +98,11 @@ def formatted_outputs(reply, model_name):
def set_manual_seed(seed): def set_manual_seed(seed):
if seed != -1: if seed == -1:
torch.manual_seed(seed) seed = random.randint(1, 2**31)
if torch.cuda.is_available(): torch.manual_seed(seed)
torch.cuda.manual_seed_all(seed) if torch.cuda.is_available():
torch.cuda.manual_seed_all(seed)
def stop_everything_event(): def stop_everything_event():