storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-09 00:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixes (#178)

Browse Source
This commit is contained in:
Sayak Paul
2025-01-04 08:57:41 +05:30
committed by GitHub
parent 071c140879
commit a709a1cec1
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/trainer.py
+6 -2
View File
@@ -646,11 +646,15 @@ class Trainer:
scheduler_sigmas = get_scheduler_sigmas(self.scheduler)
scheduler_sigmas = (
scheduler_sigmas.to(device=accelerator.device, dtype=torch.float32) if scheduler_sigmas else None
scheduler_sigmas.to(device=accelerator.device, dtype=torch.float32)
if scheduler_sigmas is not None
else None
)
scheduler_alphas = get_scheduler_alphas(self.scheduler)
scheduler_alphas = (
scheduler_alphas.to(device=accelerator.device, dtype=torch.float32) if scheduler_alphas else None
scheduler_alphas.to(device=accelerator.device, dtype=torch.float32)
if scheduler_alphas is not None
else None
)
for epoch in range(first_epoch, self.state.train_epochs):