aboutsummaryrefslogtreecommitdiff
path: root/modules/textual_inversion/ui.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2022-10-11 21:50:19 +0300
committerGitHub <noreply@github.com>2022-10-11 21:50:19 +0300
commit419e539fe33e5bd1f7c7f55b477f955f6bdfc822 (patch)
tree131ea9501cefd29b7e5cb4b3e1ea36d76c64beb6 /modules/textual_inversion/ui.py
parent2536ecbb1790da2af0d61b6a26f38732cba665cd (diff)
parentd7474a5185df2af84a93a12bc7e140d24e0fc516 (diff)
Merge branch 'learning_rate-scheduling' into learnschedule
Diffstat (limited to 'modules/textual_inversion/ui.py')
-rw-r--r--modules/textual_inversion/ui.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/textual_inversion/ui.py b/modules/textual_inversion/ui.py
index f19ac5e0..70f47343 100644
--- a/modules/textual_inversion/ui.py
+++ b/modules/textual_inversion/ui.py
@@ -23,6 +23,8 @@ def preprocess(*args):
def train_embedding(*args):
+ assert not shared.cmd_opts.lowvram and not shared.cmd_opts.medvram, 'Training models with lowvram or medvram is not possible'
+
try:
sd_hijack.undo_optimizations()