KaiChen1998's picture
Update README.md
1a4ce87 verified
---
license: mit
datasets:
- KaiChen1998/nuscenes2d-time-weather-geodiffusion
library_name: diffusers
tags:
- layout-to-image
---
# GeoDiffusion Model Card
GeoDiffusion is a detection data generation model fine-tuned from [Stable Diffusion v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5). This repo contains the GeoDiffusion checkpoint trained on **nuImages** dataset under **512x512** with conditional generation support for time-of-day and weather. Check detailed usage in our [Github repo](https://github.com/KaiChen1998/GeoDiffusion).