PlanetDiffusion / README.md
otrolie's picture
Fix usage example
9083655 verified
|
Raw History Blame Contribute Delete
1.24 kB
metadata
license: mit
library_name: diffusers
pipeline_tag: image-to-image
tags:
  - terrain
  - planet
  - diffusion
  - procedural-generation

PlanetDiffusion

Pretrained weights for Authoring Terrestrial Planets with Diffusion Models (doi:10.1111/cgf.70390). Code: https://github.com/Oliver-Borg/PlanetDiffusion

Models

Folder Description
PlanetDiffusion/final Tile diffusion model (TerrainDiffusionPipeline, UNet + DDIM scheduler + terrain encoder), planet size 5.

Each folder mirrors the local layout models/<name>/final. Only inference files are included (no optimizer state).

Usage

The code downloads the weights automatically when models/PlanetDiffusion/final is missing:

from src.models.diffusion.inpaint_inference import load_pipeline

pipeline = load_pipeline("models/PlanetDiffusion/final")

To download manually:

huggingface-cli download otrolie/PlanetDiffusion --include "PlanetDiffusion/final/*" --local-dir models

The model is loaded through the custom TerrainDiffusionPipeline in the code repository and cannot be loaded with the stock diffusers.DiffusionPipeline.

License

MIT, as in the code repository.