Spaces:

mehreenmehreen
/

ArabicOCR

Build error

mehreenmehreen commited on May 26, 2025

Commit

cbfa252

unverified ·

1 Parent(s): b9afeb2

Update py3/utils/continuous_state.py

Files changed (1) hide show

py3/utils/continuous_state.py CHANGED Viewed

@@ -39,7 +39,7 @@ def init_model(config, sol_dir='best_validation', lf_dir='best_validation', hw_d
     if only_load is None or only_load == 'sol' or 'sol' in only_load:
         sol = StartOfLineFinder(base_0, base_1)
         sol_state = safe_load.torch_state(os.path.join(config['snapshot_path'], "sol.pt"))
-        sol.load_state_dict(sol_state, weights_only=True)
         sol.to(device)
     if only_load is None or only_load == 'lf' or 'lf' in only_load:

     if only_load is None or only_load == 'sol' or 'sol' in only_load:
         sol = StartOfLineFinder(base_0, base_1)
         sol_state = safe_load.torch_state(os.path.join(config['snapshot_path'], "sol.pt"))
+        sol.load_state_dict(sol_state)
         sol.to(device)
     if only_load is None or only_load == 'lf' or 'lf' in only_load: