Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ThisisBillhe/NAR
/ functions
Functions
683 in github.com/ThisisBillhe/NAR
⨍
Functions
683
◇
Types & classes
116
Function
center_crop_arr
Center cropping implementation from ADM. https://github.com/openai/guided-diffusion/blob/8fb3ad9197f16bbc40620447b2742e13458d2831/guided_diff
NAR-images/utils/data.py:4
Function
check_image_path
(image_info)
NAR-images/tools/openimage_json.py:20
Function
check_website_access_bool
(url)
NAR-videos/utils/common.py:138
Function
compute_num_params
(model, full_model=True, text=True, exclude_loss=True)
NAR-videos/utils/common.py:48
Method
compute_prior_loss
(self, x: torch.Tensor)
NAR-videos/models/gptc.py:162
Method
configure_optimizers
(self, config, load_sd=False)
NAR-videos/trainers/larp_tokenizer_trainer.py:160
Method
configure_optimizers
Following minGPT: This long function is unfortunately doing something very simple and is being very defensive: We are separat
NAR-videos/trainers/larp_ar_fp_trainer.py:132
Method
configure_optimizers
Following minGPT: This long function is unfortunately doing something very simple and is being very defensive: We are separat
NAR-videos/trainers/larp_ar_trainer.py:97
Method
configure_scalers
(self, sd=None, load_sd=False)
NAR-videos/trainers/larp_tokenizer_trainer.py:214
Function
create_deepspeed_config
(args)
NAR-images/utils/deepspeed.py:1
Method
decode
(self, z)
NAR-videos/models/bottleneck.py:199
Method
decode
(self, indices)
NAR-videos/models/bottleneck.py:343
Method
decode
(self, z_bottleneck)
NAR-videos/models/bottleneck.py:376
Method
decoder_requires_grad_
(self, requires_grad)
NAR-videos/models/larp_tokenizer.py:358
Function
decorator
(cls)
NAR-videos/trainers/trainers.py:5
Function
decorator
(cls)
NAR-videos/datasets/datasets.py:8
Function
decorator
(cls)
NAR-videos/models/models.py:9
Method
device
(self)
NAR-videos/models/larp_tokenizer.py:331
Method
dtype
(self)
NAR-videos/models/larp_ar.py:317
Method
dtype
(self)
NAR-videos/models/larp_tokenizer.py:335
Method
dtype
(self)
NAR-videos/models/embed.py:137
Method
dump_cfg
(cfg, kv_dict={}, prefix='')
NAR-videos/trainers/base_trainer.py:453
Method
evaluate
Evaluate if new feature vectors are at the manifold.
NAR-images/evaluations/c2i/evaluator.py:299
Method
evaluate_step
(self, data)
NAR-videos/trainers/larp_tokenizer_trainer.py:390
Method
evaluate_step
(self, data)
NAR-videos/trainers/larp_ar_fp_trainer.py:238
Method
evaluate_step
(self, data)
NAR-videos/trainers/larp_ar_trainer.py:205
Method
extra_repr
(self)
NAR-images/utils/drop_path.py:35
Method
extra_repr
(self)
NAR-videos/models/larp_ar.py:123
Method
forward
(self, x)
NAR-images/utils/drop_path.py:32
Method
forward
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
NAR-images/tokenizer/vqgan/quantize.py:34
Method
forward
(self, z, temp=None, rescale_logits=False, return_logits=False)
NAR-images/tokenizer/vqgan/quantize.py:168
Method
forward
(self, x)
NAR-images/tokenizer/vqgan/layer.py:28
Method
forward
(self, x)
NAR-images/tokenizer/vqgan/layer.py:47
Method
forward
(self, x, temb)
NAR-images/tokenizer/vqgan/layer.py:96
Method
forward
(self, x)
NAR-images/tokenizer/vqgan/layer.py:147
Method
forward
(self, x)
NAR-images/tokenizer/vqgan/layer.py:239
Method
forward
(self, z)
NAR-images/tokenizer/vqgan/layer.py:339
Method
forward
(self, input)
NAR-images/tokenizer/vqgan/model.py:85
Method
forward
(self, input)
NAR-images/tokenizer/tokenizer_image/vq_model.py:57
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/vq_model.py:105
Method
forward
(self, z)
NAR-images/tokenizer/tokenizer_image/vq_model.py:173
Method
forward
(self, z)
NAR-images/tokenizer/tokenizer_image/vq_model.py:215
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/vq_model.py:299
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/vq_model.py:327
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/vq_model.py:374
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/vq_model.py:389
Method
forward
Standard forward.
NAR-images/tokenizer/tokenizer_image/discriminator.py:74
Method
forward
(self, input, reverse=False)
NAR-images/tokenizer/tokenizer_image/discriminator.py:112
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/discriminator.py:203
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/discriminator.py:229
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/discriminator.py:244
Method
forward
Standard forward.
NAR-images/tokenizer/tokenizer_image/discriminator_patchgan.py:65
Method
forward
(self, input, reverse=False)
NAR-images/tokenizer/tokenizer_image/discriminator_patchgan.py:103
Method
forward
(self, codebook_loss, inputs, reconstructions, optimizer_idx, global_step, last_layer=None, l
NAR-images/tokenizer/tokenizer_image/vq_loss.py:117
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/discriminator_stylegan.py:48
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/discriminator_stylegan.py:74
Method
forward
(self, x)
NAR-images/tokenizer/tokenizer_image/discriminator_stylegan.py:90
Method
forward
(self, input, target)
NAR-images/tokenizer/tokenizer_image/lpips.py:83
Method
forward
(self, inp)
NAR-images/tokenizer/tokenizer_image/lpips.py:105
Method
forward
(self, X)
NAR-images/tokenizer/tokenizer_image/lpips.py:142
Method
forward
(self, labels, train, force_drop_ids=None)
NAR-images/autoregressive/models/gpt.py:80
Method
forward
(self, caption, train, force_drop_ids=None)
NAR-images/autoregressive/models/gpt.py:112
Method
forward
(self, x)
NAR-images/autoregressive/models/gpt.py:129
Method
forward
(self, x)
NAR-images/autoregressive/models/gpt.py:148
Method
forward
(self, x)
NAR-images/autoregressive/models/gpt.py:168
Method
forward
( self, x: torch.Tensor, freqs_cis: torch.Tensor = None, input_pos: Optional[torch.Tensor] =
NAR-images/autoregressive/models/gpt.py:209
Method
forward
( self, x: torch.Tensor, freqs_cis: torch.Tensor, start_pos: int, mask: Optional[torch.Tensor] = None)
NAR-images/autoregressive/models/gpt.py:255
Method
forward
( self, idx: torch.Tensor, cond_idx: torch.Tensor, # cond_idx_or_embed accept
NAR-images/autoregressive/models/gpt.py:359
Method
forward
Get Inception feature maps Parameters ---------- inp : torch.autograd.Variable Input tensor of shape Bx3xHxW. Val
NAR-videos/utils/fid/inception.py:131
Method
forward
(self, x)
NAR-videos/utils/fid/inception.py:229
Method
forward
(self, x)
NAR-videos/utils/fid/inception.py:256
Method
forward
(self, x)
NAR-videos/utils/fid/inception.py:286
Method
forward
(self, x)
NAR-videos/utils/fid/inception.py:321
Method
forward
x: (b, c, t, h, w)
NAR-videos/models/loss.py:192
Method
forward
( self, inputs, reconstructions, global_step, for_discriminator=False,
NAR-videos/models/loss.py:338
Method
forward
(self, x)
NAR-videos/models/larp_ar.py:80
Method
forward
(self, x)
NAR-videos/models/larp_ar.py:120
Method
forward
(self, x)
NAR-videos/models/larp_ar.py:146
Method
forward
( self, x: torch.Tensor, input_pos: Optional[torch.Tensor] = None, mask: Optional[tor
NAR-videos/models/larp_ar.py:187
Method
forward
( self, x: torch.Tensor, start_pos: int, mask: Optional[torch.Tensor] = None)
NAR-videos/models/larp_ar.py:230
Method
forward
( self, idx: Optional[torch.Tensor], # (b, n) cond_idx: Optional[torch.Tensor], # co
NAR-videos/models/larp_ar.py:371
Method
forward
(self, x)
NAR-videos/models/larp_tokenizer.py:36
Method
forward
(self, data, **kwargs)
NAR-videos/models/larp_tokenizer.py:480
Method
forward
(self, x)
NAR-videos/models/norm.py:16
Method
forward
(self, x)
NAR-videos/models/transformer.py:30
Method
forward
(self, context, query)
NAR-videos/models/transformer.py:62
Method
forward
(self, x)
NAR-videos/models/embed.py:26
Method
forward
(self, t)
NAR-videos/models/embed.py:161
Method
forward
(self, labels, train, force_drop_ids=None)
NAR-videos/models/embed.py:190
Method
forward
(self, embs, train, force_drop_ids=None)
NAR-videos/models/embed.py:220
Method
forward
(self, labels, train, force_drop_ids=None)
NAR-videos/models/embed.py:251
Method
forward
(self, x)
NAR-videos/models/bottleneck.py:170
Method
forward
(self, z)
NAR-videos/models/bottleneck.py:196
Method
forward
(self, z)
NAR-videos/models/bottleneck.py:263
Method
forward
(self, z)
NAR-videos/models/bottleneck.py:358
Method
forward
(self, x, layer_past=None)
NAR-videos/models/gptc.py:54
Method
forward
(self, x, layer_past=None, return_present=False)
NAR-videos/models/gptc.py:93
Method
forward_perceptual
(self, inputs, reconstructions)
NAR-videos/models/loss.py:332
Method
from_checkpoint
(cls, ckpt_path, load_state_dict=True, version='sd')
NAR-videos/models/larp_tokenizer.py:371
Function
func_none
()
NAR-videos/datasets/video_dataset.py:27
← previous
next →
501–600 of 683, ranked by callers