vipr_reflectometry.reflectorch.load_model package

Submodules

vipr_reflectometry.reflectorch.load_model.reflectorch_model_loader module

class vipr_reflectometry.reflectorch.load_model.reflectorch_model_loader.ReflectorchModelLoader(**kw: Any)

Bases: ModelLoaderHandler

class Meta

Bases: object

label = 'reflectorch'

class vipr_reflectometry.reflectorch.load_model.reflectorch_model_loader.ReflectorchModelLoaderParams(*, config_name: str = 'b_mc_point_xray_conv_standard_L2_InputQ', model_name: str | None = None, root_dir: str | None = None, weights_format: str = 'safetensors', repo_id: str = 'valentinsingularity/reflectivity', device: str = 'cpu')

Bases: BaseModel

Parameters for the reflectorch model loader.

config_name: str

device: str

model_config: ClassVar[ConfigDict] = {'extra': 'forbid'}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

model_name: str | None

repo_id: str

root_dir: str | None

weights_format: str

Module contents