plismbench.models.histai module#

Models from HistAI company.

class plismbench.models.histai.HibouBase(device: int | list[int] | None = -1, mixed_precision: bool = False)[source]#

Bases: Extractor

Hibou-Base model developped by HistAI available on Hugging-Face (1).

Note

  1. https://huggingface.co/histai/hibou-b

Parameters:

  • device (int | list[int] | None = DEFAULT_DEVICE,) – Compute resources to use. If None, will use all available GPUs. If -1, extraction will run on CPU.

  • mixed_precision (bool = True) – Whether to use mixed_precision.

process(image) → Tensor[source]#

Process input images.

property transform: Lambda#

Transform method to apply element wise.

class plismbench.models.histai.HibouLarge(device: int | list[int] | None = -1, mixed_precision: bool = False)[source]#

Bases: Extractor

Hibou-Large model developped by HistAI available on Hugging-Face (1).

Note

  1. https://huggingface.co/histai/hibou-l

Parameters:

  • device (int | list[int] | None = DEFAULT_DEVICE,) – Compute resources to use. If None, will use all available GPUs. If -1, extraction will run on CPU.

  • mixed_precision (bool = True) – Whether to use mixed_precision.

process(image) → Tensor[source]#

Process input images.

property transform: Lambda#

Transform method to apply element wise.