ollamadiffuser 2.0.3__tar.gz → 2.0.5__tar.gz

{ollamadiffuser-2.0.3/ollamadiffuser.egg-info → ollamadiffuser-2.0.5}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: ollamadiffuser
-Version: 2.0.3
+Version: 2.0.5
 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
-Author-email: OllamaDiffuser Team <ollamadiffuser@gmail.com>
+Author: LocalKinAI Team
+Author-email: LocalKinAI Team <contact@localkin.ai>
 License: MIT
 Project-URL: Homepage, https://www.ollamadiffuser.com/
 Project-URL: Website, https://www.ollamadiffuser.com/

{ollamadiffuser-2.0.3 → ollamadiffuser-2.0.5}/ollamadiffuser/__init__.py

@@ -4,9 +4,9 @@ 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.3"
-__author__ = "OllamaDiffuser Team"
-__email__ = "ollamadiffuser@gmail.com"
+__version__ = "2.0.5"
+__author__ = "LocalKinAI Team"
+__email__ = "contact@localkin.ai"
 __description__ = "🎨 Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX.1, and LoRA support"
 __url__ = "https://www.ollamadiffuser.com/"
 __repository__ = "https://github.com/ollamadiffuser/ollamadiffuser"

{ollamadiffuser-2.0.3 → ollamadiffuser-2.0.5}/ollamadiffuser/core/inference/strategies/flux_strategy.py

@@ -39,10 +39,8 @@ class FluxStrategy(InferenceStrategy):
                 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["torch_dtype"] = torch.bfloat16
                 load_kwargs["use_safetensors"] = True
-                load_kwargs["low_cpu_mem_usage"] = True
             else:
                 load_kwargs["torch_dtype"] = torch.bfloat16
                 load_kwargs["use_safetensors"] = True
@@ -107,9 +105,7 @@ class FluxStrategy(InferenceStrategy):
 
         max_seq_len = kwargs.get("max_sequence_length", params.get("max_sequence_length", 512))
 
-        # 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)
+        generator, used_seed = self._make_generator(seed, self.device)
 
         gen_kwargs = {
             "prompt": prompt,

{ollamadiffuser-2.0.3 → ollamadiffuser-2.0.5}/ollamadiffuser/core/inference/strategies/generic_strategy.py

@@ -64,6 +64,8 @@ class GenericPipelineStrategy(InferenceStrategy):
             load_kwargs = {"torch_dtype": dtype, "low_cpu_mem_usage": True}
             if dtype in (torch.float16, torch.bfloat16):
                 load_kwargs["use_safetensors"] = True
+            if model_config.variant:
+                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(

{ollamadiffuser-2.0.3 → ollamadiffuser-2.0.5}/ollamadiffuser/core/inference/strategies/hidream_strategy.py

@@ -35,8 +35,7 @@ class HiDreamStrategy(InferenceStrategy):
             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
+                load_kwargs["torch_dtype"] = torch.bfloat16
             else:
                 load_kwargs["torch_dtype"] = torch.bfloat16
 
@@ -79,8 +78,7 @@ 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))
 
-        gen_device = "cpu" if self.device == "mps" else self.device
-        generator, used_seed = self._make_generator(seed, gen_device)
+        generator, used_seed = self._make_generator(seed, self.device)
 
         gen_kwargs = {
             "prompt": prompt,

{ollamadiffuser-2.0.3 → ollamadiffuser-2.0.5/ollamadiffuser.egg-info}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: ollamadiffuser
-Version: 2.0.3
+Version: 2.0.5
 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
-Author-email: OllamaDiffuser Team <ollamadiffuser@gmail.com>
+Author: LocalKinAI Team
+Author-email: LocalKinAI Team <contact@localkin.ai>
 License: MIT
 Project-URL: Homepage, https://www.ollamadiffuser.com/
 Project-URL: Website, https://www.ollamadiffuser.com/

{ollamadiffuser-2.0.3 → ollamadiffuser-2.0.5}/pyproject.toml

@@ -4,9 +4,9 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ollamadiffuser"
-version = "2.0.3"
+version = "2.0.5"
 authors = [
-    {name = "OllamaDiffuser Team", email = "ollamadiffuser@gmail.com"}
+    {name = "LocalKinAI Team", email = "contact@localkin.ai"}
 ]
 description = "Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX, and LoRA support"
 readme = "README.md"

{ollamadiffuser-2.0.3 → ollamadiffuser-2.0.5}/setup.py

@@ -44,8 +44,8 @@ REQUIRED = [
 setup(
     name="ollamadiffuser",
     version=get_version(),
-    author="OllamaDiffuser Team",
-    author_email="ollamadiffuser@gmail.com",
+    author="LocalKinAI Team",
+    author_email="contact@localkin.ai",
     description="🎨 Ollama-like image generation model management tool with LoRA support",
     long_description=long_description,
     long_description_content_type="text/markdown",