MCPcopy Create free account

hub / github.com/FoundationVision/LlamaGen / functions

Functions465 in github.com/FoundationVision/LlamaGen

⨍ Functions 465◇ Types & classes 88
Method__init__
(self, n_e, e_dim, beta, entropy_loss_ratio, l2_norm, show_usage)
tokenizer/tokenizer_image/vq_model.py:198
Method__init__
(self, in_channels, out_channels=None, conv_shortcut=False, dropout=0.0, norm_type='group')
tokenizer/tokenizer_image/vq_model.py:280
Method__init__
(self, in_channels, norm_type='group')
tokenizer/tokenizer_image/vq_model.py:318
Method__init__
(self, in_channels, with_conv)
tokenizer/tokenizer_image/vq_model.py:368
Method__init__
(self, in_channels, with_conv)
tokenizer/tokenizer_image/vq_model.py:382
Method__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
tokenizer/tokenizer_image/discriminator.py:21
Method__init__
(self, num_features, logdet=False, affine=True, allow_reverse_init=False)
tokenizer/tokenizer_image/discriminator.py:80
Method__init__
(self, input_nc=3, ndf=64, n_layers=3, channel_multiplier=1, image_size=256)
tokenizer/tokenizer_image/discriminator.py:169
Method__init__
(self, input_channels, filters, downsample=True)
tokenizer/tokenizer_image/discriminator.py:213
Method__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
tokenizer/tokenizer_image/discriminator_patchgan.py:12
Method__init__
(self, disc_start, disc_loss="hinge", disc_dim=64, disc_type='patchgan', image_size=256, disc
tokenizer/tokenizer_image/vq_loss.py:50
Method__init__
(self, input_nc=3, ndf=64, n_layers=3, channel_multiplier=1, image_size=256)
tokenizer/tokenizer_image/discriminator_stylegan.py:14
Method__init__
(self, input_channels, filters, downsample=True)
tokenizer/tokenizer_image/discriminator_stylegan.py:58
Method__init__
(self)
tokenizer/tokenizer_image/lpips.py:100
Method__init__
(self, chn_in, chn_out=1, use_dropout=False)
tokenizer/tokenizer_image/lpips.py:111
Method__init__
(self, requires_grad=False, pretrained=True)
tokenizer/tokenizer_image/lpips.py:119
Method__init__
(self, directory, transform=None)
tokenizer/consistencydecoder/reconstruction_cd_ddp.py:23
Method__init__
( self, args: argparse.ArgumentParser, model: str, tokenizer: Optional[str] =
autoregressive/serve/llm.py:82
Method__init__
( self, model_config: ModelConfig, parallel_config: ParallelConfig, scheduler_
autoregressive/serve/model_runner.py:105
Method__init__
(self, model: nn.Module)
autoregressive/serve/model_runner.py:1100
Method__init__
( self, args: argparse.ArgumentParser, model_config: ModelConfig, cache_config
autoregressive/serve/gpu_executor.py:18
Method__init__
(self, num_classes, hidden_size, dropout_prob)
autoregressive/serve/gpt_model.py:54
Method__init__
(self, config: ModelArgs)
autoregressive/serve/gpt_model.py:99
Method__init__
(self, config: ModelArgs)
autoregressive/serve/gpt_model.py:185
Method__init__
(self, config: ModelArgs)
autoregressive/serve/gpt_model.py:228
Method__init__
(self, config: ModelArgs)
autoregressive/serve/gpt_model.py:245
Method__init__
(self, cfg_scale=1.0)
autoregressive/serve/sampler.py:38
Method__init__
( self, args: argparse.ArgumentParser, model_config: ModelConfig, cache_config
autoregressive/serve/llm_engine.py:85
Method__init__
( self, model_config: ModelConfig, parallel_config: ParallelConfig, scheduler_
autoregressive/serve/worker.py:35
Method__init__
(self, num_classes, hidden_size, dropout_prob)
autoregressive/models/gpt.py:60
Method__init__
(self, in_channels, hidden_size, uncond_prob, token_num=120)
autoregressive/models/gpt.py:93
Method__init__
(self, dim: int, eps: float = 1e-5)
autoregressive/models/gpt.py:138
Method__init__
(self, config: ModelArgs)
autoregressive/models/gpt.py:152
Method__init__
(self, max_batch_size, max_seq_length, n_head, head_dim, dtype)
autoregressive/models/gpt.py:171
Method__init__
(self, config: ModelArgs)
autoregressive/models/gpt.py:189
Method__init__
(self, config: ModelArgs, drop_path: float)
autoregressive/models/gpt.py:245
Method__init__
(self, config: ModelArgs)
autoregressive/models/gpt.py:261
Method__init__
(self, lst_dir, start, end, transform)
autoregressive/train/extract_codes_t2i.py:26
Method__iter__
(self)
evaluations/c2i/evaluator.py:475
Method__len__
(self)
evaluations/c2i/evaluator.py:472
Method__len__
(self)
evaluations/t2i/evaluation.py:100
Method__len__
(self)
dataset/imagenet.py:29
Method__len__
(self)
dataset/t2i.py:39
Method__len__
(self)
dataset/t2i.py:85
Method__len__
(self)
dataset/openimage.py:20
Method__len__
(self)
dataset/coco.py:15
Method__len__
(self)
language/extract_t5_feature.py:41
Method__len__
(self)
tokenizer/vae/reconstruction_vae_ddp.py:30
Method__len__
(self)
tokenizer/vqgan/reconstruction_vqgan_ddp.py:32
Method__len__
(self)
tokenizer/validation/val_ddp.py:25
Method__len__
(self)
tokenizer/consistencydecoder/reconstruction_cd_ddp.py:30
Method__len__
(self)
autoregressive/train/extract_codes_t2i.py:41
Method__reduce__
(self)
autoregressive/serve/llm_engine.py:293
Method__repr__
(self)
evaluations/t2i/evaluation.py:40
Method_init_weights
(self, module)
tokenizer/tokenizer_image/discriminator.py:67
Method_init_weights
(self, module)
tokenizer/tokenizer_image/discriminator_patchgan.py:58
Method_init_weights
(self, module)
autoregressive/models/gpt.py:307
Function_sample_with_triton_kernel
( probs: torch.Tensor, logprobs: torch.Tensor, sampling_metadata: SamplingMetadata, sampling_t
autoregressive/serve/sampler.py:520
Method_should_modify_greedy_probs_inplace
Whether or not the sampler should modify the probability distribution of greedily-sampled tokens such that multinomial sampling would sample
autoregressive/serve/sampler.py:128
Method_verify_args
(self)
autoregressive/serve/llm_engine.py:318
Methodabort_request
Aborts a request(s) with the given ID. Args: request_id: The ID(s) of the request to abort. Details: - Refer
autoregressive/serve/llm_engine.py:439
Methodadd_lora
(self, lora_request: LoRARequest)
autoregressive/serve/model_runner.py:966
Methodadd_lora
(self, lora_request: LoRARequest)
autoregressive/serve/gpu_executor.py:170
Methodadd_lora
(self, lora_request: LoRARequest)
autoregressive/serve/llm_engine.py:661
Methodadd_lora
(self, lora_request: LoRARequest)
autoregressive/serve/worker.py:257
Functionapply_rotary_emb
(x: torch.Tensor, freqs_cis: torch.Tensor)
autoregressive/serve/gpt_model.py:360
Functioncenter_crop_arr
Center cropping implementation from ADM. https://github.com/openai/guided-diffusion/blob/8fb3ad9197f16bbc40620447b2742e13458d2831/guided_diff
utils/data.py:4
Methodcheck_health
(self)
autoregressive/serve/gpu_executor.py:181
Methodcheck_health
(self)
autoregressive/serve/llm_engine.py:670
Functioncheck_image_path
(image_info)
tools/openimage_json.py:20
Functioncreate_deepspeed_config
(args)
utils/deepspeed.py:1
Methoddetermine_num_available_blocks
Determine the number of available KV blocks by invoking the underlying worker.
autoregressive/serve/gpu_executor.py:136
Methoddo_log_stats
Forced log when no requests active.
autoregressive/serve/llm_engine.py:583
Methodempty
(cls)
autoregressive/serve/model_runner.py:81
Methodevaluate
Evaluate if new feature vectors are at the manifold.
evaluations/c2i/evaluator.py:295
Methodexecute_model
( self, seq_group_metadata_list: List[SequenceGroupMetadata], blocks_to_swap_in: Dict[
autoregressive/serve/gpu_executor.py:153
Methodexecute_model_async
( self, seq_group_metadata_list: List[SequenceGroupMetadata], blocks_to_swap_in: Dict[
autoregressive/serve/gpu_executor.py:189
Methodextra_repr
(self)
utils/drop_path.py:35
Methodforward
(self, x)
utils/drop_path.py:32
Methodforward
Inputs the output of the encoder network z and maps it to a discrete one-hot vector that is the index of the closest embedding vector
tokenizer/vqgan/quantize.py:34
Methodforward
(self, z, temp=None, rescale_logits=False, return_logits=False)
tokenizer/vqgan/quantize.py:168
Methodforward
(self, x)
tokenizer/vqgan/layer.py:28
Methodforward
(self, x)
tokenizer/vqgan/layer.py:47
Methodforward
(self, x, temb)
tokenizer/vqgan/layer.py:96
Methodforward
(self, x)
tokenizer/vqgan/layer.py:147
Methodforward
(self, x)
tokenizer/vqgan/layer.py:239
Methodforward
(self, z)
tokenizer/vqgan/layer.py:339
Methodforward
(self, input)
tokenizer/vqgan/model.py:85
Methodforward
(self, input)
tokenizer/tokenizer_image/vq_model.py:57
Methodforward
(self, x)
tokenizer/tokenizer_image/vq_model.py:105
Methodforward
(self, z)
tokenizer/tokenizer_image/vq_model.py:173
Methodforward
(self, z)
tokenizer/tokenizer_image/vq_model.py:215
Methodforward
(self, x)
tokenizer/tokenizer_image/vq_model.py:299
Methodforward
(self, x)
tokenizer/tokenizer_image/vq_model.py:327
Methodforward
(self, x)
tokenizer/tokenizer_image/vq_model.py:374
Methodforward
(self, x)
tokenizer/tokenizer_image/vq_model.py:389
Methodforward
Standard forward.
tokenizer/tokenizer_image/discriminator.py:74
Methodforward
(self, input, reverse=False)
tokenizer/tokenizer_image/discriminator.py:112
Methodforward
(self, x)
tokenizer/tokenizer_image/discriminator.py:203
Methodforward
(self, x)
tokenizer/tokenizer_image/discriminator.py:229
← previousnext →301–400 of 465, ranked by callers