Instructions to use dn6/RFDiffusion-3 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use dn6/RFDiffusion-3 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("dn6/RFDiffusion-3", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
| [build-system] | |
| requires = ["setuptools>=61.0", "wheel"] | |
| build-backend = "setuptools.build_meta" | |
| [project] | |
| name = "modular-rfdiffusion" | |
| version = "0.1.0" | |
| description = "Modular Diffusers Pipeline for RFDiffusion protein structure generation" | |
| readme = "README.md" | |
| license = {text = "Apache-2.0"} | |
| requires-python = ">=3.8" | |
| authors = [ | |
| {name = "Dhruv Nair"} | |
| ] | |
| keywords = ["diffusion", "protein", "rfdiffusion", "deep-learning", "pytorch"] | |
| classifiers = [ | |
| "Development Status :: 3 - Alpha", | |
| "Intended Audience :: Science/Research", | |
| "License :: OSI Approved :: Apache Software License", | |
| "Programming Language :: Python :: 3", | |
| "Programming Language :: Python :: 3.8", | |
| "Programming Language :: Python :: 3.9", | |
| "Programming Language :: Python :: 3.10", | |
| "Programming Language :: Python :: 3.11", | |
| "Topic :: Scientific/Engineering :: Artificial Intelligence", | |
| "Topic :: Scientific/Engineering :: Bio-Informatics", | |
| ] | |
| dependencies = [ | |
| "torch>=2.0.0", | |
| "numpy>=1.21.0", | |
| "diffusers>=0.25.0", | |
| "huggingface-hub>=0.20.0", | |
| "scipy>=1.7.0", | |
| "hydra-core>=1.0.0", | |
| "omegaconf>=2.0.0", | |
| ] | |
| [project.optional-dependencies] | |
| dev = [ | |
| "pytest>=7.0.0", | |
| "pytest-cov>=4.0.0", | |
| "black>=23.0.0", | |
| "ruff>=0.1.0", | |
| "mypy>=1.0.0", | |
| ] | |
| [project.urls] | |
| Homepage = "https://github.com/DN6/modular-diffusers" | |
| Repository = "https://github.com/DN6/modular-diffusers" | |
| [tool.setuptools.packages.find] | |
| where = ["."] | |
| include = ["*"] | |
| [tool.black] | |
| line-length = 119 | |
| target-version = ["py38", "py39", "py310", "py311"] | |
| [tool.ruff] | |
| line-length = 119 | |
| target-version = "py38" | |
| [tool.ruff.lint] | |
| select = ["E", "F", "W", "I", "N"] | |
| ignore = ["E501"] | |