ollamadiffuser 2.0.5__py3-none-any.whl → 2.0.6__py3-none-any.whl

ollamadiffuser/__init__.py

@@ -4,7 +4,7 @@ OllamaDiffuser - Local AI Image Generation with Ollama-style CLI
 A tool for managing and running Stable Diffusion, FLUX.1, and other AI image generation models locally.
 """
 
-__version__ = "2.0.5"
+__version__ = "2.0.6"
 __author__ = "LocalKinAI Team"
 __email__ = "contact@localkin.ai"
 __description__ = "🎨 Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX.1, and LoRA support"

ollamadiffuser/core/inference/strategies/generic_strategy.py

@@ -68,9 +68,19 @@ class GenericPipelineStrategy(InferenceStrategy):
                 load_kwargs["variant"] = model_config.variant
 
             logger.info(f"Loading {pipeline_class_name} from {model_config.path} (dtype={dtype})")
-            self.pipeline = pipeline_cls.from_pretrained(
-                model_config.path, **load_kwargs
-            )
+            try:
+                self.pipeline = pipeline_cls.from_pretrained(
+                    model_config.path, **load_kwargs
+                )
+            except (OSError, ValueError):
+                if "variant" in load_kwargs:
+                    logger.info(f"No {model_config.variant} variant files found, loading without variant")
+                    load_kwargs.pop("variant")
+                    self.pipeline = pipeline_cls.from_pretrained(
+                        model_config.path, **load_kwargs
+                    )
+                else:
+                    raise
 
             # Device placement
             enable_offload = params.get("enable_cpu_offload", False)

{ollamadiffuser-2.0.5.dist-info → ollamadiffuser-2.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ollamadiffuser
-Version: 2.0.5
+Version: 2.0.6
 Summary: Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX, and LoRA support
 Home-page: https://github.com/ollamadiffuser/ollamadiffuser
 Author: LocalKinAI Team

{ollamadiffuser-2.0.5.dist-info → ollamadiffuser-2.0.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ollamadiffuser/__init__.py,sha256=yjQGtY5CBRV3GJ1AF-Dsqnt-UMYo0RayYBG8BuHjidw,1118
+ollamadiffuser/__init__.py,sha256=0eGsryuywTWlzKEhS98ci0ilAkC5-9TpSsxa1iSY9Po,1118
 ollamadiffuser/__main__.py,sha256=tNWMvEHq4ddtKLp7DrhIoOdnFw3F8RNrETC_u5xpkFI,141
 ollamadiffuser/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ollamadiffuser/api/server.py,sha256=MqEKjckz8x8pU9dhZlLXET5XWt7ERfVcAfvdVfDLFWw,15464
@@ -20,7 +20,7 @@ ollamadiffuser/core/inference/engine.py,sha256=xUCSQmGke9yAIoKGzh9jRaH1XE3GP9-uG
 ollamadiffuser/core/inference/strategies/__init__.py,sha256=5LQgTeS5JVin-HiGX7UvjlPzd0awombKrNhvAVr3SSw,53
 ollamadiffuser/core/inference/strategies/controlnet_strategy.py,sha256=_lGlCaYhrrdn7N6Aw0X9a4L90wKPjCrr6EBfQqPVH2E,6712
 ollamadiffuser/core/inference/strategies/flux_strategy.py,sha256=V11exUxKw256YGACGcLQD0Kzla6ZHF0HnxrKT5Ri1Cw,5351
-ollamadiffuser/core/inference/strategies/generic_strategy.py,sha256=loUDqgVIfYqSaIIY-1IRPeRRYJ6kU2yIgFaDJCzZ5zI,6228
+ollamadiffuser/core/inference/strategies/generic_strategy.py,sha256=CLriTGxv8ccvojB_ki97Cf5NcBl00i78DjxYQExcnLs,6691
 ollamadiffuser/core/inference/strategies/gguf_strategy.py,sha256=kIGT85tDCcSsliXdaxEJoQz4Gm7Xt7TfEcu6xcmTvJg,3893
 ollamadiffuser/core/inference/strategies/hidream_strategy.py,sha256=27XSeXQDvlDl7yMlVhkHzf7tdDr7udQyPyYowkDdZmM,3696
 ollamadiffuser/core/inference/strategies/sd15_strategy.py,sha256=qz5eGA2xkcA_3oNywP-rCliXzP7jYpH60728QmOT5fw,4966
@@ -53,9 +53,9 @@ ollamadiffuser/ui/samples/scribble/face_sketch.png,sha256=MVVYy_aS48xoS_RnIDzLUa
 ollamadiffuser/ui/samples/scribble/tree_sketch.png,sha256=3P-NGgW25xRwreDxiBYKcDhd2oHZAwKSkjNVM5oPTWY,3017
 ollamadiffuser/ui/templates/index.html,sha256=XcrYZqtDR65dAiu959Ea19t3MbtYmXl9PVyMnR1Telk,42358
 ollamadiffuser/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ollamadiffuser-2.0.5.dist-info/licenses/LICENSE,sha256=cnGL9l2P510Uk3TCnv62kot6vAfdSawhOZh7Y-oYoIE,1071
-ollamadiffuser-2.0.5.dist-info/METADATA,sha256=m8Tx6yEFssX6IP3RmU5IPPh3dUTVrB7UIEOvaZgRRy8,31084
-ollamadiffuser-2.0.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ollamadiffuser-2.0.5.dist-info/entry_points.txt,sha256=Bp-ZzV3F7QpQu02Mcafeza-oTMjDslomz9qrhvfcQUA,116
-ollamadiffuser-2.0.5.dist-info/top_level.txt,sha256=97wOGgTCxDE765Nr_o7B4Kwr_M_jy8fCCeQ81sMKlC4,15
-ollamadiffuser-2.0.5.dist-info/RECORD,,
+ollamadiffuser-2.0.6.dist-info/licenses/LICENSE,sha256=cnGL9l2P510Uk3TCnv62kot6vAfdSawhOZh7Y-oYoIE,1071
+ollamadiffuser-2.0.6.dist-info/METADATA,sha256=ix9N9vYQvdXxFTlXCACYTV-27Crnblp_rkNz6oNPF48,31084
+ollamadiffuser-2.0.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ollamadiffuser-2.0.6.dist-info/entry_points.txt,sha256=Bp-ZzV3F7QpQu02Mcafeza-oTMjDslomz9qrhvfcQUA,116
+ollamadiffuser-2.0.6.dist-info/top_level.txt,sha256=97wOGgTCxDE765Nr_o7B4Kwr_M_jy8fCCeQ81sMKlC4,15
+ollamadiffuser-2.0.6.dist-info/RECORD,,