                                                                                                                                                                                                   
  num_inference_steps = 20                                                                                                                                                                                                                    

  return {"p