Vijish commited on
Commit
18eee8e
Β·
verified Β·
1 Parent(s): e1f4f54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -271,7 +271,7 @@ def process_image_batch(images, pipe, prompt, negative_prompt, progress, batch_s
271
  prompt=prompt,
272
  negative_prompt=negative_prompt,
273
  ref_image=img,
274
- num_inference_steps=20
275
  ).images
276
  processed_batch.extend(result)
277
  else:
@@ -279,7 +279,7 @@ def process_image_batch(images, pipe, prompt, negative_prompt, progress, batch_s
279
  prompt=batch_prompt,
280
  negative_prompt=batch_negative_prompt,
281
  image=batch,
282
- num_inference_steps=20
283
  ).images
284
 
285
  all_processed_images.extend(processed_batch)
@@ -347,7 +347,7 @@ def generate_images_with_progress(prompt, negative_prompt, batch_count, use_cont
347
 
348
  images = []
349
  for i in tqdm(range(batch_count), desc="Generating images"):
350
- generated = pipe(prompt=[prompt], negative_prompt=[negative_prompt], num_inference_steps=20, width=1024, height=1024).images
351
  images.extend(generated)
352
  progress((i + 1) / batch_count) # Update progress bar
353
  clear_memory() # Clear memory after each image, even in single image mode
 
271
  prompt=prompt,
272
  negative_prompt=negative_prompt,
273
  ref_image=img,
274
+ num_inference_steps=30
275
  ).images
276
  processed_batch.extend(result)
277
  else:
 
279
  prompt=batch_prompt,
280
  negative_prompt=batch_negative_prompt,
281
  image=batch,
282
+ num_inference_steps=30
283
  ).images
284
 
285
  all_processed_images.extend(processed_batch)
 
347
 
348
  images = []
349
  for i in tqdm(range(batch_count), desc="Generating images"):
350
+ generated = pipe(prompt=[prompt], negative_prompt=[negative_prompt], num_inference_steps=30, width=1024, height=1024).images
351
  images.extend(generated)
352
  progress((i + 1) / batch_count) # Update progress bar
353
  clear_memory() # Clear memory after each image, even in single image mode