Update infer/lib/predictors/DJCM/model.py

infer/lib/predictors/DJCM/model.py

@@ -5,9 +5,9 @@ import torch.nn as nn
 
 sys.path.append(os.getcwd())
 
-from main.library.predictors.DJCM.utils import init_bn
-from main.library.predictors.DJCM.decoder import PE_Decoder, SVS_Decoder
-from main.library.predictors.DJCM.encoder import ResEncoderBlock, Encoder
+from infer.lib.predictors.DJCM.utils import init_bn
+from infer.lib.predictors.DJCM.decoder import PE_Decoder, SVS_Decoder
+from infer.lib.predictors.DJCM.encoder import ResEncoderBlock, Encoder
 
 class LatentBlocks(nn.Module):
     def __init__(