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"],
    },
)