akrao9 commited on
Commit
65727ca
·
verified ·
1 Parent(s): 1440023

Upload 2 files

Browse files
Files changed (2) hide show
  1. README.md +49 -0
  2. 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