Zhouyan248 commited on
Commit
701a4f7
1 Parent(s): 1ee756b

Update image_to_video/__init__.py

Browse files
Files changed (1) hide show
  1. image_to_video/__init__.py +1 -1
image_to_video/__init__.py CHANGED
@@ -209,7 +209,7 @@ def get_input(path,args):
209
  # raise ValueError('Need to give a video or some images')
210
  print('given video is None, using text to video')
211
  video_frames = torch.zeros(16,3,args.latent_h,args.latent_w,dtype=torch.uint8)
212
- args.mask_type = 'all'
213
  video_frames = transform_video(video_frames)
214
  n = 0
215
  return video_frames, n
 
209
  # raise ValueError('Need to give a video or some images')
210
  print('given video is None, using text to video')
211
  video_frames = torch.zeros(16,3,args.latent_h,args.latent_w,dtype=torch.uint8)
212
+ args.mask_type = 'first1'
213
  video_frames = transform_video(video_frames)
214
  n = 0
215
  return video_frames, n