Fix import error and add source_files to config

foundation_bert.py

@@ -8,9 +8,9 @@ if current_dir not in sys.path:
 
 import torch
 import yaml
-from utils.masked_data_modeling_loss import MaskedDataLossWithSoftmax
+from .utils.masked_data_modeling_loss import MaskedDataLossWithSoftmax
 # from ..utils.contrastive_loss import ContrastiveLoss
-from utils.yaml_util import MyLoader
+from .utils.yaml_util import MyLoader
 from dataclasses import dataclass
 from transformers import BertModel, BertConfig, PretrainedConfig
 from typing import Optional, Union