Ahsen Khaliq commited on
Commit
1fb1dee
·
1 Parent(s): c6b75cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -85,7 +85,7 @@ model_config.update({
85
  'class_cond': False,
86
  'diffusion_steps': 1000,
87
  'rescale_timesteps': True,
88
- 'timestep_respacing': '300', # Modify this value to decrease the number of
89
  # timesteps.
90
  'image_size': 256,
91
  'learn_sigma': True,
@@ -208,7 +208,7 @@ def inference(text, init_image, skip_timesteps):
208
  all_frames.append(img)
209
  tqdm.write(f'Batch {i}, step {j}, output {k}:')
210
  #display.display(display.Image(filename))
211
- writer = imageio.get_writer('video.mp4', fps=20)
212
  for im in all_frames:
213
  writer.append_data(np.array(im))
214
  writer.close()
 
85
  'class_cond': False,
86
  'diffusion_steps': 1000,
87
  'rescale_timesteps': True,
88
+ 'timestep_respacing': '90', # Modify this value to decrease the number of
89
  # timesteps.
90
  'image_size': 256,
91
  'learn_sigma': True,
 
208
  all_frames.append(img)
209
  tqdm.write(f'Batch {i}, step {j}, output {k}:')
210
  #display.display(display.Image(filename))
211
+ writer = imageio.get_writer('video.mp4', fps=5)
212
  for im in all_frames:
213
  writer.append_data(np.array(im))
214
  writer.close()