OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-05 05:02:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

produce error when training with medvram/lowvram enabled

Browse Source
This commit is contained in:
AUTOMATIC 2022-10-11 18:33:57 +03:00
parent d682444ecc
commit 6d09b8d1df
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/hypernetworks/ui.py
View File

@ -25,6 +25,8 @@ def train_hypernetwork(*args):
initial_hypernetwork = shared.loaded_hypernetwork 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: try:
sd_hijack.undo_optimizations() sd_hijack.undo_optimizations()

3
modules/textual_inversion/ui.py
View File

@ -22,6 +22,9 @@ def preprocess(*args):
def train_embedding(*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: try:
sd_hijack.undo_optimizations() sd_hijack.undo_optimizations()