Update inference/load_model.py

inference/load_model.py

@@ -32,7 +32,7 @@ class ShardedModelLoader:
             model_path: Path to the inference directory containing shards
         """
         self.model_path = Path(model_path)
-        self.config_path = self.model_path / "config.json"
+        self.config_path = self.model_path / "model_config.json"
         self.index_path = self.model_path / "model.safetensors.index.json"
         
         # Load configuration