INNER CODE UNIT · Python

tficon

Shilin-LU/TF-ICON · scripts/app.py:133

def tficon(init_img, ref_img, seg, prompt, dpm_order, dpm_steps, tau_a, tau_b, domain, seed, scale):

    n_samples = 1
    precision = "autocast"
    gpu = 'cuda'
    ddim_eta = 0.0
    dpm_order = int(dpm_order[0])
    
    # The scale used in the paper
    # if domain == 'Cross Domain':
    #     scale = 5.0
    # elif domain == 'Real Domain':
    #     scale = 2.5

    scale = scale
    
    device = DEVICE
    model = MODEL

View source record →