zwv9 commited on
Commit
fb27579
1 Parent(s): 9095ff2

fixed warmup

Browse files
Files changed (1) hide show
  1. wfx.py +5 -3
wfx.py CHANGED
@@ -110,9 +110,11 @@ class WFX():
110
 
111
  if warmed < warmed_total:
112
  logger.info(f'warming up T2I pipeline...')
113
- self.T2IPipeline(**warmup_kwargs)
114
- warmed += 1
115
- logger.info(f'T2I pipeline warmed in {warmed}/{warmed_total} iterations')
 
 
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__':