def get_diffusers_pipeline()

in src/sagemaker_huggingface_inference_toolkit/diffusers_utils.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

def get_diffusers_pipeline(task=None, model_dir=None, device=-1, **kwargs):
    """Get a pipeline for Diffusers models."""
    device = "cuda" if device == 0 else "cpu"
    pipeline = DIFFUSERS_TASKS[task](model_dir=model_dir, device=device)
    return pipeline