i'm a dumbass pass it on

2023-06-09 21:56:05 -05:00 · 2023-06-09 21:56:05 -05:00 · 2045bcc7c8
parent f9640d7c51
commit 2045bcc7c8
1 changed files with 1 additions and 1 deletions
--- a/optimizer/optimizers.py
+++ b/optimizer/optimizers.py
@ -308,7 +308,7 @@ class EveryDreamOptimizer():
                import dowg
                if optimizer_name == "coordinate_dowg":                    
                    opt_class = dowg.CoordinateDoWG
-                if optimizer_name == "scalar_dowg":
+                elif optimizer_name == "scalar_dowg":
                    opt_class = dowg.ScalarDoWG
                else:
                    raise ValueError(f"Unknown DoWG optimizer {optimizer_name}. Available options are coordinate_dowg and scalar_dowg")