You can deploy this model and start earning money today!

CosmosDream/
GalaxySD

Name: GalaxySD
Availability: InStock

text-to-image

2506.16233

stable-diffusion-v1-5/stable-diffusion-v1-5

finetune

apache-2.0

Views

No views yet

🌌 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:

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

pip install diffusers transformers accelerate safetensors

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 or contact authors.

🔗 Project Resources

📄 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},
}

CosmosDream/GalaxySD