Update README.md

README.md

@@ -13,27 +13,37 @@ tags:
 - fine-tuned
 - preview
 - 6b
+- s3-dit
 base_model: Tongyi-MAI/Z-Image
-model-index:
-- name: Helio-Image-Preview
-  results: []
 ---
 
-<h1 align="center">🌞 Helio-Image-Preview — 6B<br><sub><sup>Русскоязычная модель генерации изображений на ~6 млрд параметров — предварительная версия</sup></sub></h1>
-
 <div align="center">
+  <h1>🌞 Helio-Image-Preview</h1>
+  <h3>Русскоязычная модель генерации изображений — 6B параметров</h3>
+  <p><i>Предварительная версия. Архитектура S3-DiT. Кириллица в изображениях.</i></p>
+  <br>
+  <p>
+    <img src="https://img.shields.io/badge/Parameters-6B-green" alt="params"/>
+    <img src="https://img.shields.io/badge/Architecture-S3--DiT-blue" alt="arch"/>
+    <img src="https://img.shields.io/badge/Resolution-1024×1024-purple" alt="res"/>
+    <img src="https://img.shields.io/badge/Language-RU%20%7C%20EN-orange" alt="lang"/>
+    <img src="https://img.shields.io/badge/License-Apache%202.0-lightgrey" alt="license"/>
+  </p>
+</div>
+
+---
 
+<div align="center">
 <img src="https://huggingface.co/HelioAI/Helio-Image-Preview/resolve/main/teaser.png" alt="Helio-Image-Preview Examples" width="800">
+</div>
 
 **Helio-Image-Preview** — первый публичный выпуск семейства Helio-Image.
-Модель содержит **~6 млрд (6B) параметров** и построена на архитектуре S3-DiT.
+Модель содержит **~6 млрд (6B) параметров**, построена на архитектуре S3-DiT.
 Это предварительная версия, демонстрирующая возможности архитектуры перед полноценным релизом.
 
-</div>
-
 ---
 
-## 📋 Краткие характеристики
+## 📋 Характеристики
 
 | | |
 |---|---|
@@ -51,8 +61,6 @@ model-index:
 
 **Helio-Image** — семейство моделей генерации изображений на базе архитектуры S3-DiT (**~6B параметров**), ориентированное на качественную работу с русским языком: генерация надписей на кириллице, понимание русскоязычных описаний сцен и стилей.
 
-**Preview-версия** — это ранний доступ к модели, который показывает текущее направление разработки и уже сейчас демонстрирует впечатляющие результаты на широком спектре задач.
-
 ### 🔮 Что впереди
 
 Полноценная модель **Helio-Image** находится в активной разработке и будет обучена на **500–800K изображений** с расширенным покрытием стилей, жанров и сценариев. Preview-версия обучена на ~110K изображений за одну эпоху и уже покрывает большинство популярных направлений генерации — от фотореализма до аниме, от игровых сцен до киберпанка.
@@ -84,7 +92,7 @@ model-index:
 
 ### Требования к оборудованию
 
-> Модель содержит **~6 млрд параметров**. В формате `bf16` она занимает около **~12 ГБ VRAM** только для весов трансформера (плюс VAE, текстовый энкодер и рабочая память). Рекомендуется GPU с ≥ 24 ГБ VRAM (RTX 3090 / 4090 / A5000 и выше).
+> Модель содержит **~6 млрд параметров**. В формате `bf16` — около **~12 ГБ VRAM** только для весов трансформера (плюс VAE, текстовый энкодер и рабочая память). Рекомендуется GPU с ≥ 24 ГБ VRAM (RTX 3090 / 4090 / A5000 и выше).
 
 ### Установка
 
@@ -101,7 +109,7 @@ from diffusers import ZImagePipeline
 
 # Загрузка модели (~6B параметров)
 pipe = ZImagePipeline.from_pretrained(
-    "HelioAI/Helio-Image-Preview",  # или локальный путь
+    "HelioAI/Helio-Image-Preview",
     torch_dtype=torch.bfloat16,
 )
 pipe.to("cuda")
@@ -149,7 +157,7 @@ Star Wars штурмовики маршируют с надписью ИМПЕР
 | Параметр | Значение |
 |---|---|
 | **Архитектура** | **S3-DiT (~6B параметров)** |
-| Базовая модель | [Z-Image](https://huggingface.co/Tongyi-MAI/Z-Image) |
+| **Базовая модель** | [Z-Image](https://huggingface.co/Tongyi-MAI/Z-Image) |
 | Датасет | ~110K изображений 1024×1024 |
 | GPU | 4× RTX PRO 6000 96GB |
 | DeepSpeed | ZeRO-2 |
@@ -186,7 +194,7 @@ Apache 2.0. Базовая архитектура: [Z-Image](https://huggingface
 ---
 
 <div align="center">
-
-*Helio-Image-Preview (6B) — это только начало. Следите за обновлениями!* ☀️
-
-</div>
+  <h3>☀️ Built by Helios</h3>
+  <p>4× RTX PRO 6000 • 110K images • S3-DiT 6B • Preview</p>
+  <p><b><i>Helio-Image-Preview — это только начало. Следите за обновлениями!</i></b></p>
+</div>