Special cast to SparseModelDescription

I8dNLo · I8dNLo · commit 1c9d9be9c0c8 · 2025-02-11T20:13:58.000+01:00
diff --git a/fastembed/common/model_management.py b/fastembed/common/model_management.py
@@ -407,7 +407,7 @@ def download_model(
                 try:
                     return cls.retrieve_model_gcs(
                         model.model,
-                        url_source,
+                        str(url_source),
                         str(cache_dir),
                         local_files_only=local_files_only,
                     )
diff --git a/fastembed/late_interaction_multimodal/colpali.py b/fastembed/late_interaction_multimodal/colpali.py
@@ -140,7 +140,7 @@ def _post_process_onnx_image_output(
             Iterable[NumpyArray]: Post-processed output as NumPy arrays.
         """
         return output.model_output.reshape(
-            output.model_output.shape[0], -1, self.model_description.dim
+            output.model_output.shape[0], -1, int(self.model_description.dim)
         ).astype(np.float32)
 
     def _post_process_onnx_text_output(

Original file line number	Diff line number	Diff line change
`@@ -407,7 +407,7 @@ def download_model(`
`407`	`407`	`try:`
`408`	`408`	`return cls.retrieve_model_gcs(`
`409`	`409`	`model.model,`
`410`		`- url_source,`
	`410`	`+ str(url_source),`
`411`	`411`	`str(cache_dir),`
`412`	`412`	`local_files_only=local_files_only,`
`413`	`413`	`)`