Spaces:
Running on Zero
Running on Zero
Upload 2 files
Browse files- README.md +49 -0
- requirements.txt +9 -0
README.md
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
title: Boomer FLA T2I
|
| 3 |
+
emoji: 🎨
|
| 4 |
+
colorFrom: indigo
|
| 5 |
+
colorTo: purple
|
| 6 |
+
sdk: gradio
|
| 7 |
+
sdk_version: 6.15.2
|
| 8 |
+
app_file: app.py
|
| 9 |
+
pinned: false
|
| 10 |
+
license: other
|
| 11 |
+
short_description: 1024px text-to-image with Boomer FLA on ZeroGPU
|
| 12 |
+
models:
|
| 13 |
+
- akrao9/Boomer-T2I
|
| 14 |
+
- google/gemma-4-E2B-it
|
| 15 |
+
- mit-han-lab/dc-ae-f32c32-sana-1.1-diffusers
|
| 16 |
+
preload_from_hub:
|
| 17 |
+
- akrao9/Boomer-T2I
|
| 18 |
+
---
|
| 19 |
+
|
| 20 |
+
# Boomer FLA — Text to Image Demo
|
| 21 |
+
|
| 22 |
+
Interactive demo for [akrao9/Boomer-T2I](https://huggingface.co/akrao9/Boomer-T2I): a 1024×1024 text-to-image flow-matching model with Flash Linear Attention.
|
| 23 |
+
|
| 24 |
+
## Space setup
|
| 25 |
+
|
| 26 |
+
1. Create a new **Gradio** Space on Hugging Face.
|
| 27 |
+
2. Upload the contents of this `Space/` folder (or point the Space repo here).
|
| 28 |
+
3. In **Settings → Hardware**, select **ZeroGPU** (requires HF PRO for hosting). This allocates a shared **NVIDIA RTX Pro 6000 Blackwell** GPU (48 GB via `@spaces.GPU(size="large")`).
|
| 29 |
+
4. Add a Space secret **`HF_TOKEN`** with a token that has accepted the [Gemma Terms of Use](https://ai.google.dev/gemma/terms).
|
| 30 |
+
5. Restart the Space after the first build (model + VAE + text encoder download can take several minutes).
|
| 31 |
+
|
| 32 |
+
## Local test
|
| 33 |
+
|
| 34 |
+
```bash
|
| 35 |
+
cd Space
|
| 36 |
+
pip install -r requirements.txt
|
| 37 |
+
export HF_TOKEN=hf_...
|
| 38 |
+
python app.py
|
| 39 |
+
```
|
| 40 |
+
|
| 41 |
+
The `@spaces.GPU` decorator is a no-op locally; use a CUDA machine for full inference.
|
| 42 |
+
|
| 43 |
+
Pinned deps for ZeroGPU: do **not** pin `torch` or `torchvision` in `requirements.txt` — the Space builder injects `torch<=2.11.0` automatically. Other pins: diffusers 0.38.0, transformers 5.9.0, flash-linear-attention 0.5.0.
|
| 44 |
+
|
| 45 |
+
## Notes
|
| 46 |
+
|
| 47 |
+
- Uses `@spaces.GPU(size="large", duration=150)` — RTX Pro 6000 Blackwell 48 GB tier — for 32-step STORK-2 at 1024px.
|
| 48 |
+
- Example prompts run generation on click (`run_on_click=True`).
|
| 49 |
+
- VAE and Gemma 4 2B are fetched from their upstream repos on first startup.
|
requirements.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
gradio==6.15.2
|
| 2 |
+
diffusers==0.38.0
|
| 3 |
+
transformers==5.9.0
|
| 4 |
+
accelerate==1.13.0
|
| 5 |
+
safetensors==0.8.0rc0
|
| 6 |
+
pillow>=11.3.0
|
| 7 |
+
huggingface-hub==1.16.1
|
| 8 |
+
einops>=0.8.2
|
| 9 |
+
flash-linear-attention==0.5.0
|