Functions — github.com/ByteDance-Seed/Cola-DLM

CodeHub MCPcopy Create free account

hub / github.com/ByteDance-Seed/Cola-DLM / functions

Functions221 in github.com/ByteDance-Seed/Cola-DLM

⨍ Functions 221◇ Types & classes 40 ↳ Endpoints 3

Methodnum_examples

tasks/gsm8k.py:15

Methodnum_examples

tasks/smoltalk.py:14

Methodnum_examples

tasks/mmlu.py:16

Methodnum_examples

tasks/customjson.py:30

Methodnum_examples

tasks/common.py:47

Methodnum_examples

tasks/spellingbee.py:45

Methodnum_examples

tasks/spellingbee.py:103

Methodoutput_multiplier

cola_dlm/modeling_cola_vae.py:235

Methodset_kv_cache

(self, flag: bool)

cola_dlm/modeling_cola_vae.py:697

Methodset_kv_cache

(self, flag: bool)

cola_dlm/modeling_cola_dit.py:405

Methodset_kv_cache

cola_dlm/modeling_cola_dit.py:561

cola_dlm/optim.py:353

Functionstr2bool

scripts/cola_sft.py:24

Functiontest_configs_instantiate

ColaDiTConfig / ColaTextVAEConfig should construct with their defaults.

tests/test_package.py:53

Functiontest_cu_seqlens_shape

tests/test_attention_utils.py:29

Functiontest_get_seqlen_and_max_seqlen

tests/test_attention_utils.py:37

Functiontest_na_block_causal_mask_multi_sample_no_crossing

Two samples of length 2 should never attend across samples.

tests/test_attention_utils.py:74

Functiontest_na_block_causal_mask_single_sample

tests/test_attention_utils.py:44

Functiontest_na_block_causal_mask_suffix_q

Q == block_size should align to the tail of K within each sample.

tests/test_attention_utils.py:103

Functiontest_public_api_importable

tests/test_package.py:24

Functiontest_version_string

tests/test_package.py:41

← previous201–221 of 221, ranked by callers