update: model configs

wandb · Sep 4, 2024 · d6a4e27 · d6a4e27
1 parent 46b1f92
commit d6a4e27
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hemm/eval_pipelines/model.py b/hemm/eval_pipelines/model.py
@@ -45,6 +45,7 @@ def __init__(
             configs=configs,
             pipeline_configs=pipeline_configs,
         )
+        self.configs["torch_dtype"] = str(self._torch_dtype)
         pipeline_init_kwargs = {
             "pretrained_model_name_or_path": self.diffusion_model_name_or_path,
             "torch_dtype": self._torch_dtype,