Functions — github.com/FoundationVision/LlamaGen

hub / github.com/FoundationVision/LlamaGen / functions

Functions465 in github.com/FoundationVision/LlamaGen

⨍ Functions 465◇ Types & classes 88

↓ 1 callersMethodnatural_sort

(self, l)

evaluations/t2i/evaluation.py:77

↓ 1 callersFunctionopen_npz_array

(path: str, arr_name: str)

evaluations/c2i/evaluator.py:529

↓ 1 callersFunctionprecompute_freqs_cis_2d

(grid_size: int, n_elem: int, base: int = 10000, cls_token_num=120)

autoregressive/serve/gpt_model.py:344

↓ 1 callersFunctionprefill

(model, cond_idx: torch.Tensor, input_pos: torch.Tensor, cfg_scale: float, **sampling_kwargs)

autoregressive/models/generate.py:77

↓ 1 callersMethodprepare_input_tensors

( self, seq_group_metadata_list: List[SequenceGroupMetadata], )

autoregressive/serve/model_runner.py:676

↓ 1 callersMethodprofile_run

(self)

autoregressive/serve/model_runner.py:889

↓ 1 callersFunctionraise_if_cache_size_invalid

(num_gpu_blocks, block_size, max_model_len)

autoregressive/serve/worker.py:336

↓ 1 callersFunctionrandom_crop_arr

(pil_image, image_size, min_crop_frac=0.8, max_crop_frac=1.0)

dataset/augmentation.py:29

↓ 1 callersMethodread_batch

(self, batch_size: int)

evaluations/c2i/evaluator.py:447

↓ 1 callersMethodread_batches

(self, batch_size: int)

evaluations/c2i/evaluator.py:454

↓ 1 callersMethodremaining

(self)

evaluations/c2i/evaluator.py:451

↓ 1 callersMethodremap_to_used

(self, inds)

tokenizer/vqgan/quantize.py:144

↓ 1 callersMethodremove_all_loras

(self)

autoregressive/serve/model_runner.py:955

↓ 1 callersMethodreverse

(self, output)

tokenizer/tokenizer_image/discriminator.py:140

↓ 1 callersMethodreverse

(self, output)

tokenizer/tokenizer_image/discriminator_patchgan.py:131

↓ 1 callersFunctionsample_cfg

()

tools/draw_figure.py:108

↓ 1 callersFunctionsample_topk

()

tools/draw_figure.py:80

↓ 1 callersMethodset_block_size

(self, block_size: int)

autoregressive/serve/model_runner.py:237

↓ 1 callersMethodsetup_caches

(self, max_batch_size, max_seq_length, dtype)

autoregressive/models/gpt.py:316

↓ 1 callersFunctionsetup_for_distributed

This function disables printing when not in master process

utils/distributed.py:6

↓ 1 callersFunctionsetup_fsdp_sync

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

autoregressive/train/train_c2i_fsdp.py:31

↓ 1 callersFunctionspatial_average

(x, keepdim=True)

tokenizer/tokenizer_image/lpips.py:163

↓ 1 callersFunctiontensor2pil

output image : tensor to PIL

evaluations/t2i/evaluation.py:117

↓ 1 callersMethodtext_preprocessing

(self, text)

language/t5.py:81

↓ 1 callersMethodtoken_drop

Drops labels to enable classifier-free guidance.

autoregressive/models/gpt.py:67

↓ 1 callersMethodtoken_drop

Drops labels to enable classifier-free guidance.

autoregressive/models/gpt.py:99

↓ 1 callersMethodunmap_to_all

(self, inds)

tokenizer/vqgan/quantize.py:158

↓ 1 callersMethodwarmup

(self)

evaluations/c2i/evaluator.py:147

FunctionGPT_1B

(**kwargs)

autoregressive/serve/gpt_model.py:396

FunctionGPT_1B

(**kwargs)

autoregressive/models/gpt.py:444

FunctionGPT_1B

(**kwargs)

autoregressive/models/gpt_hf.py:19

FunctionGPT_3B

(**kwargs)

autoregressive/serve/gpt_model.py:393

FunctionGPT_3B

(**kwargs)

autoregressive/models/gpt.py:441

FunctionGPT_3B

(**kwargs)

autoregressive/models/gpt_hf.py:16

FunctionGPT_7B

(**kwargs)

autoregressive/serve/gpt_model.py:390

FunctionGPT_7B

(**kwargs)

autoregressive/models/gpt.py:438

FunctionGPT_7B

(**kwargs)

autoregressive/models/gpt_hf.py:13

FunctionGPT_B

(**kwargs)

autoregressive/serve/gpt_model.py:412

FunctionGPT_B

(**kwargs)

autoregressive/models/gpt.py:460

FunctionGPT_B

(**kwargs)

autoregressive/models/gpt_hf.py:35

FunctionGPT_L

(**kwargs)

autoregressive/serve/gpt_model.py:409

FunctionGPT_L

(**kwargs)

autoregressive/models/gpt.py:457

FunctionGPT_L

