cleanup reqs

gimmvfi/generalizable_INR/flowformer/core/FlowFormer/LatentCostFormer/decoder.py

@@ -1,22 +1,17 @@
-import loguru
 import torch
-import math
 import torch.nn as nn
 import torch.nn.functional as F
-from torch import einsum
 
-from einops.layers.torch import Rearrange
 from einops import rearrange
 
-from ...utils.utils import coords_grid, bilinear_sampler, upflow8
+from ...utils.utils import coords_grid, bilinear_sampler
 from .attention import (
     MultiHeadAttention,
     LinearPositionEmbeddingSine,
     ExpPositionEmbeddingSine,
 )
-from typing import Optional, Tuple
 
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.models.layers import DropPath
 
 from .gru import BasicUpdateBlock, GMAUpdateBlock
 from .gma import Attention

gimmvfi/generalizable_INR/flowformer/core/FlowFormer/LatentCostFormer/encoder.py

@@ -1,15 +1,11 @@
-import loguru
 import torch
-import math
 import torch.nn as nn
 import torch.nn.functional as F
 from torch import einsum
 import numpy as np
 
-from einops.layers.torch import Rearrange
 from einops import rearrange
-import sys
-from ...utils.utils import coords_grid, bilinear_sampler, upflow8
+from ...utils.utils import coords_grid
 from .attention import (
     BroadMultiHeadAttention,
     MultiHeadAttention,
@@ -17,14 +13,13 @@ from .attention import (
     ExpPositionEmbeddingSine,
 )
 from ..encoders import twins_svt_large
-from typing import Optional, Tuple
-from .twins import Size_, PosConv
-from .cnn import TwinsSelfAttentionLayer, TwinsCrossAttentionLayer, BasicEncoder
+from typing import Tuple
+from .twins import Size_
+from .cnn import BasicEncoder
 from .mlpmixer import MLPMixerLayer
 from .convnext import ConvNextLayer
-import time
 
-from timm.models.layers import Mlp, DropPath, to_2tuple, trunc_normal_
+from timm.models.layers import DropPath
 
 
 class PatchEmbed(nn.Module):

gimmvfi/generalizable_INR/flowformer/core/FlowFormer/LatentCostFormer/transformer.py

@@ -1,25 +1,8 @@
-import loguru
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
-from torch import einsum
 
-from einops.layers.torch import Rearrange
-from einops import rearrange
-
-from ...utils.utils import coords_grid, bilinear_sampler, upflow8
-from ..common import (
-    FeedForward,
-    pyramid_retrieve_tokens,
-    sampler,
-    sampler_gaussian_fix,
-    retrieve_tokens,
-    MultiHeadAttention,
-    MLP,
-)
-from ..encoders import twins_svt_large_context, twins_svt_large
-from ...position_encoding import PositionEncodingSine, LinearPositionEncoding
-from .twins import PosConv
+from ...utils.utils import coords_grid
+from ..encoders import twins_svt_large
 from .encoder import MemoryEncoder
 from .decoder import MemoryDecoder
 from .cnn import BasicEncoder

gimmvfi/generalizable_INR/flowformer/core/FlowFormer/common.py

@@ -3,13 +3,9 @@ import torch.nn as nn
 import torch.nn.functional as F
 from torch import einsum
 
-from einops.layers.torch import Rearrange
 from einops import rearrange
 
-from ..utils.utils import coords_grid, bilinear_sampler, indexing
-from loguru import logger
-
-import math
+from ..utils.utils import bilinear_sampler, indexing
 
 
 def nerf_encoding(x, L=6, NORMALIZE_FACOR=1 / 300):

gimmvfi/generalizable_INR/flowformer/core/position_encoding.py

@@ -1,4 +1,3 @@
-from loguru import logger
 import math
 import torch
 from torch import nn

gimmvfi/generalizable_INR/flowformer/core/utils/logger.py

@@ -1,5 +1,4 @@
 from torch.utils.tensorboard import SummaryWriter
-from loguru import logger as loguru_logger
 
 
 class Logger:
@@ -22,7 +21,7 @@ class Logger:
         metrics_str = ("{:10.4f}, " * len(metrics_data)).format(*metrics_data)
 
         # print the training status
-        loguru_logger.info(training_str + metrics_str)
+        print(training_str + metrics_str)
 
         if self.writer is None:
             if self.cfg.log_dir is None:

nodes.py

@@ -5,11 +5,9 @@ import folder_paths
 import yaml
 import comfy.model_management as mm
 from comfy.utils import ProgressBar, load_torch_file
-from PIL import Image
 
 from omegaconf import OmegaConf
 from tqdm import tqdm
-import numpy as np
 import cv2
 
 from .gimmvfi.generalizable_INR.gimmvfi_r import GIMMVFI_R

requirements.txt

@@ -5,5 +5,4 @@ cupy-cuda12x>=13.3.0
 timm
 omegaconf
 yacs
-loguru
 easydict