import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("mespinosami/controlearth", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Model description
ControlNet model conditioned on OpenStreetMaps (OSM) to generate the corresponding satellite images.
Trained on the region of the Central Belt.
Dataset used for training
The dataset used for the training procedure is the WorldImagery Clarity dataset.
The code for the dataset construction can be accessed in https://github.com/miquel-espinosa/map-sat.
- Downloads last month
- -
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support