lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 11:11:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update initialization.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 15:13:32 -08:00
parent ad778e688a
commit 16a3b4233d
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/initialization.py
View File

@@ -22,4 +22,8 @@ def initialize_forge():
import modules_forge.patch_precision import modules_forge.patch_precision
modules_forge.patch_precision.patch_all_precision() modules_forge.patch_precision.patch_all_precision()
if model_management.directml_enabled:
model_management.lowvram_available = True
model_management.OOM_EXCEPTION = Exception
return return