shared.py

from models import load_sd, sam

use_fp16 = False
use_dpm = True

sd_key = "gligen/diffusers-generation-text-box"

print(f"Using SD: {sd_key}")
model_dict = load_sd(key=sd_key, use_fp16=use_fp16, use_dpm_multistep_scheduler=use_dpm, load_inverse_scheduler=False)

sam_model_dict = sam.load_sam()