Clean imports and comments

2024-05-18 21:55:03 +00:00
parent 88839aa64f
commit 6052b00a10
1 changed files with 0 additions and 16 deletions
@@ -1,5 +1,3 @@
-from dataclasses import dataclass
-
 import torch
 from comfy import model_management

@@ -108,20 +106,6 @@ class ProPainterInpaint:
            device,
            image_config.process_size,
        )
-        # node_config = ProPainterConfig(
-        #     width,
-        #     height,
-        #     mask_dilates,
-        #     flow_mask_dilates,
-        #     ref_stride,
-        #     neighbor_length,
-        #     subvideo_length,
-        #     raft_iter,
-        #     fp16,
-        #     video_length,
-        #     input_size,
-        #     device,
-        # )

        frames_tensor, flow_masks_tensor, masks_dilated_tensor, original_frames = (
            prepare_frames_and_masks(frames, mask, image_config, device)