GeoSynth
Collection
8 items • Updated
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("MVRL/GeoSynth", dtype=torch.bfloat16, device_map="cuda")
prompt = "city neighborhood"
image = pipe(prompt).images[0]Model Card for Model ID
This is a StableDiffusion based model that synthesizes satellite images given text prompts. The base stable diffusion model used is stable-diffusion-2-1-base (v2-1_512-ema-pruned.ckpt).
- Use it with 🧨 diffusers
- Use it with stablediffusion repository
Model Sources [optional]
- Repository: stable-diffusion
Examples
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained("MVRL/GeoSynth")
pipe = pipe.to("cuda:0")
image = pipe(
"Satellite image features a city neighborhood",
).images[0]
image.save("generated_city.jpg")
- Downloads last month
- 340