INNER CODE UNIT · Python

load_model_and_get_prompt_embedding

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

def load_model_and_get_prompt_embedding(model, scale, device, prompts, inv=False):
           
    if inv:
        inv_emb = model.get_learned_conditioning(prompts, inv)
        c = uc = inv_emb
    else:
        inv_emb = None
        
    if scale != 1.0:
        uc = model.get_learned_conditioning([""])
    else:
        uc = None
    c = model.get_learned_conditioning(prompts)
        
    return c, uc, inv_emb
    
    
def chunk(it, size):

View source record →