Require argmaxtools 0.1.13 or greater

Author: ZachNagengast

requirements.txt

@@ -1,4 +1,4 @@
-argmaxtools
+argmaxtools>=0.1.13
 torch
 safetensors
 mlx

setup.py

@@ -1,6 +1,18 @@
+import os
+
 from setuptools import find_packages, setup
+from setuptools.command.install import install
+
+VERSION = "0.2.16"
+
+
+class VersionInstallCommand(install):
+    def run(self):
+        install.run(self)
+        version_file = os.path.join(self.install_lib, "diffusionkit", "version.py")
+        with open(version_file, "w") as f:
+            f.write(f"__version__ = '{VERSION}'\n")
 
-VERSION = "0.2.13"
 
 with open("README.md") as f:
     readme = f.read()
@@ -14,7 +26,7 @@
     long_description_content_type="text/markdown",
     author="Argmax, Inc.",
     install_requires=[
-        "argmaxtools",
+        "argmaxtools>=0.1.13",
         "torch",
         "safetensors",
         "mlx",
@@ -30,6 +42,9 @@
             "diffusionkit-cli=diffusionkit.mlx.scripts.generate_images:cli",
         ],
     },
+    cmdclass={
+        "install": VersionInstallCommand,
+    },
     classifiers=[
         "Development Status :: 4 - Beta",
         "Intended Audience :: Developers",