Skip to content

Commit 1c6e6b9

Browse files
committed
Restored allowed random seed range (entropy reduction applied only where needed)
1 parent b150931 commit 1c6e6b9

File tree

3 files changed

+10
-3
lines changed

3 files changed

+10
-3
lines changed

modules/constants.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
SWITCH_SPEED = 20
44
SWITCH_QUALITY = 40
55

6-
# limited by modules\expansion.py -> transformers\trainer_utils.py
76
MIN_SEED = 0
8-
MAX_SEED = 2**32 - 1
7+
MAX_SEED = 2**63 - 1
8+
9+
# exclusive, needed by modules\expansion.py -> transformers\trainer_utils.py -> np.random.seed()
10+
SEED_LIMIT_NUMPY = 2**32

modules/expansion.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import torch
22
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, set_seed
33
from modules.path import fooocus_expansion_path
4+
import modules.constants as constants
45

56

67
fooocus_magic_split = [
@@ -31,7 +32,7 @@ def __init__(self):
3132
print('Fooocus Expansion engine loaded.')
3233

3334
def __call__(self, prompt, seed):
34-
seed = int(seed)
35+
seed = int(seed) % constants.SEED_LIMIT_NUMPY
3536
set_seed(seed)
3637

3738
prompt = safe_str(prompt) + fooocus_magic_split[seed % len(fooocus_magic_split)]

update_log_mre.md

+4
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,7 @@
1+
### 1.0.61 MRE
2+
3+
* Restored allowed random seed range (entropy reduction applied only to transformers / numpy related calls)
4+
15
### 1.0.51 MRE
26

37
* Added support for adjusting text prompt strengths (useful in Revision mode).

0 commit comments

Comments
 (0)