Merge pull request #25 from secutron/hotfix

ControlNet · web-flow · commit 398060f011f6 · 2025-03-27T20:01:04.000+11:00
hotfix: add lightning dependancy for _cosine_scheduler_fn(..)
diff --git a/model/marlin.py b/model/marlin.py
@@ -307,6 +307,7 @@ def _cosine_scheduler_factors(self):
         return values_factors
 
     def _cosine_scheduler_fn(self, epoch):
+        if epoch > 0: epoch -= 1  # lightning dependancy
         return self.lr_scheduler_factors[epoch]
 
     def configure_optimizers(self):
