Functions — github.com/DreamEditBenchTeam/DreamEdit

hub / github.com/DreamEditBenchTeam/DreamEdit / functions

Functions598 in github.com/DreamEditBenchTeam/DreamEdit

⨍ Functions 598◇ Types & classes 104

Functionconvert_module_to_f32

(x)

src/ldm/modules/diffusionmodules/openaimodel.py:27

Methodconvert_to_fp16

Convert the torso of the model to float16.

src/ldm/modules/diffusionmodules/openaimodel.py:694

Methodconvert_to_fp16

Convert the torso of the model to float16.

src/ldm/modules/diffusionmodules/openaimodel.py:924

Methodconvert_to_fp32

Convert the torso of the model to float32.

src/ldm/modules/diffusionmodules/openaimodel.py:702

Methodconvert_to_fp32

Convert the torso of the model to float32.

src/ldm/modules/diffusionmodules/openaimodel.py:931

Methodcount_flops

(model, _x, y)

src/ldm/modules/diffusionmodules/openaimodel.py:375

Methodcount_flops

(model, _x, y)

src/ldm/modules/diffusionmodules/openaimodel.py:409

Methoddecode

(self, text)

src/ldm/modules/encoders/modules.py:76

Methoddecode

(self, h, force_not_quantize=False)

src/ldm/models/autoencoder.py:274

Methoddecode

(self, x, *args, **kwargs)

src/ldm/models/autoencoder.py:434

Methoddecode_code

(self, code_b)

src/ldm/models/autoencoder.py:112

Functiondegradation_bsrgan

This is the degradation model of BSRGAN from the paper "Designing a Practical Degradation Model for Deep Blind Image Super-Resolution" --

src/ldm/modules/image_degradation/bsrgan_light.py:442

Functiondegradation_bsrgan

This is the degradation model of BSRGAN from the paper "Designing a Practical Degradation Model for Deep Blind Image Super-Resolution" --

src/ldm/modules/image_degradation/bsrgan.py:438

Functiondegradation_bsrgan_plus

This is an extended degradation model by combining the degradation models of BSRGAN and Real-ESRGAN ---------- img: HXWXC, [0, 1], it

src/ldm/modules/image_degradation/bsrgan.py:617

Functiondegradation_bsrgan_variant

This is the degradation model of BSRGAN from the paper "Designing a Practical Degradation Model for Deep Blind Image Super-Resolution" --

src/ldm/modules/image_degradation/bsrgan_light.py:534

Functiondegradation_bsrgan_variant

This is the degradation model of BSRGAN from the paper "Designing a Practical Degradation Model for Deep Blind Image Super-Resolution" --

src/ldm/modules/image_degradation/bsrgan.py:530

Methoddifferentiable_decode_first_stage

(self, z, predict_cids=False, force_not_quantize=False)

src/ldm/models/diffusion/ddpm.py:766

Functiondisabled_train

Overwrite model.train with this function to make sure train/eval mode does not change anymore.

src/ldm/models/diffusion/classifier.py:22

Functiondisabled_train

Overwrite model.train with this function to make sure train/eval mode does not change anymore.

src/ldm/models/diffusion/ddpm.py:34

Functiondpsr_degradation

bicubic downsampling + blur Args: x: HxWxC image, [0, 1] k: hxw, double sf: down-scale factor Return: downsam

src/ldm/modules/image_degradation/bsrgan_light.py:262

Functiondpsr_degradation

bicubic downsampling + blur Args: x: HxWxC image, [0, 1] k: hxw, double sf: down-scale factor Return: downsam

src/ldm/modules/image_degradation/bsrgan.py:262

Methodencode

(self, *args, **kwargs)

src/ldm/modules/encoders/modules.py:16

Methodencode

(self, x)

src/ldm/modules/encoders/modules.py:49

Methodencode

(self, text)

src/ldm/modules/encoders/modules.py:70

Methodencode

(self, x)

src/ldm/modules/encoders/modules.py:134

Methodencode

(self, text)

src/ldm/modules/encoders/modules.py:161

Methodencode

(self, text)

src/ldm/modules/encoders/modules.py:189

Methodencode

(self, x)

src/ldm/models/autoencoder.py:269

Methodencode

(self, x, *args, **kwargs)

src/ldm/models/autoencoder.py:431

Methodencode_to_prequant

(self, x)

src/ldm/models/autoencoder.py:102

Methodforward

(self,model)

src/ldm/modules/ema.py:25

Methodforward

(self, x)

src/ldm/modules/x_transformer.py:34

Methodforward

(self, x, seq_dim=1, offset=0)

src/ldm/modules/x_transformer.py:45

Methodforward

(self, x, **kwargs)

src/ldm/modules/x_transformer.py:123

Methodforward

(self, x, **kwargs)

src/ldm/modules/x_transformer.py:134

Methodforward

(self, x)

src/ldm/modules/x_transformer.py:146

Methodforward

(self, x)

src/ldm/modules/x_transformer.py:158

Methodforward

(self, x, residual)

src/ldm/modules/x_transformer.py:164

Methodforward

(self, x, residual)

src/ldm/modules/x_transformer.py:173

Methodforward

(self, x)

src/ldm/modules/x_transformer.py:189

Methodforward

(self, x)

src/ldm/modules/x_transformer.py:210

Methodforward

