MCPcopy Create free account

hub / github.com/Hzfinfdu/Diffusion-BERT / functions

Functions340 in github.com/Hzfinfdu/Diffusion-BERT

Methodprepare_inputs_for_generation
(self, input_ids, attention_mask=None, **model_kwargs)
models/modeling_bert_new_timestep.py:1422
Functionprocess_fn_in_collate
(wf)
predict_downstream_condition.py:115
Methodproduct_fn
(i, state)
diffusion_word_freq.py:161
Methodproduct_fn
(i, state)
diffusion_condition.py:161
Methodqt_reverse
Get q(x_{t+1} | x_t), the one-step posterior efficiently. Args: qt_plus_1: an array of floats specifying a distribution over p(x_0).
diffusion_word_freq.py:399
Methodqt_reverse
Get q(x_{t+1} | x_t), the one-step posterior efficiently. Args: qt_plus_1: an array of floats specifying a distribution over p(x_0).
diffusion_condition.py:434
Methodsample
(self, logits, x_0)
sample.py:6
Methodsample
(self, logits, x_0)
sample.py:27
Methodsample_and_compute_posterior_q
Samples from q(x_{t+1} | x_0), then computes q(x_t | x_{t+1}, x_0). Args: x_0: an array containing x_0 samples. These are expected t
diffusion_word_freq.py:197
Methodsample_and_compute_posterior_q
Samples from q(x_{t+1} | x_0), then computes q(x_t | x_{t+1}, x_0). Args: x_0: an array containing x_0 samples. These are expected t
diffusion_condition.py:197
Methodsample_stationary
Draws a sample from the stationary distribution (q(x_T)).
diffusion_word_freq.py:43
Methodsample_stationary
Draws a sample from the stationary distribution (q(x_T)).
diffusion_condition.py:43
Methodsample_stationary
(self, size)
diffusion_condition.py:410
Methodsample_t
Samples batches of time steps to use.
diffusion_condition.py:46
Functionsampling_step
(step, state)
diffusion_word_freq.py:906
Functionsampling_step
(step, state)
diffusion_condition.py:921
Functionscan
(f, init, xs, length=None)
utils.py:12
Functionschedule_fn
(step)
diffusion_word_freq.py:501
Functionschedule_fn
(step)
diffusion_condition.py:536
Functionself_bleu_for_unconditional_generation
This function is a canonical implementation of self-BLEU. The deviation from the above one is that the references are ALL THE REST sentences
compute_metric.py:92
Methodset_input_embeddings
(self, value)
models/modeling_bert.py:900
Methodset_input_embeddings
(self, value)
models/modeling_roberta.py:729
Methodset_input_embeddings
(self, value)
models/modeling_bert_new_timestep.py:931
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_bert.py:1070
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_bert.py:1181
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_bert.py:1319
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_roberta.py:906
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_roberta.py:1061
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_bert_new_timestep.py:1102
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_bert_new_timestep.py:1213
Methodset_output_embeddings
(self, new_embeddings)
models/modeling_bert_new_timestep.py:1351
Methodstationary_probs
Returns probs for the stationary distribution.
diffusion_word_freq.py:39
Methodstationary_probs
Returns probs for the stationary distribution.
diffusion_condition.py:39
Methodsupports_efficient_get
Returns true if get() is implemented/efficient.
diffusion_word_freq.py:50
Methodsupports_efficient_get
Returns true if get() is implemented/efficient.
diffusion_condition.py:50
Methodsupports_efficient_inference
Returns true if custom_product_fn is implemented. The ontology of efficient_get and efficient_inference is this: * if efficient_infe
diffusion_word_freq.py:54
Methodsupports_efficient_inference
Returns true if custom_product_fn is implemented. The ontology of efficient_get and efficient_inference is this: * if efficient_infe
diffusion_condition.py:54
Methodupdate_loss
(self, t, loss)
diffusion_condition.py:382
Methodweights
(self)
diffusion_condition.py:391
Functionword_frequency
(path, basic_freq=.5)
utils.py:24
← previous301–340 of 340, ranked by callers