plismbench.models.genbio_ai module
Models from GenBio AI company.
-
class plismbench.models.genbio_ai.GenBioPathFM(device: int | list[int] | None = -1, mixed_precision: bool = False)[source]
Bases: Extractor
GenBio_pathlib.PathFM model developped by GenBio AI available on Hugging-Face (1).
- 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.
-
property transform: Compose
Transform method to apply element wise.