( self, x, context=None, mask=None, context_mask=N

src/ldm/modules/x_transformer.py:268

Methodforward

( self, x, context=None, mask=None, context_mask=N

src/ldm/modules/x_transformer.py:481

Methodforward

( self, x, return_embeddings=False, mask=None, ret

src/ldm/modules/x_transformer.py:598

Methodforward

(self, x)

src/ldm/modules/attention.py:42

Methodforward

(self, x)

src/ldm/modules/attention.py:63

Methodforward

(self, x)

src/ldm/modules/attention.py:88

Methodforward

(self, x)

src/ldm/modules/attention.py:126

Methodforward

(self, x, context=None, mask=None)

src/ldm/modules/attention.py:170

Methodforward

(self, x, context=None)

src/ldm/modules/attention.py:208

Methodforward

(self, x, context=None)

src/ldm/modules/attention.py:250

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/openaimodel.py:51

Methodforward

Apply the module to `x` given `emb` timestep embeddings.

src/ldm/modules/diffusionmodules/openaimodel.py:68

Methodforward

(self, x, emb, context=None)

src/ldm/modules/diffusionmodules/openaimodel.py:80

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/openaimodel.py:109

Methodforward

(self,x)

src/ldm/modules/diffusionmodules/openaimodel.py:130

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/openaimodel.py:158

Methodforward

Apply the block to a Tensor, conditioned on a timestep embedding. :param x: an [N x C x ...] Tensor of features. :param emb:

src/ldm/modules/diffusionmodules/openaimodel.py:243

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/openaimodel.py:314

Methodforward

Apply QKV attention. :param qkv: an [N x (H * 3 * C) x T] tensor of Qs, Ks, and Vs. :return: an [N x (H * C) x T] tensor afte

src/ldm/modules/diffusionmodules/openaimodel.py:356

Methodforward

Apply QKV attention. :param qkv: an [N x (3 * H * C) x T] tensor of Qs, Ks, and Vs. :return: an [N x (H * C) x T] tensor afte

src/ldm/modules/diffusionmodules/openaimodel.py:388

Methodforward

Apply the model to an input batch. :param x: an [N x C x ...] Tensor of inputs. :param timesteps: a 1-D batch of timesteps.

src/ldm/modules/diffusionmodules/openaimodel.py:710

Methodforward

Apply the model to an input batch. :param x: an [N x C x ...] Tensor of inputs. :param timesteps: a 1-D batch of timesteps.

src/ldm/modules/diffusionmodules/openaimodel.py:938

Methodforward

(ctx, run_function, length, *args)

src/ldm/modules/diffusionmodules/util.py:121

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/util.py:215

Methodforward

(self, c_concat, c_crossattn)

src/ldm/modules/diffusionmodules/util.py:258

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:53

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:72

Methodforward

(self, x, temb)

src/ldm/modules/diffusionmodules/model.py:121

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:178

Methodforward

(self, x, t=None, context=None)

src/ldm/modules/diffusionmodules/model.py:316

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:434

Methodforward

(self, z)

src/ldm/modules/diffusionmodules/model.py:535

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:594

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:641

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:680

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:705

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:722

Methodforward

(self, x)

src/ldm/modules/diffusionmodules/model.py:741

Methodforward

(self, x, scale_factor=1.0)

src/ldm/modules/diffusionmodules/model.py:763

Methodforward

(self,x)

src/ldm/modules/diffusionmodules/model.py:822

Methodforward

(self, batch, key=None)

src/ldm/modules/encoders/modules.py:27

Methodforward

(self, tokens)

src/ldm/modules/encoders/modules.py:44

Methodforward

(self, text)

src/ldm/modules/encoders/modules.py:63

Methodforward

(self, text)

src/ldm/modules/encoders/modules.py:93

Methodforward

(self,x)

src/ldm/modules/encoders/modules.py:125

Methodforward

(self, text)

src/ldm/modules/encoders/modules.py:152

Methodforward

(self, text)

src/ldm/modules/encoders/modules.py:182

Methodforward

(self, x)

src/ldm/modules/encoders/modules.py:226

Methodforward

(self, codebook_loss, inputs, reconstructions, optimizer_idx, global_step, last_layer=None, co

src/ldm/modules/losses/vqperceptual.py:98

Methodforward

(self, inputs, reconstructions, posteriors, optimizer_idx, global_step, last_layer=None, cond=

src/ldm/modules/losses/contperceptual.py:45

Methodforward

(self, input, return_pred_indices=False)

src/ldm/models/autoencoder.py:117

Methodforward

(self, input, sample_posterior=True)

src/ldm/models/autoencoder.py:335

Methodforward

(self, x, *args, **kwargs)

src/ldm/models/autoencoder.py:442

Methodforward

(self, x_noisy, t, *args, **kwargs)

src/ldm/models/diffusion/classifier.py:120

Methodforward

(self, x, *args, **kwargs)

src/ldm/models/diffusion/ddpm.py:323

Methodforward

(self, x, c, *args, **kwargs)

src/ldm/models/diffusion/ddpm.py:870

Methodforward

(self, x, t, c_concat: list = None, c_crossattn: list = None)

src/ldm/models/diffusion/ddpm.py:1402

Methodfreeze

(self)

src/ldm/modules/encoders/modules.py:177

Functiongen_kernel

# modified version of https://github.com/assafshocher/BlindSR_dataset_generator # Kai Zhang # min_var = 0.175 * sf # variance of the gau

src/ldm/modules/image_degradation/bsrgan_light.py:145

← previous next →401–500 of 598, ranked by callers