RageshAntony commited on
Commit
efe92e7
·
verified ·
1 Parent(s): 1eddd5f

tqdm false

Browse files
Files changed (1) hide show
  1. check_app.py +2 -2
check_app.py CHANGED
@@ -121,7 +121,7 @@ MODEL_CONFIGS = {
121
  }
122
  }
123
 
124
- def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_scale=3.5, seed=None,negative_prompt=None, randomize_seed=None, width=1024, height=1024, num_inference_steps=40, progress=gr.Progress(track_tqdm=True)):
125
  generator = None
126
  if randomize_seed:
127
  seed = random.randint(0, MAX_SEED)
@@ -177,7 +177,7 @@ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_sc
177
  @spaces.GPU(duration=170)
178
  def create_pipeline_logic(prompt_text, model_name, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=4.5, num_inference_steps=40,):
179
  print(f"starting {model_name}")
180
- progress = gr.Progress(track_tqdm=True)
181
  config = MODEL_CONFIGS[model_name]
182
  pipe_class = config["pipeline_class"]
183
  pipe = None
 
121
  }
122
  }
123
 
124
+ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_scale=3.5, seed=None,negative_prompt=None, randomize_seed=None, width=1024, height=1024, num_inference_steps=40, progress=gr.Progress(track_tqdm=False)):
125
  generator = None
126
  if randomize_seed:
127
  seed = random.randint(0, MAX_SEED)
 
177
  @spaces.GPU(duration=170)
178
  def create_pipeline_logic(prompt_text, model_name, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=4.5, num_inference_steps=40,):
179
  print(f"starting {model_name}")
180
+ progress = gr.Progress(track_tqdm=False)
181
  config = MODEL_CONFIGS[model_name]
182
  pipe_class = config["pipeline_class"]
183
  pipe = None