cellmap_flow.models.configs.huggingface

HuggingFaceModelConfig: a cellmap_models export on the Hugging Face Hub.

It is downloaded, then built as a CellMapModelConfig. to_dict() adds the repo’s metadata.json for the pipeline builder; building a launch command never downloads it.

Attributes

logger

Classes

HuggingFaceModelConfig

Configuration class for a Hugging Face model.

Module Contents

cellmap_flow.models.configs.huggingface.logger

class cellmap_flow.models.configs.huggingface.HuggingFaceModelConfig(repo, revision=None, name=None, scale=None)

Configuration class for a Hugging Face model.

cli_name = 'huggingface'

finetunable = True

METADATA_RETRY_SECONDS = 60

repo

revision = None

name = None

scale = None

to_dict(): This config as a model entry, which registry.build_model rebuilds.