---
base_model:
- stable-diffusion-v1-5/stable-diffusion-v1-5
license: apache-2.0
pipeline_tag: text-to-image
library_name: diffusers
---
# 🌌 GalaxySD: Can AI Dream of Unseen Galaxies? 
# Conditional Diffusion Model for Galaxy Morphology Augmentation

## 🔍 Model Description

This repository provides the trained weights for **GalaxySD**, a fine-tuned Stable Diffusion model designed to generate high-fidelity galaxy images from text prompts. 
The model demonstrates strong performance in synthesizing visually realistic galaxies that adhere closely to specified morphological characteristics, such as spiral arms, bulge prominence, and edge-on orientation.

## 💻 How to Use

The file `galaxsd_final.safetensors` contains the fine-tuned Stable Diffusion weights for the **GalaxySD** model. To use this model in your own pipeline:

1. Make sure you have `diffusers`, `transformers`, `accelerate`, and `safetensors` installed:

```bash
pip install diffusers transformers accelerate safetensors
```

2. Load and run the model:
```bash
from diffusers import StableDiffusionPipeline
import torch

pipe = StableDiffusionPipeline.from_single_file(
    "galaxsd_final.safetensors",
    torch_dtype=torch.float16
).to("cuda")

prompt = "sdss, spiral galaxy, loosely wound spiral arms, 2 spiral arms"
image = pipe(prompt).images[0]
image.save("output.png")
```
If you encounter any issues, feel free to raise them in the [GalaxySD Repo](https://github.com/chenruiRae/GalaxySD) or contact authors.

## 🔗 Project Resources
- 🏠 [Homepage](https://galaxysd-webpage.streamlit.app/)
- 🚀 [GalaxySD Repository](https://github.com/chenruiRae/GalaxySD)
- 🛠️ [Trained Galaxy Embedding Tool](https://huggingface.co/CosmosDream/GalaxyEmb)
- 🗂️ [Training Dataset](https://zenodo.org/records/15669465)
- 📊 [A Contributed Catalog](https://zenodo.org/records/15636756)

## 📄 Citation
```
@article{Ma_2026,
        doi = {10.3847/1538-4365/ae1f10},
        url = {https://doi.org/10.3847/1538-4365/ae1f10},
        year = {2026},
        month = {jan},
        publisher = {The American Astronomical Society},
        volume = {282},
        number = {2},
        pages = {25},
        author = {Ma, Chenrui and Sun, Zechang and Jing, Tao and Cai, Zheng and Ting, Yuan-Sen and Huang, Song and Li, Mingyu},
        title = {Can AI Dream of Unseen Galaxies? Conditional Diffusion Model for Galaxy Morphology Augmentation},
        journal = {The Astrophysical Journal Supplement Series},
}
```