↓ 4 callersMethod__init__(
self,
channels,
num_res_blocks: int,
hidden_size,
hidden_dropout,
src/diffusers/models/unets/uvit_2d.py:308
↓ 4 callersMethod_combiner""" Combines a latent iamge img_vae of shape (B, C, H, W) and a CLIP-embedded image img_clip of shape (B, 1, clip_img_dim) into a sin
src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:790
↓ 4 callersMethodenable_sequential_cpu_offloadr""" Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet, text_encoder, vae and safety
src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py:182
↓ 4 callersMethodenable_sequential_cpu_offloadr""" Offloads all models (`unet`, `text_encoder`, `vae`, and `safety checker` state dicts) to CPU using 🤗 Accelerate, significantly re
src/diffusers/pipelines/kandinsky/pipeline_kandinsky_combined.py:196
↓ 4 callersMethodget_inputs(self, seed=0, get_fixed_latents=False, device="cpu", dtype=torch.float32, shape=(1, 3, 64, 64))
tests/pipelines/consistency_models/test_consistency_models.py:183