Skip to content

Commit d591406

Browse files
committed
make loading from raw RO model work in contrib script
1 parent 390a1f1 commit d591406

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kraken/contrib/add_neural_ro.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def cli(format_type, model, files):
3737
except:
3838
from kraken.lib.ro import ROModel
3939
net = ROModel.load_from_checkpoint(model)
40-
ro_class_mapping = net.class_mapping
40+
ro_class_mapping = net.hparams.class_mapping
4141
ro_model = net.ro_net
4242

4343
for doc in files:

0 commit comments

Comments
 (0)