Skip to content

Commit

Permalink
Change the error message in memory limit (#499)
Browse files Browse the repository at this point in the history
* fix memory

* fix memory
  • Loading branch information
wanliAlex authored May 31, 2023
1 parent 95f8d60 commit ba2e99a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/marqo/s2_inference/s2_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -287,9 +287,9 @@ def _check_memory_threshold_for_model(device: str, model_size: Union[float, int]
f"without device cache check. This might break down Marqo if too many models are loaded.")
if model_size > threshold:
raise ModelCacheManagementError(
f"You are trying to load a model with size = `{model_size}` into device = `{device}`, which is larger than the device threshlod = `{threshold}`."
f"You are trying to load a model with size = `{model_size}` into device = `{device}`, which is larger than the device threshold = `{threshold}`. "
f"Marqo CANNOT find enough space for the model. Please change the threshold by adjusting the environment variables.\n"
f"You can find more detailed information at `https://docs.marqo.ai/0.0.17/Advanced-Usage/configuration/`.")
f"You can find more detailed information at `https://docs.marqo.ai/0.0.21/Advanced-Usage/configuration/`.")
return (used_memory + model_size) < threshold


Expand Down

0 comments on commit ba2e99a

Please sign in to comment.