fix: 🚑️ TF by default fills vram

This was causing all kind of issues
melMass · Jul 14, 2023 · c225da5 · c225da5
1 parent b0fb522
commit c225da5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/__init__.py b/__init__.py
@@ -1,3 +1,7 @@
+import os
+
+os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
+
 import traceback
 from .log import log, blue_text, cyan_text, get_summary, get_label
 from .utils import here

diff --git a/web/color_widget.js b/web/color_widget.js
@@ -29,7 +29,7 @@ function getBrightness(rgbObj) {
 
 
 /**
- * @returns {import("/types/litegraph").IWidget} widget
+ * @returns {import("./types/litegraph").IWidget} widget
  */
 const custom = (key, val, compute = false) => {
     /** @type {import("/types/litegraph").IWidget} */