vipr.plugins.huggingface package

Submodules

vipr.plugins.huggingface.controller module

class vipr.plugins.huggingface.controller.HuggingFaceController(*args: Any, **kw: Any)

Bases: ArgparseController

class Meta

Bases: object

description = 'My Plugin Controller'

label = 'huggingface'

stacked_on = 'base'

stacked_type = 'nested'

download()

Download models for HuggingFace.

vipr.plugins.huggingface.download module

vipr.plugins.huggingface.download.download_models(extension_name: str, model_names: List[str] | None = None, app=None, force_download: bool = False)

Download models for a specific extension from Hugging Face.

Parameters:

• extension_name – Name of the extension (e.g., ‘reflectometry’)

• model_names – Optional list of specific model names to download

• app – The app instance

• force_download – Whether to force download even if files exist

vipr.plugins.huggingface.huggingface module

class vipr.plugins.huggingface.huggingface.HuggingFace(app)

Bases: object

download_for_extension(extension_name, model_names=None, force_download=False)

Utility method to download models for a specific extension

Parameters:

• extension_name – Name of the extension

• model_names – Optional list of specific model names to download

• force_download – If True, download even if files already exist

ensure_models_downloaded(force_download=False)

Ensure all configured models are downloaded

Parameters:

force_download – If True, download even if files already exist

init_hook(app)

Hook called after command line arguments are parsed

Module contents

vipr.plugins.huggingface.load(app)