fix directory permissions after install

Author: lstein

ldm/invoke/config/invokeai_configure.py

@@ -669,6 +669,12 @@ def initialize_rootdir(root: str, yes_to_all: bool = False):
                         dirs_exist_ok=True,
                         copy_function=shutil.copyfile,
                         )
+    # Fix up directory permissions so that they are writable
+    # This can happen when running under Nix environment which
+    # makes the runtime directory template immutable.
+    for root,dirs,_ in os.walk(os.path.join(root,name)):
+        for d in dirs:
+            Path(root,d).chmod(0o775)
 
 # -------------------------------------
 def run_console_ui(

ldm/invoke/config/invokeai_update.py

@@ -47,9 +47,13 @@ def do_post_install():
     '''
     Run postinstallation script.
     '''
-    from ldm.invoke.config.post_install.py import post_install
-    post_install()
-    
+    print("Looking for postinstallation script to run on this version...")
+    try:
+        from ldm.invoke.config.post_install.py import post_install
+        post_install()
+    except:
+        print("Postinstallation script not available for this version of InvokeAI")
+        
 def welcome(versions: dict):
 
     @group()

ldm/invoke/config/model_install_backend.py

@@ -394,6 +394,12 @@ def update_config_file(successfully_downloaded: dict, config_file: Path):
                         dirs_exist_ok=True,
                         copy_function=shutil.copyfile,
                         )
+    # Fix up directory permissions so that they are writable
+    # This can happen when running under Nix environment which
+    # makes the runtime directory template immutable.
+    for root,dirs,_ in os.walk(default_config_file().parent):
+        for d in dirs:
+            Path(root,d).chmod(0o775)
 
     yaml = new_config_file_contents(successfully_downloaded, config_file)
 

pyproject.toml

@@ -34,7 +34,7 @@ dependencies = [
   "clip_anytorch",
   "compel~=1.1.0",
   "datasets",
-  "diffusers[torch]~=0.14",
+  "diffusers[torch]==0.14",
   "dnspython==2.2.1",
   "einops",
   "eventlet",