Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Re-added coordinate dowg with 0.3.1's release and fix

This commit is contained in:
SargeZT 2023-06-09 21:40:17 -05:00
parent e5b08f92b4
commit f9640d7c51
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
optimizer/optimizers.py
View File

@ -306,8 +306,8 @@ class EveryDreamOptimizer():
if "dowg" in optimizer_name: 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. # coordinate_dowg, scalar_dowg require no additional parameters. Epsilon is overrideable but is unnecessary in all stable diffusion training situations.
import dowg import dowg
#if optimizer_name == "coordinate_dowg": if optimizer_name == "coordinate_dowg":
#opt_class = dowg.CoordinateDoWG opt_class = dowg.CoordinateDoWG
if optimizer_name == "scalar_dowg": if optimizer_name == "scalar_dowg":
opt_class = dowg.ScalarDoWG opt_class = dowg.ScalarDoWG
else: else: