Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use simple lcg in xor

This commit is contained in:
DepFA 2022-10-11 04:17:36 +01:00 committed by GitHub
parent e0fbe6d27e
commit 7aa8fcac1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/textual_inversion/textual_inversion.py
View File

@ -39,9 +39,15 @@ def embeddingFromB64(data):
d = base64.b64decode(data) d = base64.b64decode(data)
return json.loads(d,cls=EmbeddingDecoder) 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): def xorBlock(block):
return np.bitwise_xor(block.astype(np.uint8), g = lcg()
((np.random.RandomState(0xDEADBEEF).random(block.shape)*255).astype(np.uint8)) & 0x0F ) 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): def styleBlock(block,sequence):
im = Image.new('RGB',(block.shape[1],block.shape[0])) im = Image.new('RGB',(block.shape[1],block.shape[0]))