Simple Diffusion XS (sdxs-2b alpha version)

XS Size, Excess Quality

Train status: 4xRTX5090 / we need more gold / support us please..

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)

Samples with seed 0 and good prompts

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 = "worst quality, low quality, loli, low details, blurry, jpeg artifacts, unfinished, sketch, sepia, missing limb, text, bad anatomy, bad proportions, bad hands, missing fingers"

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: 0$

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},
  url={https://huggingface.co/AiArtLab/sdxs-2b},
  year={2026}
}