update requirements

README.md

@@ -59,7 +59,6 @@ Hand-picked list of extensions that are deeply integrated into core workflows:
 - [Dynamic Thresholding](https://github.com/mcmonkeyprojects/sd-dynamic-thresholding)
 - [Steps Animation](https://github.com/vladmandic/sd-extension-steps-animation)
 - [Seed Travel](https://github.com/yownas/seed_travel)
-- [Model Keyword](https://github.com/mix1009/model-keyword)
 - [Multi-Diffusion Upscaler](https://github.com/pkuliyi2015/multidiffusion-upscaler-for-automatic1111)
 
 ### User Interface

modules/ui_extra_networks_textual_inversion.py

@@ -18,11 +18,12 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
         if len(embeddings) == 0: # maybe not loaded yet, so lets just look them up
             for root, _dirs, fns in os.walk(shared.opts.embeddings_dir):
                 for fn in fns:
-                    embedding = Embedding(0, fn)
-                    embedding.filename = os.path.join(root, fn)
-                    embeddings.append(embedding)
+                    if fn.lower().endswith(".pt"):
+                        embedding = Embedding(0, fn)
+                        embedding.filename = os.path.join(root, fn)
+                        embeddings.append(embedding)
         for embedding in embeddings:
-            path, ext = os.path.splitext(embedding.filename)
+            path, _ext = os.path.splitext(embedding.filename)
             yield {
                 "name": embedding.name,
                 "filename": embedding.filename,

requirements.txt

@@ -1,4 +1,3 @@
-accelerate
 addict
 aenum
 aiohttp
@@ -30,7 +29,6 @@ numpy
 omegaconf
 open-clip-torch
 opencv-contrib-python
-opencv-python
 piexif
 Pillow
 psutil
@@ -43,7 +41,6 @@ safetensors
 scikit-image
 scipy
 tb_nightly
-timm
 toml
 torch
 torchdiffeq
@@ -53,6 +50,9 @@ tqdm
 voluptuous
 yapf
 
+accelerate==0.18.0
+opencv-python==4.7.0.72
+timm==0.6.13
 diffusers==0.15.0
 einops==0.4.1
 gradio==3.23.0

setup.py

@@ -73,7 +73,7 @@ def installed(package, friendly: str = None):
                 if len(p) > 1:
                     ok = ok and version == p[1]
                     if not ok:
-                        log.warning(f"Package wrong version: {p[0]} required {p[1]}")
+                        log.warning(f"Package wrong version: {p[0]} {version} required {p[1]}")
             else:
                 log.debug(f"Package version not found: {p[0]}")
         return ok
@@ -192,7 +192,7 @@ def install_packages():
     clip_package = os.environ.get('CLIP_PACKAGE', "git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1")
     install(clip_package, 'clip')
     try:
-        xformers_package = os.environ.get('XFORMERS_PACKAGE', 'xformers')
+        xformers_package = os.environ.get('XFORMERS_PACKAGE', 'xformers==0.0.17')
         install(f'--no-deps {xformers_package}')
     except Exception as e:
         log.error(f'Cannot install xformers package: {e}')