fepegar commited on
Commit
9020c6f
·
1 Parent(s): 7490e4f

Cast checkpoint path to string

Browse files
Files changed (1) hide show
  1. src/colipri/model/multimodal.py +1 -1
src/colipri/model/multimodal.py CHANGED
@@ -63,7 +63,7 @@ def get_model(
63
  accelerate_logger = logging.getLogger("accelerate.utils.modeling")
64
  old_level = accelerate_logger.getEffectiveLevel()
65
  accelerate_logger.setLevel(logging.ERROR)
66
- model = load_checkpoint_and_dispatch(model, checkpoint_path)
67
  accelerate_logger.setLevel(old_level)
68
  set_verbosity(transformers_verbosity)
69
 
 
63
  accelerate_logger = logging.getLogger("accelerate.utils.modeling")
64
  old_level = accelerate_logger.getEffectiveLevel()
65
  accelerate_logger.setLevel(logging.ERROR)
66
+ model = load_checkpoint_and_dispatch(model, str(checkpoint_path))
67
  accelerate_logger.setLevel(old_level)
68
  set_verbosity(transformers_verbosity)
69