Functions — github.com/ThisisBillhe/NAR

CodeHub MCPcopy Create free account

hub / github.com/ThisisBillhe/NAR / functions

Functions683 in github.com/ThisisBillhe/NAR

⨍ Functions 683◇ Types & classes 116

↓ 1 callersMethodrepeat_to_16_frames

(self, video_tensor)

NAR-videos/eval/rfvd_evaluator.py:79

↓ 1 callersMethodreverse

NAR-images/tokenizer/tokenizer_image/discriminator.py:140

↓ 1 callersMethodreverse

NAR-images/tokenizer/tokenizer_image/discriminator_patchgan.py:131

↓ 1 callersMethodsample

NAR-videos/models/bottleneck.py:45

↓ 1 callersFunctionsample_cfg

NAR-images/tools/draw_figure.py:108

↓ 1 callersFunctionsample_topk

NAR-images/tools/draw_figure.py:80

↓ 1 callersFunctionsample_videos

( larp_ar_model: LARP_AR, larp_tokenizer: LARPTokenizer, output_dir: str, dataset_csv: str,

NAR-videos/sample_one.py:78

↓ 1 callersFunctionsample_videos

( larp_ar_model: LARP_AR, larp_tokenizer: LARPTokenizer, output_dir: str, dataset_csv: str,

NAR-videos/sample.py:78

↓ 1 callersMethodsampling

NAR-videos/models/larp_ar.py:322

↓ 1 callersFunctionserialize_to_tensor

NAR-videos/utils/common.py:151

↓ 1 callersMethodset_eval_deterministic

(self, deterministic=True)

NAR-videos/models/bottleneck.py:244

↓ 1 callersFunctionset_logger

NAR-videos/utils/common.py:26

↓ 1 callersMethodset_num_features

(self, num_features)

NAR-videos/utils/fvd/fvd.py:74

↓ 1 callersMethodset_stochastic_temperature

(self, temperature)

NAR-videos/models/bottleneck.py:248

↓ 1 callersMethodset_training_mode

(self, trainable_mode, others_mode=False)

NAR-videos/models/loss.py:320

↓ 1 callersMethodsetup_caches

(self, max_batch_size, max_seq_length, dtype)

NAR-images/autoregressive/models/gpt.py:339

↓ 1 callersMethodsetup_caches

(self, max_batch_size, max_seq_length, dtype)

NAR-videos/models/larp_ar.py:346

↓ 1 callersFunctionsetup_for_distributed

This function disables printing when not in master process

NAR-images/utils/distributed.py:6

↓ 1 callersFunctionsetup_fsdp_sync

(model: nn.Module, args: argparse.Namespace, device)

NAR-images/autoregressive/train/train_c2i_fsdp.py:33

↓ 1 callersFunctionspatial_average

(x, keepdim=True)

NAR-images/tokenizer/tokenizer_image/lpips.py:163

↓ 1 callersFunctiontensor2pil

output image : tensor to PIL

NAR-images/evaluations/t2i/evaluation.py:117

↓ 1 callersMethodtext_preprocessing

NAR-images/language/t5.py:81

↓ 1 callersMethodtimestep_embedding

Create sinusoidal timestep embeddings. :param t: a 1-D Tensor of N indices, one per batch element. These ma

NAR-videos/models/embed.py:141

↓ 1 callersMethodtoken_drop

Drops labels to enable classifier-free guidance.

NAR-images/autoregressive/models/gpt.py:69

↓ 1 callersMethodtoken_drop

Drops labels to enable classifier-free guidance.

NAR-images/autoregressive/models/gpt.py:101

↓ 1 callersMethodtoken_drop

Drops labels to enable classifier-free guidance. labels: (b, n) LongTensor of latent token indices.

NAR-videos/models/embed.py:178

↓ 1 callersMethodtoken_drop

Drops labels to enable classifier-free guidance.

NAR-videos/models/embed.py:240

↓ 1 callersFunctiontop_k_top_p_filtering

Filter a distribution of logits using top-k and/or nucleus (top-p) filtering Args: logits: logits distribution shape (batch size, vocabula

NAR-images/autoregressive/models/generate.py:18

↓ 1 callersMethodtrain_epoch

NAR-videos/trainers/base_trainer.py:607

↓ 1 callersMethodtrain_step

NAR-videos/trainers/base_trainer.py:604

↓ 1 callersFunctiontranslate_cfg_

NAR-videos/train.py:66

↓ 1 callersMethodunmap_to_all

NAR-images/tokenizer/vqgan/quantize.py:158

↓ 1 callersMethodupdate_lecam_ema

(self, real, fake, decay=0.999)

NAR-videos/models/loss.py:327

↓ 1 callersMethodupdate_model_spec

Update model_spec with cfg.

NAR-videos/trainers/base_trainer.py:349

↓ 1 callersMethodvisualize_epoch

NAR-videos/trainers/base_trainer.py:778

↓ 1 callersMethodwarmup

NAR-images/evaluations/c2i/evaluator.py:151

NAR-videos/models/gptc.py:199

NAR-videos/models/gptc.py:196

NAR-videos/models/gptc.py:202

NAR-videos/models/gptc.py:205

FunctionGPTC_XS

NAR-videos/models/gptc.py:208

FunctionGPTC_XXS

NAR-videos/models/gptc.py:211

NAR-images/autoregressive/models/gpt.py:508

NAR-images/autoregressive/models/gpt.py:505

NAR-images/autoregressive/models/gpt.py:502

NAR-images/autoregressive/models/gpt.py:527

NAR-images/autoregressive/models/gpt.py:521

NAR-images/autoregressive/models/gpt.py:524

NAR-images/autoregressive/models/gpt.py:518

FunctionGPT_XXL

NAR-images/autoregressive/models/gpt.py:515

FunctionGPT_XXXL

NAR-images/autoregressive/models/gpt.py:512

FunctionLLAMA_ABS_B

NAR-videos/models/larp_ar.py:584

FunctionLLAMA_ABS_L

NAR-videos/models/larp_ar.py:581

FunctionLLAMA_ABS_LP

NAR-videos/models/larp_ar.py:578

FunctionLLAMA_ABS_S

NAR-videos/models/larp_ar.py:587

FunctionLLAMA_ABS_XL

NAR-videos/models/larp_ar.py:575

FunctionLLAMA_ABS_XXL

NAR-videos/models/larp_ar.py:572

FunctionLLAMA_ABS_XXXL

NAR-videos/models/larp_ar.py:569

NAR-images/tokenizer/tokenizer_image/vq_model.py:421

NAR-images/tokenizer/tokenizer_image/vq_model_hf.py:14

NAR-images/tokenizer/tokenizer_image/vq_model.py:418

NAR-images/tokenizer/tokenizer_image/vq_model_hf.py:11

NAR-videos/utils/fvd/fvd.py:171

NAR-images/evaluations/t2i/evaluation.py:37

Method__getitem__

NAR-images/evaluations/t2i/evaluation.py:104

Method__getitem__

NAR-images/dataset/imagenet.py:34

Method__getitem__

NAR-images/dataset/t2i.py:42

Method__getitem__

NAR-images/dataset/t2i.py:95

Method__getitem__

NAR-images/dataset/openimage.py:23

Method__getitem__

NAR-images/dataset/coco.py:18

Method__getitem__

NAR-images/language/extract_t5_feature.py:44

Method__getitem__

NAR-images/tokenizer/vae/reconstruction_vae_ddp.py:33

Method__getitem__

NAR-images/tokenizer/vqgan/reconstruction_vqgan_ddp.py:35

Method__getitem__

NAR-images/tokenizer/validation/val_ddp.py:28

Method__getitem__

NAR-images/tokenizer/consistencydecoder/reconstruction_cd_ddp.py:33

Method__getitem__

NAR-videos/utils/fvd/fvd.py:310

Method__getitem__

NAR-videos/utils/fid/fid.py:54

Method__getitem__

NAR-videos/datasets/video_dataset.py:339

(self, mu: np.ndarray, sigma: np.ndarray)

NAR-images/evaluations/c2i/evaluator.py:84

( self, session, batch_size=64, softmax_batch_size=512, )

NAR-images/evaluations/c2i/evaluator.py:135

Estimate the manifold of given feature vectors. :param session: the TensorFlow session. :param row_batch_size: row batch siz

NAR-images/evaluations/c2i/evaluator.py:228

(self, session)

NAR-images/evaluations/c2i/evaluator.py:385

(self, gen_fn, length)

NAR-images/evaluations/c2i/evaluator.py:472

(self, arr_f, shape, dtype)

NAR-images/evaluations/c2i/evaluator.py:484

NAR-images/evaluations/c2i/evaluator.py:510

(self, data_name, data_dir, data_type, cro

NAR-images/evaluations/t2i/evaluation.py:45

(self, drop_prob: float = 0., scale_by_keep: bool = True)

NAR-images/utils/drop_path.py:27

(self, feature_dir, label_dir)

NAR-images/dataset/imagenet.py:9

(self, lst_dir, face_lst_dir, transform)

NAR-images/dataset/t2i.py:11

(self, args, transform)

NAR-images/dataset/t2i.py:51

NAR-images/dataset/t2i.py:139

(self, data_path, transform=None)

NAR-images/dataset/openimage.py:11

(self, directory, transform=None)

NAR-images/dataset/coco.py:8

(self, lst_dir, start, end, caption_key, trunc_caption=False)

NAR-images/language/extract_t5_feature.py:24

(self, device, dir_or_name='t5-v1_1-xxl', *, local_cache=False, cache_dir=None, hf_token=None, use_text_prepro

NAR-images/language/t5.py:19

(self, directory, transform=None)

NAR-images/tokenizer/vae/reconstruction_vae_ddp.py:23

(self, directory, transform=None)

NAR-images/tokenizer/vqgan/reconstruction_vqgan_ddp.py:25

(self, n_e, e_dim, beta, remap=None, unknown_index="random", sane_index_shape=False, legacy=T

NAR-images/tokenizer/vqgan/quantize.py:118

(self, in_channels, with_conv)

NAR-images/tokenizer/vqgan/layer.py:18

(self, in_channels, with_conv)

NAR-images/tokenizer/vqgan/layer.py:36

← previous next →301–400 of 683, ranked by callers