Tony Lian
Add stage 2
1f39cf9
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()