rynmurdock commited on
Commit
66cd66a
·
verified ·
1 Parent(s): 48b46a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -55,7 +55,7 @@ from transformers import CLIPVisionModelWithProjection
55
  import uuid
56
  import av
57
 
58
- def write_video(file_name, images, fps=17):
59
  print('Saving')
60
  container = av.open(file_name, mode="w")
61
 
@@ -79,6 +79,14 @@ def write_video(file_name, images, fps=17):
79
  container.close()
80
  print('Saved')
81
 
 
 
 
 
 
 
 
 
82
 
83
  image_encoder = CLIPVisionModelWithProjection.from_pretrained("h94/IP-Adapter", subfolder="sdxl_models/image_encoder", torch_dtype=dtype).to(DEVICE)
84
  #vae = AutoencoderTiny.from_pretrained("madebyollin/taesd", torch_dtype=dtype)
 
55
  import uuid
56
  import av
57
 
58
+ def av_write_video(file_name, images, fps=17):
59
  print('Saving')
60
  container = av.open(file_name, mode="w")
61
 
 
79
  container.close()
80
  print('Saved')
81
 
82
+ def write_video(file_name, images, fps=15):
83
+ print('Saving')
84
+ writer = imageio.get_writer(file_name, fps=fps)
85
+ for im in images:
86
+ writer.append_data(np.array(im))
87
+ writer.close()
88
+ print('Saved')
89
+
90
 
91
  image_encoder = CLIPVisionModelWithProjection.from_pretrained("h94/IP-Adapter", subfolder="sdxl_models/image_encoder", torch_dtype=dtype).to(DEVICE)
92
  #vae = AutoencoderTiny.from_pretrained("madebyollin/taesd", torch_dtype=dtype)