Instructions to use AiArtLab/sdxs-1b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use AiArtLab/sdxs-1b with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("AiArtLab/sdxs-1b", dtype=torch.bfloat16, device_map="cuda") prompt = "sdxs-1b" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
2604
Browse files- README.md +4 -1
- media/gandr-collage.jpg +3 -0
- media/result_grid.jpg +2 -2
- src/cuda.ipynb +2 -2
- test.ipynb +2 -2
- unet/diffusion_pytorch_model.safetensors +2 -2
README.md
CHANGED
|
@@ -81,9 +81,12 @@ image.show()
|
|
| 81 |
```
|
| 82 |
|
| 83 |
### Diversiry
|
| 84 |
-
Prompt: *cat*
|
| 85 |

|
| 86 |
|
|
|
|
|
|
|
|
|
|
| 87 |
### VAE
|
| 88 |
|
| 89 |
The VAE in Simple Diffusion utilizes an asymmetric VAE architecture featuring an 8x encoder and a 16x decoder. While a compression factor of 8 is maintained during training, the resolution is effectively doubled during inference through an additional upscaling block. This strategy reduces training costs by an order of magnitude and boosts inference speed without perceptual quality loss. Effectively, this acts as an integrated latent upscaler. To ensure a fair comparison with other VAEs, we downsampled the generated images to match the input resolution for metric evaluation. The SDXS VAE was not trained from scratch but was initialized from weights of FLUX 2 VAE, then redisigned and retrained. We also trained [16 ch vae](https://huggingface.co/AiArtLab/simplevae) with flux.1 quality based on aura vae.
|
|
|
|
| 81 |
```
|
| 82 |
|
| 83 |
### Diversiry
|
| 84 |
+
Prompt: *cat*
|
| 85 |

|
| 86 |
|
| 87 |
+
Random prompts
|
| 88 |
+

|
| 89 |
+
|
| 90 |
### VAE
|
| 91 |
|
| 92 |
The VAE in Simple Diffusion utilizes an asymmetric VAE architecture featuring an 8x encoder and a 16x decoder. While a compression factor of 8 is maintained during training, the resolution is effectively doubled during inference through an additional upscaling block. This strategy reduces training costs by an order of magnitude and boosts inference speed without perceptual quality loss. Effectively, this acts as an integrated latent upscaler. To ensure a fair comparison with other VAEs, we downsampled the generated images to match the input resolution for metric evaluation. The SDXS VAE was not trained from scratch but was initialized from weights of FLUX 2 VAE, then redisigned and retrained. We also trained [16 ch vae](https://huggingface.co/AiArtLab/simplevae) with flux.1 quality based on aura vae.
|
media/gandr-collage.jpg
ADDED
|
Git LFS Details
|
media/result_grid.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
src/cuda.ipynb
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1c49083d3defe77873103e3c528b087b93bcf202e742f699fbbceb81f4eefe51
|
| 3 |
+
size 2530
|
test.ipynb
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e2ad52116dadf67a00cc6303514409f17387875c4925d2a4b83625b23729b71d
|
| 3 |
+
size 13732127
|
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b4dc30ad1a2c8a6b060849ace4aa33d94889d7436d92892d2f21515ef00fe7d5
|
| 3 |
+
size 3210307232
|