ollamadiffuser 2.0.1__tar.gz → 2.0.2__tar.gz

{ollamadiffuser-2.0.1/ollamadiffuser.egg-info → ollamadiffuser-2.0.2}/PKG-INFO

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

{ollamadiffuser-2.0.1 → ollamadiffuser-2.0.2}/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.1"
+__version__ = "2.0.2"
 __author__ = "OllamaDiffuser Team"
 __email__ = "ollamadiffuser@gmail.com"
 __description__ = "🎨 Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX.1, and LoRA support"

{ollamadiffuser-2.0.1 → ollamadiffuser-2.0.2}/ollamadiffuser/core/inference/strategies/flux_strategy.py

@@ -38,6 +38,11 @@ class FluxStrategy(InferenceStrategy):
             if device == "cpu":
                 load_kwargs["torch_dtype"] = torch.float32
                 logger.warning("FLUX on CPU will be very slow for this 12B parameter model")
+            elif device == "mps":
+                # MPS has limited bfloat16 support; float16 avoids VAE decode crashes
+                load_kwargs["torch_dtype"] = torch.float16
+                load_kwargs["use_safetensors"] = True
+                load_kwargs["low_cpu_mem_usage"] = True
             else:
                 load_kwargs["torch_dtype"] = torch.bfloat16
                 load_kwargs["use_safetensors"] = True
@@ -102,7 +107,9 @@ class FluxStrategy(InferenceStrategy):
 
         max_seq_len = kwargs.get("max_sequence_length", params.get("max_sequence_length", 512))
 
-        generator, used_seed = self._make_generator(seed, self.device)
+        # CPU offload moves tensors between CPU/device; use CPU generator to avoid device mismatches
+        gen_device = "cpu" if self.device == "mps" else self.device
+        generator, used_seed = self._make_generator(seed, gen_device)
 
         gen_kwargs = {
             "prompt": prompt,

{ollamadiffuser-2.0.1 → ollamadiffuser-2.0.2}/ollamadiffuser/core/inference/strategies/hidream_strategy.py

@@ -34,6 +34,9 @@ class HiDreamStrategy(InferenceStrategy):
             load_kwargs = {**SAFETY_DISABLED_KWARGS}
             if device == "cpu":
                 load_kwargs["torch_dtype"] = torch.float32
+            elif device == "mps":
+                load_kwargs["torch_dtype"] = torch.float16
+                load_kwargs["low_cpu_mem_usage"] = True
             else:
                 load_kwargs["torch_dtype"] = torch.bfloat16
 
@@ -77,7 +80,8 @@ class HiDreamStrategy(InferenceStrategy):
         guidance = guidance_scale if guidance_scale is not None else params.get("guidance_scale", 5.0)
         max_seq_len = kwargs.get("max_sequence_length", params.get("max_sequence_length", 128))
 
-        generator, used_seed = self._make_generator(seed, self.device)
+        gen_device = "cpu" if self.device == "mps" else self.device
+        generator, used_seed = self._make_generator(seed, gen_device)
 
         gen_kwargs = {
             "prompt": prompt,

{ollamadiffuser-2.0.1 → ollamadiffuser-2.0.2/ollamadiffuser.egg-info}/PKG-INFO

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

{ollamadiffuser-2.0.1 → ollamadiffuser-2.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ollamadiffuser"
-version = "2.0.1"
+version = "2.0.2"
 authors = [
     {name = "OllamaDiffuser Team", email = "ollamadiffuser@gmail.com"}
 ]