From 6052b00a10c33651f7bc2a661471320a6b07855b Mon Sep 17 00:00:00 2001 From: daniabib Date: Sat, 18 May 2024 21:55:03 +0000 Subject: [PATCH] Clean imports and comments --- propainter_nodes.py | 16 ---------------- 1 file changed, 16 deletions(-) diff --git a/propainter_nodes.py b/propainter_nodes.py index 44002ba..f38747e 100644 --- a/propainter_nodes.py +++ b/propainter_nodes.py @@ -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)