fixed docstring in probe module

This commit is contained in:
Lincoln Stein 2023-12-10 21:33:54 -05:00
parent de2879f602
commit 36043bf38b

View File

@ -115,11 +115,13 @@ class ModelProbe(object):
fields: Optional[Dict[str, Any]] = None, fields: Optional[Dict[str, Any]] = None,
) -> AnyModelConfig: ) -> AnyModelConfig:
""" """
Probe the model at model_path and return sufficient information about it Probe the model at model_path and return its configuration record.
to place it somewhere in the models directory hierarchy. If the model is
already loaded into memory, you may provide it as model in order to avoid :param model_path: Path to the model file (checkpoint) or directory (diffusers).
opening it a second time. The prediction_type_helper callable is a function that receives :param fields: An optional dictionary that can be used to override probed
the path to the model and returns the SchedulerPredictionType. fields. Typically used for fields that don't probe well, such as prediction_type.
Returns: The appropriate model configuration derived from ModelConfigBase.
""" """
if fields is None: if fields is None:
fields = {} fields = {}