debug

foundation_bert.py

@@ -178,11 +178,9 @@ class FoundationBert(ModernBertModel):
         """
         if 'checkpoint' in pretrained_model_name_or_path:
             model_config = Path(pretrained_model_name_or_path).parent / 'train_config.yaml'
-        elif 'train_config.yaml' in os.listdir(pretrained_model_name_or_path):
-            model_config = Path(pretrained_model_name_or_path) / 'train_config.yaml'
         else:
-            raise ValueError(f"Could not find train_config.yaml in {pretrained_model_name_or_path}")
-        
+            model_config = Path(pretrained_model_name_or_path) / 'train_config.yaml'
+            
         with open(model_config, 'r') as f:
             config = yaml.load(f, Loader=MyLoader)
         kwargs['modalities'] = config['modalities']