From 16a3b4233d3fb016a08d90624ab7a440993bcb25 Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Thu, 25 Jan 2024 15:13:32 -0800
Subject: [PATCH] Update initialization.py

---
 modules_forge/initialization.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/modules_forge/initialization.py b/modules_forge/initialization.py
index 02497014..02914592 100644
--- a/modules_forge/initialization.py
+++ b/modules_forge/initialization.py
@@ -22,4 +22,8 @@ def initialize_forge():
     import modules_forge.patch_precision
     modules_forge.patch_precision.patch_all_precision()
 
+    if model_management.directml_enabled:
+        model_management.lowvram_available = True
+        model_management.OOM_EXCEPTION = Exception
+
     return