diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-11 03:17:36 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-11 03:17:36 +0000 |
| commit | 7aa8fcac1e45c3ad9c6a40df0e44a346afcd5032 (patch) | |
| tree | f98da3a58e515bd9755480e4ba821e804c291dcd /modules/textual_inversion/textual_inversion.py | |
| parent | e0fbe6d27e7b4505766c8cb5a4264e1114cf3721 (diff) | |
| download | stable-diffusion-webui-gfx803-7aa8fcac1e45c3ad9c6a40df0e44a346afcd5032.tar.gz stable-diffusion-webui-gfx803-7aa8fcac1e45c3ad9c6a40df0e44a346afcd5032.tar.bz2 stable-diffusion-webui-gfx803-7aa8fcac1e45c3ad9c6a40df0e44a346afcd5032.zip | |
use simple lcg in xor
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index ae807268..13416a08 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -39,9 +39,15 @@ def embeddingFromB64(data): d = base64.b64decode(data)
return json.loads(d,cls=EmbeddingDecoder)
+def lcg(m=2**32, a=1664525, c=1013904223, seed=0):
+ while True:
+ seed = (a * seed + c) % m
+ yield seed
+
def xorBlock(block):
- return np.bitwise_xor(block.astype(np.uint8),
- ((np.random.RandomState(0xDEADBEEF).random(block.shape)*255).astype(np.uint8)) & 0x0F )
+ g = lcg()
+ randblock = np.array([next(g) for _ in range(np.product(block.shape))]).astype(np.uint8).reshape(block.shape)
+ return np.bitwise_xor(block.astype(np.uint8),randblock & 0x0F)
def styleBlock(block,sequence):
im = Image.new('RGB',(block.shape[1],block.shape[0]))
|