update: fix some bug

core/model.py

@@ -115,8 +115,6 @@ class DiscrepancyEstimator(nn.Module):
         """
         Load the model's state_dict from the specified directory.
         """
-        if not os.path.exists(load_directory):
-            raise ValueError(f"Directory {load_directory} does not exist.")
         
         if 'gpt-j' in load_directory or 'GPT-J' in load_directory:
             model_kwargs = dict(