Cast checkpoint path to string
Browse files
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 |
|