Instructions to use microsoft/colipri with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- COLIPRI
How to use microsoft/colipri with COLIPRI:
pip install colipri
from colipri import get_model from colipri import get_processor from colipri import load_sample_ct from colipri import ZeroShotImageClassificationPipeline model = get_model().cuda() processor = get_processor() pipeline = ZeroShotImageClassificationPipeline("microsoft/colipri", processor) image = load_sample_ct() pipeline(image, ["No lung nodules", "Lung nodules"]) - Notebooks
- Google Colab
- Kaggle
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 |
|