Simple Diffusion XS (sdxs-2b alpha version)
XS Size, Excess Quality
Train status: thanks to theartisan.ai for provided GPU cluster!
At AiArtLab, we strive to create a free, compact and fast model that can be trained on consumer graphics cards.
- CosmosTransformer3D: 2b parameters
- Qwen3.5: 0.8b parameters
- Qwen vae: 16ch8x
- Resolution: Default 768x1152, trained from 576 to 1152 with step 64
- Limitations: trained on small datasets ~2.5kk, focused on art / illustrations / anime and photo
- Captions: danbooru, natural (short/medium), trained with 250 max toks
Key points
- Apr 2026: Started research of Cosmos Transformer
- May 2026: 1st preview (Initial model weights)
- May 2026: finetuned Qwen3.5-0.8B
- May 2026: hybrid EDM/Karras scheduler
- May 2026: img2img pipeline and txt2video pipeline (not trained)
- May 2026: telegram bot
- Jun 2026: comfyui pipeline
- Jun 2026: 2nd preview
Random samples
Text 2 image
#!pip install -U torch torchvision
#!pip install -U diffusers accelerate transformers
import torch
from diffusers import DiffusionPipeline
device = "cuda" if torch.cuda.is_available() else "cpu"
dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
pipe_id = "AiArtLab/sdxs-2b"
pipe = DiffusionPipeline.from_pretrained(
pipe_id,
torch_dtype=dtype,
trust_remote_code=True
).to(device)
refined = "A blonde-haired Red Eyes girl with a hair ribbon, half-updo, and tsurime stands solo in a flower field holding a bouquet with a serene smile, wearing green overalls, a white shirt, rolled-up sleeves, and a straw hat with a flower while looking at the viewer under volumetric and natural lighting with a Dutch angle."
negative_prompt = "low quality, bad quality, blurry, sketch, sepia, text, bad anatomy, bad proportions, bad hands, missing fingers, child drawing"
output = pipe(
prompt=refined,
negative_prompt=negative_prompt,
)
image = output.images[0]
image.show()
Prompt refiner
refined = pipe.refine_prompts("1girl")
print(refined[0])
Donations
Donated: 25$
Thanks for your support!
- Euro / dollars: sorry, we can’t accept payments via Patreon or Ko-fi. Please register on vast.ai (this is the GPU provider we use for training the model), top up any amount, and transfer: Transfer Money / User / Email: vadim-kulibaba@yandex.ru.
- RUB: donate in rub
- DOGE: DEw2DR8C7BnF8GgcrfTzUjSnGkuMeJhg83
- BTC: 3JHv9Hb8kEW8zMAccdgCdZGfrHeMhH1rpN
- USDT
- Ethereum / Polygon / BNB SmartChain: 0xD4388B6698dFaE1460E72099D4F208aaCA4f6E6C
- Tron: TD7ey4h9igPGdcrcBcnZaz56R5tNgRZNvV
- Solana: MMYFJeYEtYHrSNFHChytJDHbEDniXrnAxPNLhJ1LbkB
Contacts
Please contact with us if you may provide some GPU's or money on training
- telegram recoilme *prefered way
- mail at aiartlab.org (slow response)
Citation
@misc{sdxs,
title={Simple Diffusion XS-2b},
author={recoilme and AiArtLab Team},
title={Simple Diffusion XS-2b},
author={recoilme and AiArtLab Team},
url={https://huggingface.co/AiArtLab/sdxs-2b},
year={2026}