From f9640d7c5166f07178a6640b8a5086a690ad27e3 Mon Sep 17 00:00:00 2001 From: SargeZT Date: Fri, 9 Jun 2023 21:40:17 -0500 Subject: [PATCH] Re-added coordinate dowg with 0.3.1's release and fix --- optimizer/optimizers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optimizer/optimizers.py b/optimizer/optimizers.py index 269c3ff..94d00b4 100644 --- a/optimizer/optimizers.py +++ b/optimizer/optimizers.py @@ -306,8 +306,8 @@ class EveryDreamOptimizer(): if "dowg" in optimizer_name: # coordinate_dowg, scalar_dowg require no additional parameters. Epsilon is overrideable but is unnecessary in all stable diffusion training situations. import dowg - #if optimizer_name == "coordinate_dowg": - #opt_class = dowg.CoordinateDoWG + if optimizer_name == "coordinate_dowg": + opt_class = dowg.CoordinateDoWG if optimizer_name == "scalar_dowg": opt_class = dowg.ScalarDoWG else: