lev1 commited on
Commit
296e148
1 Parent(s): a4632b0

Limit to input video length

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -122,7 +122,7 @@ class Model:
122
  added_prompt = 'best quality, extremely detailed'
123
  negative_prompts = 'longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality'
124
 
125
- video, fps = utils.prepare_video(video_path, resolution, self.device, self.dtype, False)
126
  control = utils.pre_process_canny(video, low_threshold, high_threshold).to(self.device).to(self.dtype)
127
  f, _, h, w = video.shape
128
  self.generator.manual_seed(seed)
 
122
  added_prompt = 'best quality, extremely detailed'
123
  negative_prompts = 'longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality'
124
 
125
+ video, fps = utils.prepare_video(video_path, resolution, self.device, self.dtype, False, start_t=0, end_t=15)
126
  control = utils.pre_process_canny(video, low_threshold, high_threshold).to(self.device).to(self.dtype)
127
  f, _, h, w = video.shape
128
  self.generator.manual_seed(seed)