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
File size: 1,682 Bytes
4900749 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | [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"]
|