From a709a1cec15c241435ee6d94b5e81b07a0e609d6 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Sat, 4 Jan 2025 08:57:41 +0530 Subject: [PATCH] fixes (#178) --- finetrainers/trainer.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/finetrainers/trainer.py b/finetrainers/trainer.py index cf114b2..5b853a3 100644 --- a/finetrainers/trainer.py +++ b/finetrainers/trainer.py @@ -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):