Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 10:35:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

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

10
modules/text_generation.py
View file

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