| from setuptools import setup, find_packages | |
| setup( | |
| name="microforge", | |
| version="0.1.0", | |
| description="MicroForge: Mobile-First Image Generation with Recurrent Latent Planning", | |
| long_description=open("README.md").read(), | |
| long_description_content_type="text/markdown", | |
| packages=find_packages(), | |
| python_requires=">=3.8", | |
| install_requires=[ | |
| "torch>=2.0", | |
| "torchvision", | |
| "einops", | |
| ], | |
| extras_require={ | |
| "dev": ["matplotlib", "jupyter", "timm"], | |
| }, | |
| ) | |