ollamadiffuser 2.0.4__tar.gz → 2.0.6__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (78) hide show
  1. {ollamadiffuser-2.0.4/ollamadiffuser.egg-info → ollamadiffuser-2.0.6}/PKG-INFO +1 -1
  2. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/__init__.py +1 -1
  3. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/generic_strategy.py +15 -3
  4. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6/ollamadiffuser.egg-info}/PKG-INFO +1 -1
  5. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/pyproject.toml +1 -1
  6. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/CHANGELOG.md +0 -0
  7. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/LICENSE +0 -0
  8. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/MANIFEST.in +0 -0
  9. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/README.md +0 -0
  10. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/__main__.py +0 -0
  11. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/api/__init__.py +0 -0
  12. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/api/server.py +0 -0
  13. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/cli/__init__.py +0 -0
  14. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/cli/commands.py +0 -0
  15. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/cli/config_commands.py +0 -0
  16. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/cli/lora_commands.py +0 -0
  17. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/cli/main.py +0 -0
  18. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/cli/model_commands.py +0 -0
  19. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/cli/recommend_command.py +0 -0
  20. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/cli/registry_commands.py +0 -0
  21. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/__init__.py +0 -0
  22. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/config/__init__.py +0 -0
  23. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/config/model_registry.py +0 -0
  24. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/config/settings.py +0 -0
  25. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/__init__.py +0 -0
  26. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/base.py +0 -0
  27. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/engine.py +0 -0
  28. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/__init__.py +0 -0
  29. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/controlnet_strategy.py +0 -0
  30. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/flux_strategy.py +0 -0
  31. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/gguf_strategy.py +0 -0
  32. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/hidream_strategy.py +0 -0
  33. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/sd15_strategy.py +0 -0
  34. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/sd3_strategy.py +0 -0
  35. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/sdxl_strategy.py +0 -0
  36. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/video_strategy.py +0 -0
  37. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/models/__init__.py +0 -0
  38. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/models/gguf_loader.py +0 -0
  39. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/models/manager.py +0 -0
  40. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/utils/__init__.py +0 -0
  41. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/utils/controlnet_preprocessors.py +0 -0
  42. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/utils/download_utils.py +0 -0
  43. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/utils/lora_manager.py +0 -0
  44. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/mcp/__init__.py +0 -0
  45. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/mcp/server.py +0 -0
  46. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/__init__.py +0 -0
  47. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/canny/geometric_shapes.png +0 -0
  48. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/canny/house_outline.png +0 -0
  49. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/canny/portrait_outline.png +0 -0
  50. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/depth/linear_perspective.png +0 -0
  51. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/depth/radial_gradient.png +0 -0
  52. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/depth/sphere_3d.png +0 -0
  53. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/metadata.json +0 -0
  54. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/openpose/running_pose.png +0 -0
  55. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/openpose/sitting_pose.png +0 -0
  56. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/openpose/standing_pose.png +0 -0
  57. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/scribble/car_sketch.png +0 -0
  58. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/scribble/face_sketch.png +0 -0
  59. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/samples/scribble/tree_sketch.png +0 -0
  60. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/templates/index.html +0 -0
  61. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/ui/web.py +0 -0
  62. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/utils/__init__.py +0 -0
  63. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser.egg-info/SOURCES.txt +0 -0
  64. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser.egg-info/dependency_links.txt +0 -0
  65. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser.egg-info/entry_points.txt +0 -0
  66. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser.egg-info/not-zip-safe +0 -0
  67. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser.egg-info/requires.txt +0 -0
  68. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser.egg-info/top_level.txt +0 -0
  69. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/requirements.txt +0 -0
  70. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/setup.cfg +0 -0
  71. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/setup.py +0 -0
  72. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/tests/test_api_base64.py +0 -0
  73. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/tests/test_api_server.py +0 -0
  74. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/tests/test_engine.py +0 -0
  75. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/tests/test_mcp_server.py +0 -0
  76. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/tests/test_model_registry.py +0 -0
  77. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/tests/test_mps_support.py +0 -0
  78. {ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/tests/test_settings.py +0 -0
{ollamadiffuser-2.0.4/ollamadiffuser.egg-info → ollamadiffuser-2.0.6}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ollamadiffuser
3
- Version: 2.0.4
3
+ Version: 2.0.6
4
4
  Summary: Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX, and LoRA support
5
5
  Home-page: https://github.com/ollamadiffuser/ollamadiffuser
6
6
  Author: LocalKinAI Team
{ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/__init__.py RENAMED
@@ -4,7 +4,7 @@ OllamaDiffuser - Local AI Image Generation with Ollama-style CLI
4
4
  A tool for managing and running Stable Diffusion, FLUX.1, and other AI image generation models locally.
5
5
  """
6
6
 
7
- __version__ = "2.0.4"
7
+ __version__ = "2.0.6"
8
8
  __author__ = "LocalKinAI Team"
9
9
  __email__ = "contact@localkin.ai"
10
10
  __description__ = "🎨 Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX.1, and LoRA support"
{ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/ollamadiffuser/core/inference/strategies/generic_strategy.py RENAMED
@@ -64,11 +64,23 @@ class GenericPipelineStrategy(InferenceStrategy):
64
64
  load_kwargs = {"torch_dtype": dtype, "low_cpu_mem_usage": True}
65
65
  if dtype in (torch.float16, torch.bfloat16):
66
66
  load_kwargs["use_safetensors"] = True
67
+ if model_config.variant:
68
+ load_kwargs["variant"] = model_config.variant
67
69
 
68
70
  logger.info(f"Loading {pipeline_class_name} from {model_config.path} (dtype={dtype})")
69
- self.pipeline = pipeline_cls.from_pretrained(
70
- model_config.path, **load_kwargs
71
- )
71
+ try:
72
+ self.pipeline = pipeline_cls.from_pretrained(
73
+ model_config.path, **load_kwargs
74
+ )
75
+ except (OSError, ValueError):
76
+ if "variant" in load_kwargs:
77
+ logger.info(f"No {model_config.variant} variant files found, loading without variant")
78
+ load_kwargs.pop("variant")
79
+ self.pipeline = pipeline_cls.from_pretrained(
80
+ model_config.path, **load_kwargs
81
+ )
82
+ else:
83
+ raise
72
84
 
73
85
  # Device placement
74
86
  enable_offload = params.get("enable_cpu_offload", False)
{ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6/ollamadiffuser.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ollamadiffuser
3
- Version: 2.0.4
3
+ Version: 2.0.6
4
4
  Summary: Local AI Image Generation with Ollama-style CLI for Stable Diffusion, FLUX, and LoRA support
5
5
  Home-page: https://github.com/ollamadiffuser/ollamadiffuser
6
6
  Author: LocalKinAI Team
{ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "ollamadiffuser"
7
- version = "2.0.4"
7
+ version = "2.0.6"
8
8
  authors = [
9
9
  {name = "LocalKinAI Team", email = "contact@localkin.ai"}
10
10
  ]
{ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/LICENSE RENAMED
File without changes
{ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/README.md RENAMED
File without changes
{ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/setup.cfg RENAMED
File without changes
{ollamadiffuser-2.0.4 → ollamadiffuser-2.0.6}/setup.py RENAMED
File without changes