Add requirements-win-torch-1-11-0.txt to fix Windows installs with PyTorch 1.12.1

Author: carson-katri

__init__.py

@@ -44,9 +44,10 @@
 from .property_groups.dream_prompt import DreamPrompt
 
 requirements_path_items = (
-    ('requirements-lin-AMD.txt', 'Linux (AMD)', 'Linux with AMD GPU'),
-    ('requirements-lin-win-colab-CUDA.txt', 'Linux/Windows (CUDA)', 'Linux or Windows with NVIDIA GPU'),
-    ('requirements-mac-MPS-CPU.txt', 'Apple Silicon', 'Apple M1/M2'),
+    # Use the old version of requirements-win.txt to fix installation issues with Blender + PyTorch 1.12.1
+    ('requirements-win-torch-1-11-0.txt', 'Linux/Windows (CUDA)', 'Linux or Windows with NVIDIA GPU'),
+    ('stable_diffusion/requirements-mac-MPS-CPU.txt', 'Apple Silicon', 'Apple M1/M2'),
+    ('stable_diffusion/requirements-lin-AMD.txt', 'Linux (AMD)', 'Linux with AMD GPU'),
 )
 
 def register():
@@ -59,7 +60,7 @@ def register():
     sys.path.append(absolute_path("stable_diffusion/src/taming-transformers"))
 
     set_dependencies_installed(False)
-    bpy.types.Scene.dream_textures_requirements_path = EnumProperty(name="Platform", items=requirements_path_items, description="Specifies which set of dependencies to install", default='requirements-mac-MPS-CPU.txt' if sys.platform == 'darwin' else 'requirements-lin-win-colab-CUDA.txt')
+    bpy.types.Scene.dream_textures_requirements_path = EnumProperty(name="Platform", items=requirements_path_items, description="Specifies which set of dependencies to install", default='stable_diffusion/requirements-mac-MPS-CPU.txt' if sys.platform == 'darwin' else 'requirements-win-torch-1-11-0.txt')
 
     register_section_props()
 

help_section.py

@@ -81,6 +81,7 @@ def faq_box(is_open, title, steps):
         title="It still doesn't work!",
         steps=[
             "Reinstall Blender 3.3 fresh from blender.org then try installing the addon again.",
+            "You can completely reset your Blender system by deleting 'C:\\Program Files\\Blender Foundation' and 'C:\\Users\\[YOU]\\AppData\\Roaming\\Blender Foundation'" if sys.platform == 'win32' else None,
             "If you are still unable to fix the issue, please file a bug on GitHub. Include the following:",
             "1. The logs of the error found from the system console",
             "2. Your graphics card specifications.",

operators/dream_texture.py

@@ -124,13 +124,14 @@ async def dream_texture(self, context):
         weights = absolute_path('stable_diffusion/' + models[model].weights)
 
         global generator
-        if generator is None:
+        if generator is None or generator.full_precision != context.scene.dream_textures_prompt.full_precision:
             generator = Generate(
                 conf=models_config,
                 model=model,
                 # These args are deprecated, but we need them to specify an absolute path to the weights.
                 weights=weights,
-                config=config
+                config=config,
+                full_precision=context.scene.dream_textures_prompt.full_precision
             )
             generator.load_model()
 
@@ -223,6 +224,9 @@ def perform():
                 # path to an initial image - its dimensions override width and height
                 init_img=init_img_path,
 
+                # generate tileable/seamless textures
+                seamless=scene.dream_textures_prompt.seamless,
+
                 fit=scene.dream_textures_prompt.fit,
                 # strength for noising/unnoising init_img. 0.0 preserves image exactly, 1.0 replaces it completely
                 strength=scene.dream_textures_prompt.strength,

operators/install_dependencies.py

@@ -117,10 +117,11 @@ def members(tf):
     requirements_path = requirements_txt
     if requirements_path is None:
         if sys.platform == 'darwin': # Use MPS dependencies list on macOS
-            requirements_path = 'requirements-mac-MPS-CPU.txt'
-        else: # Use CUDA dependencies by default on Linux/Windows
-            requirements_path = 'requirements-lin-win-colab-CUDA.txt'
-    subprocess.run([sys.executable, "-m", "pip", "install", "-r", absolute_path(f"stable_diffusion/{requirements_path}")], check=True, env=environ_copy, cwd=absolute_path("stable_diffusion/"))
+            requirements_path = 'stable_diffusion/requirements-mac-MPS-CPU.txt'
+        else: # Use CUDA dependencies by default on Linux/Windows.
+            # These are not the submodule dependencies from the `development` branch, but use the `main` branch deps for PyTorch 1.11.0.
+            requirements_path = 'requirements-win-torch-1-11-0.txt'
+    subprocess.run([sys.executable, "-m", "pip", "install", "-r", absolute_path(requirements_path), "--no-cache-dir"], check=True, env=environ_copy, cwd=absolute_path("stable_diffusion/"))
 
 class InstallDependencies(bpy.types.Operator):
     bl_idname = "stable_diffusion.install_dependencies"
@@ -131,6 +132,10 @@ class InstallDependencies(bpy.types.Operator):
     bl_options = {"REGISTER", "INTERNAL"}
 
     def execute(self, context):
+        # Open the console so we can watch the progress.
+        if sys.platform == 'win32':
+            bpy.ops.wm.console_toggle()
+
         try:
             install_pip()
             install_and_import_requirements(requirements_txt=context.scene.dream_textures_requirements_path)

preferences.py

@@ -86,9 +86,6 @@ def draw(self, context):
 
         weights_installed = os.path.exists(WEIGHTS_PATH)
 
-        warning_box = layout.box()
-        warning_box.label(text="10GB+ of VRAM is recommended. Smaller images are possible on lower end cards", icon="INFO")
-        
         if are_dependencies_installed() and weights_installed:
             layout.label(text="Addon setup complete", icon="CHECKMARK")
         else:

requirements-win-torch-1-11-0.txt

@@ -0,0 +1,33 @@
+albumentations==0.4.3
+einops==0.3.0
+huggingface-hub==0.8.1
+imageio-ffmpeg==0.4.2
+imageio==2.9.0
+kornia==0.6.0
+# pip will resolve the version which matches torch
+numpy
+omegaconf==2.1.1
+opencv-python==4.6.0.66
+pillow==9.2.0
+pip>=22
+pudb==2019.2
+pytorch-lightning==1.4.2
+streamlit==1.12.0
+# "CompVis/taming-transformers" doesn't work
+# ldm\models\autoencoder.py", line 6, in <module>
+#    from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
+# ModuleNotFoundError
+taming-transformers-rom1504==0.0.6
+test-tube>=0.7.5
+torch-fidelity==0.3.0
+torchmetrics==0.6.0
+transformers==4.19.2
+git+https://github.com/openai/CLIP.git@main#egg=clip
+git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion
+git+https://github.com/lstein/GFPGAN@fix-dark-cast-images#egg=gfpgan
+# No CUDA in PyPi builds
+--extra-index-url https://download.pytorch.org/whl/cu113 --trusted-host https://download.pytorch.org
+torch==1.11.0
+# Same as numpy - let pip do its thing
+torchvision
+-e .