def load_pipeline()

in validation/compare_models.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

def load_pipeline(model_id: str):
    pipeline = StableDiffusionInstructPix2PixPipeline.from_pretrained(
        model_id, torch_dtype=torch.float16, use_auth_token=True
    ).to("cuda")
    pipeline.enable_xformers_memory_efficient_attention()
    pipeline.set_progress_bar_config(disable=True)
    return pipeline