KaiChen1998 commited on
Commit
c109d37
1 Parent(s): 1aca529

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -3,6 +3,8 @@ license: mit
3
  datasets:
4
  - KaiChen1998/nuimages-geodiffusion
5
  library_name: diffusers
 
 
6
  ---
7
  # GeoDiffusion Model Card
8
  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).
 
3
  datasets:
4
  - KaiChen1998/nuimages-geodiffusion
5
  library_name: diffusers
6
+ tags:
7
+ - layout-to-image
8
  ---
9
  # GeoDiffusion Model Card
10
  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).