fixed warmup
Browse files
wfx.py
CHANGED
@@ -110,9 +110,11 @@ class WFX():
|
|
110 |
|
111 |
if warmed < warmed_total:
|
112 |
logger.info(f'warming up T2I pipeline...')
|
113 |
-
|
114 |
-
|
115 |
-
|
|
|
|
|
116 |
|
117 |
|
118 |
if __name__ == '__main__':
|
|
|
110 |
|
111 |
if warmed < warmed_total:
|
112 |
logger.info(f'warming up T2I pipeline...')
|
113 |
+
for _ in range(warmed_total):
|
114 |
+
self.T2IPipeline(**warmup_kwargs)
|
115 |
+
warmed += 1
|
116 |
+
logger.info(f'warmed {warmed}/{warmed_total}')
|
117 |
+
|
118 |
|
119 |
|
120 |
if __name__ == '__main__':
|