aboutsummaryrefslogtreecommitdiff
path: root/modules/textual_inversion
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-01-13 15:04:37 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-01-13 15:04:37 +0300
commit82725f0ac439f7e3b67858d55900e95330bbd326 (patch)
treea3e4c2be99cc12b6a8c2f92e69dad8965f350b5f /modules/textual_inversion
parent1849f6eb806f637f783b3beee3b48772da1cfab1 (diff)
fix a bug caused by merge
Diffstat (limited to 'modules/textual_inversion')
-rw-r--r--modules/textual_inversion/textual_inversion.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py
index 85210b0e..6939efcc 100644
--- a/modules/textual_inversion/textual_inversion.py
+++ b/modules/textual_inversion/textual_inversion.py
@@ -11,6 +11,7 @@ import datetime
import csv
import safetensors.torch
+import numpy as np
from PIL import Image, PngImagePlugin
from torch.utils.tensorboard import SummaryWriter