Python function
is_diffusion_pipeline
is_diffusion_pipeline()
max.pipelines.modeling.weights.is_diffusion_pipeline(repo)
Check if a Hugging Face repository is a diffusion pipeline.
Diffusion pipelines typically have a model_index.json file that describes the pipeline components.
-
Parameters:
-
repo (HuggingFaceRepo) – The HuggingFaceRepo to check.
-
Returns:
-
True if the repository appears to be a diffusion pipeline, False otherwise.
-
Return type:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!