Spaces:
Runtime error
Runtime error
Hotfix / wrong comment
Browse files
model.py
CHANGED
@@ -225,16 +225,16 @@ class Model:
|
|
225 |
self.pipe.controlnet.set_attn_processor(
|
226 |
processor=self.controlnet_attn_proc)
|
227 |
|
228 |
-
|
229 |
-
|
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)
|