2510

README.md

@@ -16,7 +16,7 @@ At AiArtLab, we strive to create a free, compact and fast model that can be trai
 - VAE: 8x16ch, [Simple VAE](https://huggingface.co/AiArtLab/simplevae)
 
 
-The model was trained for ~3 months on 4xRTX 5090 GPUs using over 1 million images of various resolutions and styles, primarily anime and illustrations.
+The model was trained for ~3 months on (2-4)x RTX 5090 GPUs using over 1+ million images of various resolutions and styles, primarily anime and illustrations.
 
 [Gradio](https://sdxs.aiartlab.org/)
 
@@ -73,4 +73,14 @@ BTC: 3JHv9Hb8kEW8zMAccdgCdZGfrHeMhH1rpN
 
 ## More examples
 
-![result_grid](media/result_grid.jpg)
+![result_grid](media/result_grid.jpg)
+
+## Citation
+```bibtex
+@misc{recoilme_sdxs_2025,
+  title={Simple Diffusion XS},
+  author={recoilme with help of AiArtLab Team},
+  url={https://huggingface.co/AiArtLab/sdxs},
+  year={2025}
+}
+```

requirements.txt

@@ -5,4 +5,5 @@ matplotlib>=3.10.1
 wandb>=0.19.8
 huggingface_hub>=0.29.3
 bitsandbytes>=0.45.4
-transformers
+transformers
+hf_transfer

src/TRAIN.md

@@ -14,7 +14,7 @@ git config --global credential.helper store
 
 ```
 python -m pip install --upgrade pip
-pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 -U
+pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130 -U # change
 pip install flash-attn --no-build-isolation # optional
 ```
 Клонируйте репозиторий:
@@ -22,7 +22,7 @@ pip install flash-attn --no-build-isolation # optional
 ```
 git clone https://huggingface.co/AiArtLab/sdxs
 cd sdxs/
-pip install -r requirements.txt
+pip install -r requirements.txt -U
 ```
 Подготовьте датасет:
 

train.py

@@ -33,7 +33,7 @@ base_learning_rate = 5e-5
 min_learning_rate = 1e-5
 num_epochs = 40
 # samples/save per epoch
-sample_interval_share = 3
+sample_interval_share = 2
 use_wandb = True
 use_comet_ml = False
 save_model = True