diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-09 21:06:12 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-09 21:06:12 +0000 |
| commit | d0184b8f76ce492da699f1926f34b57cd095242e (patch) | |
| tree | e8f121d5ad72266c7ebea52e25f34d7c97cdb2d7 /modules/textual_inversion/textual_inversion.py | |
| parent | 5d12ec82d3e13f5ff4c55db2930e4e10aed7015a (diff) | |
| download | stable-diffusion-webui-gfx803-d0184b8f76ce492da699f1926f34b57cd095242e.tar.gz stable-diffusion-webui-gfx803-d0184b8f76ce492da699f1926f34b57cd095242e.tar.bz2 stable-diffusion-webui-gfx803-d0184b8f76ce492da699f1926f34b57cd095242e.zip | |
change json tensor key name
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 44d4e08b..ae8d207d 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -19,15 +19,15 @@ import modules.textual_inversion.dataset class EmbeddingEncoder(json.JSONEncoder):
def default(self, obj):
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)
class EmbeddingDecoder(json.JSONDecoder):
def __init__(self, *args, **kwargs):
json.JSONDecoder.__init__(self, object_hook=self.object_hook, *args, **kwargs)
def object_hook(self, d):
- if 'EMBEDDINGTENSOR' in d:
- return torch.from_numpy(np.array(d['EMBEDDINGTENSOR']))
+ if 'TORCHTENSOR' in d:
+ return torch.from_numpy(np.array(d['TORCHTENSOR']))
return d
def embeddingToB64(data):
|