produce error when training with medvram/lowvram enabled
This commit is contained in:
parent
d682444ecc
commit
6d09b8d1df
2 changed files with 5 additions and 0 deletions
|
@ -25,6 +25,8 @@ def train_hypernetwork(*args):
|
|||
|
||||
initial_hypernetwork = shared.loaded_hypernetwork
|
||||
|
||||
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()
|
||||
|
||||
|
|
|
@ -22,6 +22,9 @@ 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()
|
||||
|
||||
|
|
Loading…
Reference in a new issue