File size: 1,252 Bytes
7f1fa19
 
d2a27fd
 
 
7f1fa19
668a48f
7f1fa19
d2a27fd
7f1fa19
d2a27fd
 
 
7f1fa19
 
4e09337
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
668a48f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
---
title: Ai Fast Image Server
emoji: 🌍
colorFrom: gray
colorTo: blue
sdk: gradio
sdk_version: 5.47.2
app_file: app.py
license: mit
pinned: false
suggested_hardware: a10g-small
duplicated_from: hysts/SD-XL
load_balancing_strategy: random
---

# AI Fast Image Server

A lightweight Gradio app that serves fast **text-to-image** generation using either:

- **SDXL Base 1.0 + LCM** (default), or  
- **SSD-1B + LCM LoRA** (enable via a flag in `app.py`)

The app targets **very few inference steps** (e.g., 4) for speed while keeping good image quality. It falls back to **CPU** automatically if CUDA isn’t available.

---

## Features

- ⚑ **Fast sampling** with **LCM** schedulers  
- πŸ” **Deterministic** results via seed  
- πŸ–₯️ **Auto GPU/CPU** selection (no brittle `nvidia-smi` checks)  
- πŸ” Optional **secret token** gate to prevent abuse  
- 🧩 Switch between **SDXL** and **SSD-1B+LCM LoRA** with a flag

---

## Requirements

Dependencies are pinned for compatibility (notably `diffusers==0.23.0` + `huggingface_hub==0.14.1`):

```txt
accelerate==0.24.1
diffusers==0.23.0
gradio==3.39.0
huggingface_hub==0.14.1
invisible-watermark==0.2.0
Pillow==10.1.0
torch==2.1.0
transformers==4.35.0
safetensors==0.4.0
numpy>=1.23
ipython