Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Hzfinfdu/Diffusion-BERT
/ functions
Functions
340 in github.com/Hzfinfdu/Diffusion-BERT
⨍
Functions
340
◇
Types & classes
93
↳
Endpoints
3
Method
prepare_inputs_for_generation
(self, input_ids, attention_mask=None, **model_kwargs)
models/modeling_bert_new_timestep.py:1422
Function
process_fn_in_collate
(wf)
predict_downstream_condition.py:115
Method
product_fn
(i, state)
diffusion_word_freq.py:161
Method
product_fn
(i, state)
diffusion_condition.py:161
Method
qt_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
Method
qt_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
Method
sample
(self, logits, x_0)
sample.py:6
Method
sample
(self, logits, x_0)
sample.py:27
Method
sample_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
Method
sample_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
Method
sample_stationary
Draws a sample from the stationary distribution (q(x_T)).
diffusion_word_freq.py:43
Method
sample_stationary
Draws a sample from the stationary distribution (q(x_T)).
diffusion_condition.py:43
Method
sample_stationary
(self, size)
diffusion_condition.py:410
Method
sample_t
Samples batches of time steps to use.
diffusion_condition.py:46
Function
sampling_step
(step, state)
diffusion_word_freq.py:906
Function
sampling_step
(step, state)
diffusion_condition.py:921
Function
scan
(f, init, xs, length=None)
utils.py:12
Function
schedule_fn
(step)
diffusion_word_freq.py:501
Function
schedule_fn
(step)
diffusion_condition.py:536
Function
self_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
Method
set_input_embeddings
(self, value)
models/modeling_bert.py:900
Method
set_input_embeddings
(self, value)
models/modeling_roberta.py:729
Method
set_input_embeddings
(self, value)
models/modeling_bert_new_timestep.py:931
Method
set_output_embeddings
(self, new_embeddings)
models/modeling_bert.py:1070
Method
set_output_embeddings
(self, new_embeddings)
models/modeling_bert.py:1181
Method
set_output_embeddings
(self, new_embeddings)
models/modeling_bert.py:1319
Method
set_output_embeddings
(self, new_embeddings)
models/modeling_roberta.py:906
Method
set_output_embeddings
(self, new_embeddings)
models/modeling_roberta.py:1061
Method
set_output_embeddings
(self, new_embeddings)
models/modeling_bert_new_timestep.py:1102
Method
set_output_embeddings
(self, new_embeddings)
models/modeling_bert_new_timestep.py:1213
Method
set_output_embeddings
(self, new_embeddings)
models/modeling_bert_new_timestep.py:1351
Method
stationary_probs
Returns probs for the stationary distribution.
diffusion_word_freq.py:39
Method
stationary_probs
Returns probs for the stationary distribution.
diffusion_condition.py:39
Method
supports_efficient_get
Returns true if get() is implemented/efficient.
diffusion_word_freq.py:50
Method
supports_efficient_get
Returns true if get() is implemented/efficient.
diffusion_condition.py:50
Method
supports_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
Method
supports_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
Method
update_loss
(self, t, loss)
diffusion_condition.py:382
Method
weights
(self)
diffusion_condition.py:391
Function
word_frequency
(path, basic_freq=.5)
utils.py:24
← previous
301–340 of 340, ranked by callers