Re-added coordinate dowg with 0.3.1's release and fix
This commit is contained in:
parent
e5b08f92b4
commit
f9640d7c51
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue