Bug fixes

2024-07-03 10:56:34 -06:00
parent bb57623a35
commit acb06d6ff3
6 changed files with 133 additions and 10 deletions
--- a/toolkit/sampler.py
+++ b/toolkit/sampler.py
@@ -123,6 +123,8 @@ def get_sampler(
            "num_train_timesteps": 1000,
            "shift": 3.0
        }
+    else:
+        raise ValueError(f"Sampler {sampler} not supported")


    config = copy.deepcopy(config_to_use)