Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

change json tensor key name

Browse Source
This commit is contained in:
DepFA 2022-10-09 22:06:12 +01:00 committed by GitHub
parent 5d12ec82d3
commit d0184b8f76
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/textual_inversion/textual_inversion.py
View File

@ -19,15 +19,15 @@ import modules.textual_inversion.dataset
class EmbeddingEncoder(json.JSONEncoder): class EmbeddingEncoder(json.JSONEncoder):
def default(self, obj): def default(self, obj):
if isinstance(obj, torch.Tensor): if isinstance(obj, torch.Tensor):
return {'EMBEDDINGTENSOR':obj.cpu().detach().numpy().tolist()} return {'TORCHTENSOR':obj.cpu().detach().numpy().tolist()}
return json.JSONEncoder.default(self, o) return json.JSONEncoder.default(self, o)
class EmbeddingDecoder(json.JSONDecoder): class EmbeddingDecoder(json.JSONDecoder):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
json.JSONDecoder.__init__(self, object_hook=self.object_hook, *args, **kwargs) json.JSONDecoder.__init__(self, object_hook=self.object_hook, *args, **kwargs)
def object_hook(self, d): def object_hook(self, d):
if 'EMBEDDINGTENSOR' in d: if 'TORCHTENSOR' in d:
return torch.from_numpy(np.array(d['EMBEDDINGTENSOR'])) return torch.from_numpy(np.array(d['TORCHTENSOR']))
return d return d
def embeddingToB64(data): def embeddingToB64(data):