Run benchmark if configured

hydra_node/models.py

@@ -257,6 +257,9 @@ class StableDiffusionModel(nn.Module):
         if config.prior_path:
             self.prior = VectorAdjustPrior.load_model(config.prior_path).to(self.device)
         self.copied_ema = False
+        
+        if config.benchmark == "1":
+            self.benchmark()
 
     def benchmark(self):
         settings = self.get_default_config