mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
fixed docstring in probe module
This commit is contained in:
parent
de2879f602
commit
36043bf38b
@ -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 = {}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user