Spaces:
Sleeping
Sleeping
Increase step duration in SMC grad inference calculation from 5.0 to 6.0
Browse files- 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 =
|
| 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 |
|