Add more stats to benchmark

hydra_node/models.py

@@ -282,12 +282,10 @@ class StableDiffusionModel(nn.Module):
                     self.sample(settings)
                     self.sample(settings)
                     dur = time.perf_counter() - now
-                    results.append((w, h, n, f"{dur/2.0:.4f}s"))
-                break
-            break
+                    results.append((w, h, n, dur/2.0))
         print("results")
         for w,h,n,dur in results:
-            print(f"{w}x{h} with {n} images ({settings.steps} steps {settings.sampler}) took: {dur}")
+            print(f"{w}x{h} with {n} images ({settings.steps} steps {settings.sampler}) took: {dur:.4f}s ({1./(dur/settings.steps):.4f}it/s, {(dur/settings.steps)*28:.4f}s for 28 steps)")
 
     @property
     def get_default_config(self):