diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-13 11:32:15 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-13 11:32:15 +0000 |
| commit | a176d89487d92f5a5b152401e5c424b34ff43b96 (patch) | |
| tree | 914910f9e24f3f489d1a273ec5abf4531dcd502c /modules/textual_inversion/image_embedding.py | |
| parent | 486bda9b331d054870e2b3551f94ece7aa39574d (diff) | |
| download | stable-diffusion-webui-gfx803-a176d89487d92f5a5b152401e5c424b34ff43b96.tar.gz stable-diffusion-webui-gfx803-a176d89487d92f5a5b152401e5c424b34ff43b96.tar.bz2 stable-diffusion-webui-gfx803-a176d89487d92f5a5b152401e5c424b34ff43b96.zip | |
print bucket sizes for training without resizing images #6620
fix an error when generating a picture with embedding in it
Diffstat (limited to 'modules/textual_inversion/image_embedding.py')
| -rw-r--r-- | modules/textual_inversion/image_embedding.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/textual_inversion/image_embedding.py b/modules/textual_inversion/image_embedding.py index ea653806..5593f88c 100644 --- a/modules/textual_inversion/image_embedding.py +++ b/modules/textual_inversion/image_embedding.py @@ -76,10 +76,10 @@ def insert_image_data_embed(image, data): next_size = data_np_low.shape[0] + (h-(data_np_low.shape[0] % h))
next_size = next_size + ((h*d)-(next_size % (h*d)))
- data_np_low.resize(next_size)
+ data_np_low = np.resize(data_np_low, next_size)
data_np_low = data_np_low.reshape((h, -1, d))
- data_np_high.resize(next_size)
+ data_np_high = np.resize(data_np_high, next_size)
data_np_high = data_np_high.reshape((h, -1, d))
edge_style = list(data['string_to_param'].values())[0].cpu().detach().numpy().tolist()[0][:1024]
|