cp524 commited on
Commit
8bfeb05
·
1 Parent(s): 1117f02

Increase step duration in SMC grad inference calculation from 5.0 to 6.0

Browse files
Files changed (1) hide show
  1. src/smc/inference.py +1 -1
src/smc/inference.py CHANGED
@@ -165,7 +165,7 @@ def infer_smc_grad(config: SMCGradInferenceConfig, device='cpu'):
165
 
166
  def _get_smc_grad_duration(config: SMCGradInferenceConfig, pipe: Pipeline, device='cpu') -> int:
167
  setup_duration = 30.0
168
- step_duration = 5.0
169
  total_duration = math.ceil(setup_duration + step_duration * config.steps)
170
  return max(total_duration, MIN_GPU_DURATION)
171
 
 
165
 
166
  def _get_smc_grad_duration(config: SMCGradInferenceConfig, pipe: Pipeline, device='cpu') -> int:
167
  setup_duration = 30.0
168
+ step_duration = 6.0
169
  total_duration = math.ceil(setup_duration + step_duration * config.steps)
170
  return max(total_duration, MIN_GPU_DURATION)
171