Fix bug where scene.node_tree doesn’t exist

benrugg · Apr 14, 2023 · 9fdd374 · 9fdd374
1 parent f6d6070
commit 9fdd374
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/__init__.py b/__init__.py
@@ -2,7 +2,7 @@
     "name": "AI Render - Stable Diffusion in Blender",
     "description": "Create amazing images using Stable Diffusion AI",
     "author": "Ben Rugg",
-    "version": (0, 7, 7),
+    "version": (0, 7, 8),
     "blender": (3, 0, 0),
     "location": "Render Properties > AI Render",
     "warning": "",

diff --git a/operators.py b/operators.py
@@ -35,9 +35,10 @@ def enable_air(scene):
 
 
 def mute_legacy_compositor_node_group(scene):
-    legacy_node_group = scene.node_tree.nodes.get('AIR')
-    if legacy_node_group:
-        legacy_node_group.mute = True
+    if scene.node_tree and scene.node_tree.nodes:
+        legacy_node_group = scene.node_tree.nodes.get('AIR')
+        if legacy_node_group:
+            legacy_node_group.mute = True
 
 
 def ensure_air_workspace():