lev1 commited on
Commit
2e1b115
1 Parent(s): 62cb566

Hotfix / wrong comment

Browse files
Files changed (1) hide show
  1. model.py +4 -4
model.py CHANGED
@@ -225,16 +225,16 @@ class Model:
225
  self.pipe.controlnet.set_attn_processor(
226
  processor=self.controlnet_attn_proc)
227
 
228
- # added_prompt = 'best quality, extremely detailed'
229
- # negative_prompts = 'longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality'
230
 
231
  video, fps = utils.prepare_video(
232
  video_path, resolution, self.device, self.dtype, False)
233
  control = utils.pre_process_depth(
234
  video).to(self.device).to(self.dtype)
235
 
236
- depth_map_to_save = list(rearrange(control, 'f c w h -> f w h c').cpu().detach().numpy())
237
- _ = utils.create_video(depth_map_to_save, 4, path="ddxk.mp4", watermark=None)
238
 
239
  f, _, h, w = video.shape
240
  self.generator.manual_seed(seed)
 
225
  self.pipe.controlnet.set_attn_processor(
226
  processor=self.controlnet_attn_proc)
227
 
228
+ added_prompt = 'best quality, extremely detailed'
229
+ negative_prompts = 'longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality'
230
 
231
  video, fps = utils.prepare_video(
232
  video_path, resolution, self.device, self.dtype, False)
233
  control = utils.pre_process_depth(
234
  video).to(self.device).to(self.dtype)
235
 
236
+ # depth_map_to_save = list(rearrange(control, 'f c w h -> f w h c').cpu().detach().numpy())
237
+ # _ = utils.create_video(depth_map_to_save, 4, path="ddxk.mp4", watermark=None)
238
 
239
  f, _, h, w = video.shape
240
  self.generator.manual_seed(seed)