Types & classes — github.com/claudiom4sir/StableVSR

CodeHub MCPcopy Create free account

hub / github.com/claudiom4sir/StableVSR / types & classes

Types & classes5 in github.com/claudiom4sir/StableVSR

⨍ Functions 53 ◇ Types & classes 5

↓ 1 callersClassDDPMSchedulerOutput

Output class for the scheduler's `step` function output. Args: prev_sample (`torch.FloatTensor` of shape `(batch_size, num_channels,

scheduler/ddpm_scheduler.py:31

↓ 1 callersClassREDSRecurrentDataset

REDS dataset for training recurrent networks. The keys are generated from a meta info txt file. basicsr/data/meta_info/meta_info_REDS_GT.txt

dataset/reds_dataset.py:207

ClassDDPMScheduler

`DDPMScheduler` explores the connections between denoising score matching and Langevin dynamics sampling. This model inherits from [`Schedul

scheduler/ddpm_scheduler.py:92

ClassREDSDataset

REDS dataset for training. The keys are generated from a meta info txt file. basicsr/data/meta_info/meta_info_REDS_GT.txt Each line cont

dataset/reds_dataset.py:11

ClassStableVSRPipeline

r""" Pipeline for text-to-image generation using Stable Diffusion with ControlNet guidance. This model inherits from [`DiffusionPipeline`]. C

pipeline/stablevsr_pipeline.py:96