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: