fingerprinted now should work correctly

modules/blocks.py

@@ -1,11 +1,13 @@
+from pathlib import Path
 from typing import Union
 
 import torch
 import warnings
 
 from torch import nn
-
-from modules.utils import AutoPad, Interpolate, ShapeMixin, F_x, Flatten
+import sys
+sys.path.append(str(Path(__file__).parent))
+from .util import AutoPad, Interpolate, ShapeMixin, F_x, Flatten
 
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 

modules/model_parts.py

@@ -4,7 +4,7 @@
 import torch
 from torch import nn
 
-from modules.utils import ShapeMixin
+from .util import ShapeMixin
 
 
 class Generator(nn.Module):

modules/util.py

@@ -10,7 +10,7 @@ import pytorch_lightning as pl
 
 # Utility - Modules
 ###################
-from utils.model_io import ModelParameters
+from ..utils.model_io import ModelParameters
 
 
 class ShapeMixin: