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
2601
Browse files- samples/unet_320x640_0.jpg +2 -2
- samples/unet_352x640_0.jpg +2 -2
- samples/unet_384x640_0.jpg +2 -2
- samples/unet_416x640_0.jpg +2 -2
- samples/unet_448x640_0.jpg +2 -2
- samples/unet_480x640_0.jpg +2 -2
- samples/unet_512x640_0.jpg +2 -2
- samples/unet_544x640_0.jpg +2 -2
- samples/unet_576x640_0.jpg +2 -2
- samples/unet_608x640_0.jpg +2 -2
- samples/unet_640x320_0.jpg +2 -2
- samples/unet_640x352_0.jpg +2 -2
- samples/unet_640x384_0.jpg +2 -2
- samples/unet_640x416_0.jpg +2 -2
- samples/unet_640x448_0.jpg +2 -2
- samples/unet_640x480_0.jpg +2 -2
- samples/unet_640x512_0.jpg +2 -2
- samples/unet_640x544_0.jpg +2 -2
- samples/unet_640x576_0.jpg +2 -2
- samples/unet_640x608_0.jpg +2 -2
- samples/unet_640x640_0.jpg +2 -2
- test.ipynb +2 -2
- train.py +3 -3
- unet/diffusion_pytorch_model.safetensors +1 -1
samples/unet_320x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_352x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_384x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_416x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_448x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_480x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_512x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_544x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_608x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x352_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x416_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x448_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x480_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x512_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x544_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x608_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_640x640_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
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:bf0d89eba46a54b7cc624cf3b0f5df8ea0320aa2a8f4a5b3194a48205c10a29f
|
| 3 |
+
size 5161876
|
train.py
CHANGED
|
@@ -29,7 +29,7 @@ from transformers import AutoTokenizer, AutoModel
|
|
| 29 |
# --------------------------- Параметры ---------------------------
|
| 30 |
ds_path = "/workspace/sdxs-08b/datasets/d123_640_sd15"
|
| 31 |
project = "unet"
|
| 32 |
-
batch_size =
|
| 33 |
base_learning_rate = 4e-6
|
| 34 |
min_learning_rate = 2e-6
|
| 35 |
num_epochs = 20
|
|
@@ -37,8 +37,8 @@ sample_interval_share = 5
|
|
| 37 |
cfg_dropout = 0.10
|
| 38 |
max_length = 248
|
| 39 |
use_wandb = False
|
| 40 |
-
use_comet_ml =
|
| 41 |
-
save_model =
|
| 42 |
use_decay = True
|
| 43 |
fbp = False
|
| 44 |
optimizer_type = "adam8bit"
|
|
|
|
| 29 |
# --------------------------- Параметры ---------------------------
|
| 30 |
ds_path = "/workspace/sdxs-08b/datasets/d123_640_sd15"
|
| 31 |
project = "unet"
|
| 32 |
+
batch_size = 80
|
| 33 |
base_learning_rate = 4e-6
|
| 34 |
min_learning_rate = 2e-6
|
| 35 |
num_epochs = 20
|
|
|
|
| 37 |
cfg_dropout = 0.10
|
| 38 |
max_length = 248
|
| 39 |
use_wandb = False
|
| 40 |
+
use_comet_ml = True
|
| 41 |
+
save_model = True
|
| 42 |
use_decay = True
|
| 43 |
fbp = False
|
| 44 |
optimizer_type = "adam8bit"
|
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 3438444088
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8c184b3f80277f9a353bbf400486f25bb33b980e9f59c118b72205c5adcf280f
|
| 3 |
size 3438444088
|