576ep1

requirements.txt

@@ -1,10 +1,10 @@
-torch
-torchvision
-torchaudio
-diffusers
-accelerate
-datasets
-matplotlib
-wandb
-huggingface_hub
-bitsandbytes
+torch>=2.6.0
+torchvision>=0.21.0
+torchaudio>=2.6.0
+diffusers>=0.32.2
+accelerate>=1.5.2
+datasets>=3.5.0
+matplotlib>=3.10.1
+wandb>=0.19.8
+huggingface_hub>=0.29.3
+bitsandbytes>=0.45.4

sdxs/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0e5369b658504a0213fdf56f252426addf3311a5080f383aba4132a23a1f251b
+oid sha256:2d2357462fe46ee246701efbb8bcabbd7bf77530d5f2b2c96d7f4e3ab9fd91ab
 size 4529095968

train.py

@@ -510,9 +510,8 @@ for epoch in range(start_epoch, start_epoch + num_epochs):
             if not lowram:
                 optimizer.step()
                 lr_scheduler.step()
-
-            # Используем ограничение нормы градиентов через Accelerator
-            optimizer.zero_grad(set_to_none=True)
+                # Используем ограничение нормы градиентов через Accelerator
+                optimizer.zero_grad(set_to_none=True)
             
             # Увеличиваем счетчик глобальных шагов
             global_step += 1