(**kwargs)

autoregressive/models/gpt_hf.py:32

FunctionGPT_XL

(**kwargs)

autoregressive/serve/gpt_model.py:406

FunctionGPT_XL

(**kwargs)

autoregressive/models/gpt.py:454

FunctionGPT_XL

(**kwargs)

autoregressive/models/gpt_hf.py:29

FunctionGPT_XXL

(**kwargs)

autoregressive/serve/gpt_model.py:403

FunctionGPT_XXL

(**kwargs)

autoregressive/models/gpt.py:451

FunctionGPT_XXL

(**kwargs)

autoregressive/models/gpt_hf.py:26

FunctionGPT_XXXL

(**kwargs)

autoregressive/serve/gpt_model.py:400

FunctionGPT_XXXL

(**kwargs)

autoregressive/models/gpt.py:448

FunctionGPT_XXXL

(**kwargs)

autoregressive/models/gpt_hf.py:23

FunctionVQ_16

(**kwargs)

tokenizer/tokenizer_image/vq_model.py:421

FunctionVQ_16

(**kwargs)

tokenizer/tokenizer_image/vq_model_hf.py:14

FunctionVQ_8

(**kwargs)

tokenizer/tokenizer_image/vq_model.py:418

FunctionVQ_8

(**kwargs)

tokenizer/tokenizer_image/vq_model_hf.py:11

Method__call__

(self, img)

evaluations/t2i/evaluation.py:37

Method__call__

(self, *args, **kwargs)

autoregressive/serve/model_runner.py:1188

Method__del__

(self)

autoregressive/serve/model_runner.py:1083

Method__getitem__

(self, index)

evaluations/t2i/evaluation.py:104

Method__getitem__

(self, idx)

dataset/imagenet.py:34

dataset/openimage.py:23

language/extract_t5_feature.py:44

Method__getitem__

(self, idx)

tokenizer/vae/reconstruction_vae_ddp.py:33

Method__getitem__

(self, idx)

tokenizer/vqgan/reconstruction_vqgan_ddp.py:35

Method__getitem__

(self, idx)

tokenizer/validation/val_ddp.py:28

Method__getitem__

(self, idx)

tokenizer/consistencydecoder/reconstruction_cd_ddp.py:33

Method__getitem__

(self, index)

autoregressive/train/extract_codes_t2i.py:44

Method__init__

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

evaluations/c2i/evaluator.py:80

Method__init__

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

evaluations/c2i/evaluator.py:131

Method__init__

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

evaluations/c2i/evaluator.py:224

Method__init__

(self, session)

evaluations/c2i/evaluator.py:381

Method__init__

(self, gen_fn, length)

evaluations/c2i/evaluator.py:468

Method__init__

(self, arr_f, shape, dtype)

evaluations/c2i/evaluator.py:480

Method__init__

(self, arr)

evaluations/c2i/evaluator.py:506

Method__init__

(self, data_name, data_dir, data_type, cro

evaluations/t2i/evaluation.py:45

Method__init__

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

utils/drop_path.py:27

Method__init__

(self, feature_dir, label_dir)

dataset/imagenet.py:9

Method__init__

(self, lst_dir, face_lst_dir, transform)

dataset/t2i.py:11

Method__init__

(self, args, transform)

(self, data_path, transform=None)

dataset/openimage.py:11

Method__init__

(self, directory, transform=None)

dataset/coco.py:8

Method__init__

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

language/extract_t5_feature.py:24

Method__init__

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

language/t5.py:19

Method__init__

(self, directory, transform=None)

tokenizer/vae/reconstruction_vae_ddp.py:23

Method__init__

(self, directory, transform=None)

tokenizer/vqgan/reconstruction_vqgan_ddp.py:25

Method__init__

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

tokenizer/vqgan/quantize.py:118

Method__init__

(self, in_channels, with_conv)

tokenizer/vqgan/layer.py:18

Method__init__

(self, in_channels, with_conv)

tokenizer/vqgan/layer.py:36

Method__init__

(self, *, in_channels, out_channels=None, conv_shortcut=False, dropout, temb_channels=512)

tokenizer/vqgan/layer.py:58

Method__init__

(self, in_channels)

tokenizer/vqgan/layer.py:120

Method__init__

(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks, attn_resolutions, dropout=0.0, resam

tokenizer/vqgan/layer.py:270

Method__init__

(self, ddconfig, n_embed, embed_dim, ckpt_

tokenizer/vqgan/model.py:25

Method__init__

(self, directory, transform=None)

tokenizer/validation/val_ddp.py:18

Method__init__

(self, in_channels=3, ch=128, ch_mult=(1,1,2,2,4), num_res_blocks=2, norm_type='group', drop

tokenizer/tokenizer_image/vq_model.py:65

Method__init__

(self, z_channels=256, ch=128, ch_mult=(1,1,2,2,4), num_res_blocks=2, norm_type="group", drop

tokenizer/tokenizer_image/vq_model.py:129

← previous next →201–300 of 465, ranked by callers