# DiffusionGPT

**Repository Path**: michael-wzhu91/DiffusionGPT

## Basic Information

- **Project Name**: DiffusionGPT
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-02-18
- **Last Updated**: 2024-02-18

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# DiffusionGPT: LLM-Driven Text-to-Image Generation System

This repository contains the pytorch codes of our paper "[DiffusionGPT: LLM-Driven Text-to-Image Generation System]()".

## Overview
Diffusion-GPT leverages Large Language Models (LLM) to offer a unified generation system capable of seamlessly accommodating various types of prompts and integrating domain-expert models.
![method](imgs/method.png)

## Demo
* [Spaces Demo 1](https://huggingface.co/spaces/DiffusionGPT/DiffusionGPT-XL)
* [Spaces Demo 2](https://huggingface.co/spaces/DiffusionGPT/DiffusionGPT)

## Environment

```bash
pip install -r requirements.txt
```

## Model Weights
All the checkpoints can get from [link](https://huggingface.co/spaces/DiffusionGPT/DiffusionGPT-XL/tree/main/checkpoints).

## Get Started

```bash
python DiffusionGPT-sdxl-app.py
```
or
```bash
python DiffusionGPT-sd15-app.py
```