improve warmup

wfx.py

@@ -98,7 +98,7 @@ class WFX():
     
     def warmup(self) -> None:
         warmed = 0
-        warmed_total = 20
+        warmed_total = 5
         
         warmup_kwargs = dict(
             prompt='a photo of a cat',
@@ -112,10 +112,17 @@ class WFX():
             logger.info(f'warming up T2I pipeline...')
             for _ in range(warmed_total):
                 begin = torch.cuda.Event(enable_timing=True)
-                self.T2IPipeline(**warmup_kwargs)
                 end = torch.cuda.Event(enable_timing=True)
+                
+                begin.record()
+                self.T2IPipeline(**warmup_kwargs)
+                end.record()
+                
+                torch.cuda.synchronize()
+                elapsed_time = begin.elapsed_time(end)
+                
                 warmed += 1
-                logger.info(f'warmed {warmed}/{warmed_total} - {begin.elapsed_time(end)}ms')
+                logger.info(f'warmed {warmed}/{warmed_total} - {elapsed_time:.2f}ms')