Increased allowed random seed range

This commit is contained in:
MoonRide303
2023-10-23 12:27:12 +02:00
parent 23a7559db8
commit 81650a4305
5 changed files with 25 additions and 13 deletions
+2 -7
View File
@@ -123,13 +123,8 @@ def worker():
controlnet_cpds_path = None
clip_vision_path, ip_negative_path, ip_adapter_path = None, None, None
seed = image_seed
max_seed = int(1024 * 1024 * 1024)
if not isinstance(seed, int):
seed = random.randint(1, max_seed)
if seed < 0:
seed = - seed
seed = seed % max_seed
seed = int(image_seed)
print(f'[Parameters] Seed = {seed}')
if performance_selection == 'Speed':
steps = 30
+4 -1
View File
@@ -6,6 +6,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
from modules.path import fooocus_expansion_path
from fcbh.model_patcher import ModelPatcher
# limitation of np.random.seed(), called from transformers.set_seed()
SEED_LIMIT_NUMPY = 2**32
fooocus_magic_split = [
', extremely',
@@ -54,7 +57,7 @@ class FooocusExpansion:
print('Fooocus Expansion loaded by itself.')
model_management.load_model_gpu(self.patcher)
seed = int(seed)
seed = int(seed) % SEED_LIMIT_NUMPY
set_seed(seed)
origin = safe_str(prompt)
prompt = origin + fooocus_magic_split[seed % len(fooocus_magic_split)]