↓ 1 callersFunctionsample_sequence_conditional(model, length, context, endoftext, z=None, num_samples=1, temperature=1, top_k=0, top_p=0.0)
dialogue/run_spacefusion_gen.py:222
↓ 1 callersFunctiontrain_step(device, model, optimizer, x_tokens, input_tokens, att_mask, cond_labels)
controlgen/run_vae_ctrl_gen.py:134
↓ 1 callersFunctiontrain_step(device, model, optimizer, inputs_src, inputs_tgt, labels_tgt, src_attention_mask, tgt_attention_mask, beta, k
dialogue/run_spacefusion_gen.py:170
↓ 1 callersFunctiontrain_step(scaler, device, model, optimizer, x_tokens, input_tokens, att_mask, loss_fn, beta, kl_rate, reg_loss_type, fr
src/adaVAE.py:228
Method__init__(self, args, encoder, decoder, eos_token,
config, AdapterConfig, add_input=False,
controlgen/ctrl_gen.py:349
Method__init__(self, args, config, AdapterConfig, sep_id, pad_id, add_input=False, add_attn=False, add_softmax=False, add_me
dialogue/spacefusion.py:24
Method__init__(self, nx, n_ctx, config, AdapterConfig, add_attn=True, add_mem=False, bias_bool=True, scale=False)
src/adapters/vae.py:202
Functioncal_interpolate(args, ada_config, model, tokenizer, device, eval_dataloader, num_interpolation_steps=10, top_k=100, top_p=0.5
src/test.py:149