dependency fix

8 months ago · ab71a512de
parent 7d28a70226
commit ab71a512de
1 changed files with 0 additions and 1 deletions
--- a/src/models/rclut.py
+++ b/src/models/rclut.py
@ -5,7 +5,6 @@ import numpy as np
 from common.utils import round_func
 from common.lut import  forward_rc_conv_centered, forward_rc_conv_rot90, forward_2x2_input_SxS_output
 from pathlib import Path
-from einops import repeat

 class RCLutCentered_3x3(nn.Module):
    def __init__(