From 1921a8cf4c07d0902365a26b0d3872e970890e85 Mon Sep 17 00:00:00 2001
From: alexeykozhevin <alexey.a.kozhevin@gmail.com>
Date: Thu, 17 Jul 2025 09:43:03 +0000
Subject: [PATCH] Fix load

---
 batchflow/models/torch/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/batchflow/models/torch/base.py b/batchflow/models/torch/base.py
index 54bd5372b..6196517e9 100755
--- a/batchflow/models/torch/base.py
+++ b/batchflow/models/torch/base.py
@@ -405,7 +405,7 @@ def callable_init(module): # example of a callable for init
 
     def __init__(self, config=None):
         if config is not None and not isinstance(config, (dict, Config)):
-            config = {'load/file': config}
+            config = {'load/path': config}
         self.model_lock = Lock()
 
         # Configs