Update app_quant_latent.py

app_quant_latent.py

@@ -273,18 +273,18 @@ sigmas: list = None,
 **kwargs,
 ):
  if timesteps is not None and sigmas is not None:
- raise ValueError("Only one of timesteps or sigmas can be passed")
+  raise ValueError("Only one of timesteps or sigmas can be passed")
  if timesteps is not None:
- scheduler.set_timesteps(timesteps=timesteps, device=device, **kwargs)
- timesteps = scheduler.timesteps
- num_inference_steps = len(timesteps)
+  scheduler.set_timesteps(timesteps=timesteps, device=device, **kwargs)
+  timesteps = scheduler.timesteps
+  num_inference_steps = len(timesteps)
  elif sigmas is not None:
- scheduler.set_timesteps(sigmas=sigmas, device=device, **kwargs)
- timesteps = scheduler.timesteps
- num_inference_steps = len(timesteps)
+  scheduler.set_timesteps(sigmas=sigmas, device=device, **kwargs)
+  timesteps = scheduler.timesteps
+  num_inference_steps = len(timesteps)
  else:
- scheduler.set_timesteps(num_inference_steps, device=device, **kwargs)
- timesteps = scheduler.timesteps
+  scheduler.set_timesteps(num_inference_steps, device=device, **kwargs)
+  timesteps = scheduler.timesteps
  return timesteps, num_inference_steps
 
 @spaces.GPU