SYMBOL INDEX (6818 symbols across 470 files) FILE: 3drecon/raymarching/backend.py function find_cl_path (line 17) | def find_cl_path(): FILE: 3drecon/raymarching/raymarching.py class _near_far_from_aabb (line 19) | class _near_far_from_aabb(Function): method forward (line 22) | def forward(ctx, rays_o, rays_d, aabb, min_near=0.2): class _sph_from_ray (line 52) | class _sph_from_ray(Function): method forward (line 55) | def forward(ctx, rays_o, rays_d, radius): class _morton3D (line 83) | class _morton3D(Function): method forward (line 85) | def forward(ctx, coords): class _morton3D_invert (line 106) | class _morton3D_invert(Function): method forward (line 108) | def forward(ctx, indices): class _packbits (line 129) | class _packbits(Function): method forward (line 132) | def forward(ctx, grid, thresh, bitfield=None): class _march_rays_train (line 161) | class _march_rays_train(Function): method forward (line 164) | def forward(ctx, rays_o, rays_d, bound, density_bitfield, C, H, nears,... class _composite_rays_train (line 238) | class _composite_rays_train(Function): method forward (line 241) | def forward(ctx, sigmas, rgbs, deltas, rays, T_thresh=1e-4): method backward (line 273) | def backward(ctx, grad_weights_sum, grad_depth, grad_image): class _march_rays (line 297) | class _march_rays(Function): method forward (line 300) | def forward(ctx, n_alive, n_step, rays_alive, rays_t, rays_o, rays_d, ... class _composite_rays (line 351) | class _composite_rays(Function): method forward (line 354) | def forward(ctx, n_alive, n_step, rays_alive, rays_t, sigmas, rgbs, de... FILE: 3drecon/raymarching/setup.py function find_cl_path (line 18) | def find_cl_path(): FILE: 3drecon/raymarching/src/bindings.cpp function PYBIND11_MODULE (line 5) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: 3drecon/renderer/agg_net.py function weights_init (line 5) | def weights_init(m): class NeRF (line 11) | class NeRF(nn.Module): method __init__ (line 12) | def __init__(self, vol_n=8+8, feat_ch=8+16+32+3, hid_n=64): method forward (line 27) | def forward(self, vox_feat, img_feat_rgb_dir, source_img_mask): class Agg (line 44) | class Agg(nn.Module): method __init__ (line 45) | def __init__(self, feat_ch): method masked_mean_var (line 58) | def masked_mean_var(self, img_feat_rgb, source_img_mask): method forward (line 66) | def forward(self, img_feat_rgb_dir, source_img_mask): FILE: 3drecon/renderer/cost_reg_net.py class ConvBnReLU3D (line 3) | class ConvBnReLU3D(nn.Module): method __init__ (line 4) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,... method forward (line 10) | def forward(self, x): class CostRegNet (line 13) | class CostRegNet(nn.Module): method __init__ (line 14) | def __init__(self, in_channels, norm_act=nn.BatchNorm3d): method forward (line 44) | def forward(self, x): class MinCostRegNet (line 60) | class MinCostRegNet(nn.Module): method __init__ (line 61) | def __init__(self, in_channels, norm_act=nn.BatchNorm3d): method forward (line 84) | def forward(self, x): FILE: 3drecon/renderer/dummy_dataset.py class DummyDataset (line 5) | class DummyDataset(pl.LightningDataModule): method __init__ (line 6) | def __init__(self,seed): method setup (line 9) | def setup(self, stage): method train_dataloader (line 16) | def train_dataloader(self): method val_dataloader (line 19) | def val_dataloader(self): method test_dataloader (line 22) | def test_dataloader(self): class DummyData (line 25) | class DummyData(Dataset): method __init__ (line 26) | def __init__(self,is_train): method __len__ (line 29) | def __len__(self): method __getitem__ (line 35) | def __getitem__(self, index): FILE: 3drecon/renderer/feature_net.py class ConvBnReLU (line 4) | class ConvBnReLU(nn.Module): method __init__ (line 5) | def __init__(self, in_channels, out_channels, kernel_size=3, stride=1,... method forward (line 11) | def forward(self, x): class FeatureNet (line 14) | class FeatureNet(nn.Module): method __init__ (line 15) | def __init__(self, norm_act=nn.BatchNorm2d): method _upsample_add (line 28) | def _upsample_add(self, x, y): method forward (line 31) | def forward(self, x): FILE: 3drecon/renderer/neus_networks.py class Embedder (line 10) | class Embedder: method __init__ (line 11) | def __init__(self, **kwargs): method create_embedding_fn (line 15) | def create_embedding_fn(self): method embed (line 39) | def embed(self, inputs): function get_embedder (line 43) | def get_embedder(multires, input_dims=3): class SDFNetwork (line 60) | class SDFNetwork(nn.Module): method __init__ (line 61) | def __init__(self, d_in, d_out, d_hidden, n_layers, skip_in=(4,), mult... method forward (line 113) | def forward(self, inputs): method sdf (line 132) | def sdf(self, x): method sdf_hidden_appearance (line 135) | def sdf_hidden_appearance(self, x): method gradient (line 138) | def gradient(self, x): method sdf_normal (line 152) | def sdf_normal(self, x): class SDFNetworkWithFeature (line 166) | class SDFNetworkWithFeature(nn.Module): method __init__ (line 167) | def __init__(self, cube, dp_in, df_in, d_out, d_hidden, n_layers, skip... method forward (line 221) | def forward(self, points): method sdf (line 247) | def sdf(self, x): method sdf_hidden_appearance (line 250) | def sdf_hidden_appearance(self, x): method gradient (line 253) | def gradient(self, x): method sdf_normal (line 267) | def sdf_normal(self, x): class VanillaMLP (line 282) | class VanillaMLP(nn.Module): method __init__ (line 283) | def __init__(self, dim_in, dim_out, n_neurons, n_hidden_layers): method forward (line 295) | def forward(self, x): method make_linear (line 299) | def make_linear(self, dim_in, dim_out, is_first, is_last): method make_activation (line 320) | def make_activation(self): class SDFHashGridNetwork (line 327) | class SDFHashGridNetwork(nn.Module): method __init__ (line 328) | def __init__(self, bound=0.5, feats_dim=13): method forward (line 358) | def forward(self, x): method sdf (line 369) | def sdf(self, x): method gradient (line 372) | def gradient(self, x): method sdf_normal (line 386) | def sdf_normal(self, x): class RenderingFFNetwork (line 400) | class RenderingFFNetwork(nn.Module): method __init__ (line 401) | def __init__(self, in_feats_dim=12): method forward (line 422) | def forward(self, points, normals, view_dirs, feature_vectors): class RenderingNetwork (line 433) | class RenderingNetwork(nn.Module): method __init__ (line 434) | def __init__(self, d_feature, d_in, d_out, d_hidden, method forward (line 463) | def forward(self, points, normals, view_dirs, feature_vectors): class SingleVarianceNetwork (line 488) | class SingleVarianceNetwork(nn.Module): method __init__ (line 489) | def __init__(self, init_val, activation='exp'): method forward (line 494) | def forward(self, x): method warp (line 501) | def warp(self, x, inv_s): FILE: 3drecon/renderer/ngp_renderer.py function custom_meshgrid (line 17) | def custom_meshgrid(*args): function sample_pdf (line 24) | def sample_pdf(bins, weights, n_samples, det=False): function plot_pointcloud (line 61) | def plot_pointcloud(pc, color=None): class NGPRenderer (line 73) | class NGPRenderer(nn.Module): method __init__ (line 74) | def __init__(self, method forward (line 115) | def forward(self, x, d): method density (line 119) | def density(self, x): method color (line 122) | def color(self, x, d, mask=None, **kwargs): method reset_extra_state (line 125) | def reset_extra_state(self): method run (line 137) | def run(self, rays_o, rays_d, num_steps=128, upsample_steps=128, bg_co... method run_cuda (line 268) | def run_cuda(self, rays_o, rays_d, dt_gamma=0, bg_color=None, perturb=... method mark_untrained_grid (line 369) | def mark_untrained_grid(self, poses, intrinsic, S=64): method update_extra_state (line 434) | def update_extra_state(self, decay=0.95, S=128): method render (line 529) | def render(self, rays_o, rays_d, staged=False, max_ray_batch=4096, **k... class _trunc_exp (line 543) | class _trunc_exp(Function): method forward (line 546) | def forward(ctx, x): method backward (line 552) | def backward(ctx, g): class NGPNetwork (line 558) | class NGPNetwork(NGPRenderer): method __init__ (line 559) | def __init__(self, method forward (line 636) | def forward(self, x, d): method density (line 669) | def density(self, x): method color (line 691) | def color(self, x, d, mask=None, geo_feat=None, **kwargs): FILE: 3drecon/renderer/renderer.py function sample_pdf (line 24) | def sample_pdf(bins, weights, n_samples, det=True): function near_far_from_sphere (line 58) | def near_far_from_sphere(rays_o, rays_d, radius=DEFAULT_RADIUS): class BackgroundRemoval (line 66) | class BackgroundRemoval: method __init__ (line 67) | def __init__(self, device='cuda'): method __call__ (line 83) | def __call__(self, image): class BaseRenderer (line 91) | class BaseRenderer(nn.Module): method __init__ (line 92) | def __init__(self, train_batch_num, test_batch_num): method render_impl (line 98) | def render_impl(self, ray_batch, is_train, step): method render_with_loss (line 102) | def render_with_loss(self, ray_batch, is_train, step): method render (line 105) | def render(self, ray_batch, is_train, step): class NeuSRenderer (line 123) | class NeuSRenderer(BaseRenderer): method __init__ (line 124) | def __init__(self, train_batch_num, test_batch_num, lambda_eikonal_los... method get_vertex_colors (line 143) | def get_vertex_colors(self, vertices): method upsample (line 164) | def upsample(self, rays_o, rays_d, z_vals, sdf, n_importance, inv_s): method cat_z_vals (line 197) | def cat_z_vals(self, rays_o, rays_d, z_vals, new_z_vals, sdf, last=Fal... method sample_depth (line 214) | def sample_depth(self, rays_o, rays_d, near, far, perturb): method compute_sdf_alpha (line 242) | def compute_sdf_alpha(self, points, dists, dirs, cos_anneal_ratio, step): method get_anneal_val (line 269) | def get_anneal_val(self, step): method get_inner_mask (line 275) | def get_inner_mask(self, points): method render_impl (line 278) | def render_impl(self, ray_batch, is_train, step): method render_with_loss (line 322) | def render_with_loss(self, ray_batch, is_train, step): class NeRFRenderer (line 350) | class NeRFRenderer(BaseRenderer): method __init__ (line 351) | def __init__(self, train_batch_num, test_batch_num, bound=0.5, use_mas... method render_impl (line 363) | def render_impl(self, ray_batch, is_train, step): method render_with_loss (line 378) | def render_with_loss(self, ray_batch, is_train, step): function cartesian_to_spherical (line 396) | def cartesian_to_spherical(xyz): function get_pose (line 405) | def get_pose(target_RT): class RendererTrainer (line 412) | class RendererTrainer(pl.LightningModule): method __init__ (line 413) | def __init__(self, image_path, data_path, total_steps, warm_up_steps, ... method _construct_ray_batch (line 453) | def _construct_ray_batch(self, images_info): method load_model (line 479) | def load_model(cfg, ckpt): method _init_dataset (line 488) | def _init_dataset(self): method _shuffle_train_batch (line 548) | def _shuffle_train_batch(self): method _shuffle_train_fg_batch (line 554) | def _shuffle_train_fg_batch(self): method training_step (line 561) | def training_step(self, batch, batch_idx): method _slice_images_info (line 581) | def _slice_images_info(self, index): method validation_step (line 585) | def validation_step(self, batch, batch_idx): method configure_optimizers (line 612) | def configure_optimizers(self): FILE: 3drecon/train_renderer.py class ResumeCallBacks (line 22) | class ResumeCallBacks(Callback): method __init__ (line 23) | def __init__(self): method on_train_start (line 26) | def on_train_start(self, trainer, pl_module): function render_images (line 29) | def render_images(model, output,): function extract_fields (line 77) | def extract_fields(bound_min, bound_max, resolution, query_func, batch_s... function extract_geometry (line 97) | def extract_geometry(bound_min, bound_max, resolution, threshold, query_... function extract_mesh (line 107) | def extract_mesh(model, output, resolution=512): function main (line 118) | def main(): FILE: 3drecon/util.py function instantiate_from_config (line 6) | def instantiate_from_config(config): function get_obj_from_str (line 16) | def get_obj_from_str(string, reload=False): function read_pickle (line 23) | def read_pickle(pkl_path): function output_points (line 27) | def output_points(fn,pts,colors=None): function concat_images (line 35) | def concat_images(img0,img1,vert=False): function concat_images_list (line 49) | def concat_images_list(*args,vert=False): FILE: 4DoF/CN_encoder.py class CN_encoder (line 6) | class CN_encoder(ConvNextV2Model): method __init__ (line 7) | def __init__(self, config): method forward (line 10) | def forward( FILE: 4DoF/dataset.py class ObjaverseDataLoader (line 15) | class ObjaverseDataLoader(): method __init__ (line 16) | def __init__(self, root_dir, batch_size, total_view=12, num_workers=4): method train_dataloader (line 27) | def train_dataloader(self): method val_dataloader (line 34) | def val_dataloader(self): function cartesian_to_spherical (line 40) | def cartesian_to_spherical(xyz): function get_pose (line 49) | def get_pose(target_RT): class ObjaverseData (line 63) | class ObjaverseData(Dataset): method __init__ (line 64) | def __init__(self, method __len__ (line 106) | def __len__(self): method cartesian_to_spherical (line 109) | def cartesian_to_spherical(self, xyz): method get_T (line 118) | def get_T(self, target_RT, cond_RT): method get_pose (line 135) | def get_pose(self, target_RT): method load_im (line 148) | def load_im(self, path, color): method __getitem__ (line 161) | def __getitem__(self, index): method process_im (line 226) | def process_im(self, im): FILE: 4DoF/diffusers/commands/__init__.py class BaseDiffusersCLICommand (line 19) | class BaseDiffusersCLICommand(ABC): method register_subcommand (line 22) | def register_subcommand(parser: ArgumentParser): method run (line 26) | def run(self): FILE: 4DoF/diffusers/commands/diffusers_cli.py function main (line 21) | def main(): FILE: 4DoF/diffusers/commands/env.py function info_command_factory (line 25) | def info_command_factory(_): class EnvironmentCommand (line 29) | class EnvironmentCommand(BaseDiffusersCLICommand): method register_subcommand (line 31) | def register_subcommand(parser: ArgumentParser): method run (line 35) | def run(self): method format_dict (line 83) | def format_dict(d): FILE: 4DoF/diffusers/configuration_utils.py class FrozenDict (line 50) | class FrozenDict(OrderedDict): method __init__ (line 51) | def __init__(self, *args, **kwargs): method __delitem__ (line 59) | def __delitem__(self, *args, **kwargs): method setdefault (line 62) | def setdefault(self, *args, **kwargs): method pop (line 65) | def pop(self, *args, **kwargs): method update (line 68) | def update(self, *args, **kwargs): method __setattr__ (line 71) | def __setattr__(self, name, value): method __setitem__ (line 76) | def __setitem__(self, name, value): class ConfigMixin (line 82) | class ConfigMixin: method register_to_config (line 104) | def register_to_config(self, **kwargs): method __getattr__ (line 121) | def __getattr__(self, name: str) -> Any: method save_config (line 139) | def save_config(self, save_directory: Union[str, os.PathLike], push_to... method from_config (line 160) | def from_config(cls, config: Union[FrozenDict, Dict[str, Any]] = None,... method get_config_dict (line 246) | def get_config_dict(cls, *args, **kwargs): method load_config (line 255) | def load_config( method _get_init_keys (line 421) | def _get_init_keys(cls): method extract_init_dict (line 425) | def extract_init_dict(cls, config_dict, **kwargs): method _dict_from_json_file (line 513) | def _dict_from_json_file(cls, json_file: Union[str, os.PathLike]): method __repr__ (line 518) | def __repr__(self): method config (line 522) | def config(self) -> Dict[str, Any]: method to_json_string (line 531) | def to_json_string(self) -> str: method to_json_file (line 557) | def to_json_file(self, json_file_path: Union[str, os.PathLike]): function register_to_config (line 569) | def register_to_config(init): function flax_register_to_config (line 619) | def flax_register_to_config(cls): FILE: 4DoF/diffusers/dependency_versions_check.py function dep_version_check (line 46) | def dep_version_check(pkg, hint=None): FILE: 4DoF/diffusers/experimental/rl/value_guided_sampling.py class ValueGuidedRLPipeline (line 25) | class ValueGuidedRLPipeline(DiffusionPipeline): method __init__ (line 42) | def __init__( method normalize (line 70) | def normalize(self, x_in, key): method de_normalize (line 73) | def de_normalize(self, x_in, key): method to_torch (line 76) | def to_torch(self, x_in): method reset_x0 (line 83) | def reset_x0(self, x_in, cond, act_dim): method run_diffusion (line 88) | def run_diffusion(self, x, conditions, n_guide_steps, scale): method __call__ (line 121) | def __call__(self, obs, batch_size=64, planning_horizon=32, n_guide_st... FILE: 4DoF/diffusers/image_processor.py class VaeImageProcessor (line 27) | class VaeImageProcessor(ConfigMixin): method __init__ (line 48) | def __init__( method numpy_to_pil (line 59) | def numpy_to_pil(images: np.ndarray) -> PIL.Image.Image: method pil_to_numpy (line 75) | def pil_to_numpy(images: Union[List[PIL.Image.Image], PIL.Image.Image]... method numpy_to_pt (line 87) | def numpy_to_pt(images: np.ndarray) -> torch.FloatTensor: method pt_to_numpy (line 98) | def pt_to_numpy(images: torch.FloatTensor) -> np.ndarray: method normalize (line 106) | def normalize(images): method denormalize (line 113) | def denormalize(images): method convert_to_rgb (line 120) | def convert_to_rgb(image: PIL.Image.Image) -> PIL.Image.Image: method resize (line 127) | def resize( method preprocess (line 147) | def preprocess( method postprocess (line 215) | def postprocess( class VaeImageProcessorLDM3D (line 255) | class VaeImageProcessorLDM3D(VaeImageProcessor): method __init__ (line 273) | def __init__( method numpy_to_pil (line 283) | def numpy_to_pil(images): method rgblike_to_depthmap (line 299) | def rgblike_to_depthmap(image): method numpy_to_depth (line 309) | def numpy_to_depth(self, images): method postprocess (line 329) | def postprocess( FILE: 4DoF/diffusers/loaders.py class AttnProcsLayers (line 70) | class AttnProcsLayers(torch.nn.Module): method __init__ (line 71) | def __init__(self, state_dict: Dict[str, torch.Tensor]): class UNet2DConditionLoadersMixin (line 112) | class UNet2DConditionLoadersMixin: method load_attn_procs (line 116) | def load_attn_procs(self, pretrained_model_name_or_path_or_dict: Union... method save_attn_procs (line 329) | def save_attn_procs( class TextualInversionLoaderMixin (line 407) | class TextualInversionLoaderMixin: method maybe_convert_prompt (line 412) | def maybe_convert_prompt(self, prompt: Union[str, List[str]], tokenize... method _maybe_convert_prompt (line 439) | def _maybe_convert_prompt(self, prompt: str, tokenizer: "PreTrainedTok... method load_textual_inversion (line 469) | def load_textual_inversion( class LoraLoaderMixin (line 738) | class LoraLoaderMixin: method load_lora_weights (line 746) | def load_lora_weights(self, pretrained_model_name_or_path_or_dict: Uni... method lora_scale (line 909) | def lora_scale(self) -> float: method text_encoder_lora_attn_procs (line 915) | def text_encoder_lora_attn_procs(self): method _remove_text_encoder_monkey_patch (line 920) | def _remove_text_encoder_monkey_patch(self): method _modify_text_encoder (line 933) | def _modify_text_encoder(self, attn_processors: Dict[str, LoRAAttnProc... method _lora_attn_processor_attr_to_text_encoder_attr (line 970) | def _lora_attn_processor_attr_to_text_encoder_attr(self): method _load_text_encoder_attn_procs (line 978) | def _load_text_encoder_attn_procs( method save_lora_weights (line 1151) | def save_lora_weights( method _convert_kohya_lora_to_diffusers (line 1228) | def _convert_kohya_lora_to_diffusers(self, state_dict): class FromSingleFileMixin (line 1279) | class FromSingleFileMixin: method from_ckpt (line 1285) | def from_ckpt(cls, *args, **kwargs): method from_single_file (line 1291) | def from_single_file(cls, pretrained_model_link_or_path, **kwargs): FILE: 4DoF/diffusers/models/activations.py function get_activation (line 4) | def get_activation(act_fn): FILE: 4DoF/diffusers/models/attention.py class BasicTransformerBlock (line 27) | class BasicTransformerBlock(nn.Module): method __init__ (line 48) | def __init__( method set_chunk_feed_forward (line 126) | def set_chunk_feed_forward(self, chunk_size: Optional[int], dim: int): method forward (line 131) | def forward( class FeedForward (line 210) | class FeedForward(nn.Module): method __init__ (line 223) | def __init__( method forward (line 256) | def forward(self, hidden_states): class GELU (line 262) | class GELU(nn.Module): method __init__ (line 267) | def __init__(self, dim_in: int, dim_out: int, approximate: str = "none"): method gelu (line 272) | def gelu(self, gate): method forward (line 278) | def forward(self, hidden_states): class GEGLU (line 284) | class GEGLU(nn.Module): method __init__ (line 293) | def __init__(self, dim_in: int, dim_out: int): method gelu (line 297) | def gelu(self, gate): method forward (line 303) | def forward(self, hidden_states): class ApproximateGELU (line 308) | class ApproximateGELU(nn.Module): method __init__ (line 315) | def __init__(self, dim_in: int, dim_out: int): method forward (line 319) | def forward(self, x): class AdaLayerNorm (line 324) | class AdaLayerNorm(nn.Module): method __init__ (line 329) | def __init__(self, embedding_dim, num_embeddings): method forward (line 336) | def forward(self, x, timestep): class AdaLayerNormZero (line 343) | class AdaLayerNormZero(nn.Module): method __init__ (line 348) | def __init__(self, embedding_dim, num_embeddings): method forward (line 357) | def forward(self, x, timestep, class_labels, hidden_dtype=None): class AdaGroupNorm (line 364) | class AdaGroupNorm(nn.Module): method __init__ (line 369) | def __init__( method forward (line 383) | def forward(self, x, emb): FILE: 4DoF/diffusers/models/attention_flax.py function _query_chunk_attention (line 23) | def _query_chunk_attention(query, key, value, precision, key_chunk_size:... function jax_memory_efficient_attention (line 74) | def jax_memory_efficient_attention( class FlaxAttention (line 119) | class FlaxAttention(nn.Module): method setup (line 145) | def setup(self): method reshape_heads_to_batch_dim (line 157) | def reshape_heads_to_batch_dim(self, tensor): method reshape_batch_dim_to_heads (line 165) | def reshape_batch_dim_to_heads(self, tensor): method __call__ (line 173) | def __call__(self, hidden_states, context=None, deterministic=True): class FlaxBasicTransformerBlock (line 221) | class FlaxBasicTransformerBlock(nn.Module): method setup (line 251) | def setup(self): method __call__ (line 266) | def __call__(self, hidden_states, context, deterministic=True): class FlaxTransformer2DModel (line 288) | class FlaxTransformer2DModel(nn.Module): method setup (line 322) | def setup(self): method __call__ (line 363) | def __call__(self, hidden_states, context, deterministic=True): class FlaxFeedForward (line 388) | class FlaxFeedForward(nn.Module): method setup (line 409) | def setup(self): method __call__ (line 415) | def __call__(self, hidden_states, deterministic=True): class FlaxGEGLU (line 421) | class FlaxGEGLU(nn.Module): method setup (line 438) | def setup(self): method __call__ (line 443) | def __call__(self, hidden_states, deterministic=True): FILE: 4DoF/diffusers/models/attention_processor.py function rotate_every_two (line 36) | def rotate_every_two(x): function cape (line 42) | def cape(x, p): function cape_embed (line 48) | def cape_embed(p1, p2, qq, kk): class Attention (line 75) | class Attention(nn.Module): method __init__ (line 90) | def __init__( method set_use_memory_efficient_attention_xformers (line 205) | def set_use_memory_efficient_attention_xformers( method set_attention_slice (line 323) | def set_attention_slice(self, slice_size): method set_processor (line 344) | def set_processor(self, processor: "AttnProcessor"): method forward (line 357) | def forward(self, hidden_states, encoder_hidden_states=None, attention... method batch_to_head_dim (line 369) | def batch_to_head_dim(self, tensor): method head_to_batch_dim (line 376) | def head_to_batch_dim(self, tensor, out_dim=3): method get_attention_scores (line 387) | def get_attention_scores(self, query, key, attention_mask=None): method prepare_attention_mask (line 421) | def prepare_attention_mask(self, attention_mask, target_length, batch_... method norm_encoder_hidden_states (line 462) | def norm_encoder_hidden_states(self, encoder_hidden_states): class AttnProcessor (line 482) | class AttnProcessor: method __call__ (line 487) | def __call__( class LoRALinearLayer (line 548) | class LoRALinearLayer(nn.Module): method __init__ (line 549) | def __init__(self, in_features, out_features, rank=4, network_alpha=No... method forward (line 565) | def forward(self, hidden_states): class LoRAAttnProcessor (line 578) | class LoRAAttnProcessor(nn.Module): method __init__ (line 593) | def __init__(self, hidden_size, cross_attention_dim=None, rank=4, netw... method __call__ (line 605) | def __call__( class CustomDiffusionAttnProcessor (line 661) | class CustomDiffusionAttnProcessor(nn.Module): method __init__ (line 680) | def __init__( method __call__ (line 706) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class AttnAddedKVProcessor (line 757) | class AttnAddedKVProcessor: method __call__ (line 763) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class AttnAddedKVProcessor2_0 (line 811) | class AttnAddedKVProcessor2_0: method __init__ (line 817) | def __init__(self): method __call__ (line 823) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class LoRAAttnAddedKVProcessor (line 874) | class LoRAAttnAddedKVProcessor(nn.Module): method __init__ (line 889) | def __init__(self, hidden_size, cross_attention_dim=None, rank=4, netw... method __call__ (line 903) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class XFormersAttnAddedKVProcessor (line 955) | class XFormersAttnAddedKVProcessor: method __init__ (line 967) | def __init__(self, attention_op: Optional[Callable] = None): method __call__ (line 970) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class XFormersAttnProcessor (line 1020) | class XFormersAttnProcessor: method __init__ (line 1032) | def __init__(self, attention_op: Optional[Callable] = None): method __call__ (line 1035) | def __call__( class AttnProcessor2_0 (line 1135) | class AttnProcessor2_0: method __init__ (line 1140) | def __init__(self): method __call__ (line 1144) | def __call__( class LoRAXFormersAttnProcessor (line 1219) | class LoRAXFormersAttnProcessor(nn.Module): method __init__ (line 1240) | def __init__( method __call__ (line 1255) | def __call__( class LoRAAttnProcessor2_0 (line 1312) | class LoRAAttnProcessor2_0(nn.Module): method __init__ (line 1328) | def __init__(self, hidden_size, cross_attention_dim=None, rank=4, netw... method __call__ (line 1342) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class CustomDiffusionXFormersAttnProcessor (line 1403) | class CustomDiffusionXFormersAttnProcessor(nn.Module): method __init__ (line 1426) | def __init__( method __call__ (line 1454) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class SlicedAttnProcessor (line 1510) | class SlicedAttnProcessor: method __init__ (line 1520) | def __init__(self, slice_size): method __call__ (line 1523) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class SlicedAttnAddedKVProcessor (line 1591) | class SlicedAttnAddedKVProcessor: method __init__ (line 1601) | def __init__(self, slice_size): method __call__ (line 1604) | def __call__(self, attn: "Attention", hidden_states, encoder_hidden_st... class SpatialNorm (line 1694) | class SpatialNorm(nn.Module): method __init__ (line 1699) | def __init__( method forward (line 1709) | def forward(self, f, zq): FILE: 4DoF/diffusers/models/autoencoder_kl.py class AutoencoderKLOutput (line 28) | class AutoencoderKLOutput(BaseOutput): class AutoencoderKL (line 41) | class AutoencoderKL(ModelMixin, ConfigMixin): method __init__ (line 72) | def __init__( method _set_gradient_checkpointing (line 127) | def _set_gradient_checkpointing(self, module, value=False): method enable_tiling (line 131) | def enable_tiling(self, use_tiling: bool = True): method disable_tiling (line 139) | def disable_tiling(self): method enable_slicing (line 146) | def enable_slicing(self): method disable_slicing (line 153) | def disable_slicing(self): method attn_processors (line 162) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 186) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 221) | def set_default_attn_processor(self): method encode (line 228) | def encode(self, x: torch.FloatTensor, return_dict: bool = True) -> Au... method _decode (line 246) | def _decode(self, z: torch.FloatTensor, return_dict: bool = True) -> U... method decode (line 259) | def decode(self, z: torch.FloatTensor, return_dict: bool = True) -> Un... method blend_v (line 271) | def blend_v(self, a, b, blend_extent): method blend_h (line 277) | def blend_h(self, a, b, blend_extent): method tiled_encode (line 283) | def tiled_encode(self, x: torch.FloatTensor, return_dict: bool = True)... method tiled_decode (line 337) | def tiled_decode(self, z: torch.FloatTensor, return_dict: bool = True)... method forward (line 385) | def forward( FILE: 4DoF/diffusers/models/controlnet.py class ControlNetOutput (line 39) | class ControlNetOutput(BaseOutput): class ControlNetConditioningEmbedding (line 58) | class ControlNetConditioningEmbedding(nn.Module): method __init__ (line 68) | def __init__( method forward (line 90) | def forward(self, conditioning): class ControlNetModel (line 103) | class ControlNetModel(ModelMixin, ConfigMixin): method __init__ (line 159) | def __init__( method from_unet (line 344) | def from_unet( method attn_processors (line 400) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 424) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 459) | def set_default_attn_processor(self): method set_attention_slice (line 466) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 531) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 535) | def forward( function zero_module (line 702) | def zero_module(module): FILE: 4DoF/diffusers/models/controlnet_flax.py class FlaxControlNetOutput (line 34) | class FlaxControlNetOutput(BaseOutput): class FlaxControlNetConditioningEmbedding (line 47) | class FlaxControlNetConditioningEmbedding(nn.Module): method setup (line 52) | def setup(self): method __call__ (line 90) | def __call__(self, conditioning): class FlaxControlNetModel (line 104) | class FlaxControlNetModel(nn.Module, FlaxModelMixin, ConfigMixin): method init_weights (line 171) | def init_weights(self, rng: jax.random.KeyArray) -> FrozenDict: method setup (line 185) | def setup(self): method __call__ (line 313) | def __call__( FILE: 4DoF/diffusers/models/cross_attention.py class CrossAttention (line 41) | class CrossAttention(Attention): method __init__ (line 42) | def __init__(self, *args, **kwargs): class CrossAttnProcessor (line 48) | class CrossAttnProcessor(AttnProcessorRename): method __init__ (line 49) | def __init__(self, *args, **kwargs): class LoRACrossAttnProcessor (line 55) | class LoRACrossAttnProcessor(LoRAAttnProcessor): method __init__ (line 56) | def __init__(self, *args, **kwargs): class CrossAttnAddedKVProcessor (line 62) | class CrossAttnAddedKVProcessor(AttnAddedKVProcessor): method __init__ (line 63) | def __init__(self, *args, **kwargs): class XFormersCrossAttnProcessor (line 69) | class XFormersCrossAttnProcessor(XFormersAttnProcessor): method __init__ (line 70) | def __init__(self, *args, **kwargs): class LoRAXFormersCrossAttnProcessor (line 76) | class LoRAXFormersCrossAttnProcessor(LoRAXFormersAttnProcessor): method __init__ (line 77) | def __init__(self, *args, **kwargs): class SlicedCrossAttnProcessor (line 83) | class SlicedCrossAttnProcessor(SlicedAttnProcessor): method __init__ (line 84) | def __init__(self, *args, **kwargs): class SlicedCrossAttnAddedKVProcessor (line 90) | class SlicedCrossAttnAddedKVProcessor(SlicedAttnAddedKVProcessor): method __init__ (line 91) | def __init__(self, *args, **kwargs): FILE: 4DoF/diffusers/models/dual_transformer_2d.py class DualTransformer2DModel (line 21) | class DualTransformer2DModel(nn.Module): method __init__ (line 48) | def __init__( method forward (line 97) | def forward( FILE: 4DoF/diffusers/models/embeddings.py function get_timestep_embedding (line 24) | def get_timestep_embedding( function get_2d_sincos_pos_embed (line 67) | def get_2d_sincos_pos_embed(embed_dim, grid_size, cls_token=False, extra... function get_2d_sincos_pos_embed_from_grid (line 84) | def get_2d_sincos_pos_embed_from_grid(embed_dim, grid): function get_1d_sincos_pos_embed_from_grid (line 96) | def get_1d_sincos_pos_embed_from_grid(embed_dim, pos): class PatchEmbed (line 117) | class PatchEmbed(nn.Module): method __init__ (line 120) | def __init__( method forward (line 148) | def forward(self, latent): class TimestepEmbedding (line 157) | class TimestepEmbedding(nn.Module): method __init__ (line 158) | def __init__( method forward (line 189) | def forward(self, sample, condition=None): class Timesteps (line 204) | class Timesteps(nn.Module): method __init__ (line 205) | def __init__(self, num_channels: int, flip_sin_to_cos: bool, downscale... method forward (line 211) | def forward(self, timesteps): class GaussianFourierProjection (line 221) | class GaussianFourierProjection(nn.Module): method __init__ (line 224) | def __init__( method forward (line 238) | def forward(self, x): class ImagePositionalEmbeddings (line 251) | class ImagePositionalEmbeddings(nn.Module): method __init__ (line 275) | def __init__( method forward (line 293) | def forward(self, index): class LabelEmbedding (line 316) | class LabelEmbedding(nn.Module): method __init__ (line 326) | def __init__(self, num_classes, hidden_size, dropout_prob): method token_drop (line 333) | def token_drop(self, labels, force_drop_ids=None): method forward (line 344) | def forward(self, labels: torch.LongTensor, force_drop_ids=None): class TextImageProjection (line 352) | class TextImageProjection(nn.Module): method __init__ (line 353) | def __init__( method forward (line 366) | def forward(self, text_embeds: torch.FloatTensor, image_embeds: torch.... class ImageProjection (line 379) | class ImageProjection(nn.Module): method __init__ (line 380) | def __init__( method forward (line 392) | def forward(self, image_embeds: torch.FloatTensor): class CombinedTimestepLabelEmbeddings (line 402) | class CombinedTimestepLabelEmbeddings(nn.Module): method __init__ (line 403) | def __init__(self, num_classes, embedding_dim, class_dropout_prob=0.1): method forward (line 410) | def forward(self, timestep, class_labels, hidden_dtype=None): class TextTimeEmbedding (line 421) | class TextTimeEmbedding(nn.Module): method __init__ (line 422) | def __init__(self, encoder_dim: int, time_embed_dim: int, num_heads: i... method forward (line 429) | def forward(self, hidden_states): class TextImageTimeEmbedding (line 437) | class TextImageTimeEmbedding(nn.Module): method __init__ (line 438) | def __init__(self, text_embed_dim: int = 768, image_embed_dim: int = 7... method forward (line 444) | def forward(self, text_embeds: torch.FloatTensor, image_embeds: torch.... class ImageTimeEmbedding (line 455) | class ImageTimeEmbedding(nn.Module): method __init__ (line 456) | def __init__(self, image_embed_dim: int = 768, time_embed_dim: int = 1... method forward (line 461) | def forward(self, image_embeds: torch.FloatTensor): class ImageHintTimeEmbedding (line 468) | class ImageHintTimeEmbedding(nn.Module): method __init__ (line 469) | def __init__(self, image_embed_dim: int = 768, time_embed_dim: int = 1... method forward (line 491) | def forward(self, image_embeds: torch.FloatTensor, hint: torch.FloatTe... class AttentionPooling (line 499) | class AttentionPooling(nn.Module): method __init__ (line 502) | def __init__(self, num_heads, embed_dim, dtype=None): method forward (line 512) | def forward(self, x): FILE: 4DoF/diffusers/models/embeddings_flax.py function get_sinusoidal_embeddings (line 20) | def get_sinusoidal_embeddings( class FlaxTimestepEmbedding (line 58) | class FlaxTimestepEmbedding(nn.Module): method __call__ (line 72) | def __call__(self, temb): class FlaxTimesteps (line 79) | class FlaxTimesteps(nn.Module): method __call__ (line 92) | def __call__(self, timesteps): FILE: 4DoF/diffusers/models/modeling_flax_pytorch_utils.py function rename_key (line 28) | def rename_key(key): function rename_key_and_reshape_tensor (line 43) | def rename_key_and_reshape_tensor(pt_tuple_key, pt_tensor, random_flax_s... function convert_pytorch_state_dict_to_flax (line 90) | def convert_pytorch_state_dict_to_flax(pt_state_dict, flax_model, init_k... FILE: 4DoF/diffusers/models/modeling_flax_utils.py class FlaxModelMixin (line 45) | class FlaxModelMixin: method _from_config (line 59) | def _from_config(cls, config, **kwargs): method _cast_floating_to (line 65) | def _cast_floating_to(self, params: Union[Dict, FrozenDict], dtype: jn... method to_bf16 (line 89) | def to_bf16(self, params: Union[Dict, FrozenDict], mask: Any = None): method to_fp32 (line 128) | def to_fp32(self, params: Union[Dict, FrozenDict], mask: Any = None): method to_fp16 (line 155) | def to_fp16(self, params: Union[Dict, FrozenDict], mask: Any = None): method init_weights (line 194) | def init_weights(self, rng: jax.random.KeyArray) -> Dict: method from_pretrained (line 198) | def from_pretrained( method save_pretrained (line 495) | def save_pretrained( FILE: 4DoF/diffusers/models/modeling_pytorch_flax_utils.py function load_flax_checkpoint_in_pytorch_model (line 37) | def load_flax_checkpoint_in_pytorch_model(pt_model, model_file): function load_flax_weights_in_pytorch_model (line 58) | def load_flax_weights_in_pytorch_model(pt_model, flax_state): FILE: 4DoF/diffusers/models/modeling_utils.py function get_parameter_device (line 63) | def get_parameter_device(parameter: torch.nn.Module): function get_parameter_dtype (line 79) | def get_parameter_dtype(parameter: torch.nn.Module): function load_state_dict (line 101) | def load_state_dict(checkpoint_file: Union[str, os.PathLike], variant: O... function _load_state_dict_into_model (line 132) | def _load_state_dict_into_model(model_to_load, state_dict): class ModelMixin (line 153) | class ModelMixin(torch.nn.Module): method __init__ (line 167) | def __init__(self): method __getattr__ (line 170) | def __getattr__(self, name: str) -> Any: method is_gradient_checkpointing (line 189) | def is_gradient_checkpointing(self) -> bool: method enable_gradient_checkpointing (line 195) | def enable_gradient_checkpointing(self): method disable_gradient_checkpointing (line 204) | def disable_gradient_checkpointing(self): method set_use_memory_efficient_attention_xformers (line 212) | def set_use_memory_efficient_attention_xformers( method enable_xformers_memory_efficient_attention (line 229) | def enable_xformers_memory_efficient_attention(self, attention_op: Opt... method disable_xformers_memory_efficient_attention (line 265) | def disable_xformers_memory_efficient_attention(self): method save_pretrained (line 271) | def save_pretrained( method from_pretrained (line 332) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method _load_pretrained_model (line 732) | def _load_pretrained_model( method device (line 836) | def device(self) -> device: method dtype (line 844) | def dtype(self) -> torch.dtype: method num_parameters (line 850) | def num_parameters(self, only_trainable: bool = False, exclude_embeddi... method _convert_deprecated_attention_blocks (line 888) | def _convert_deprecated_attention_blocks(self, state_dict): method _temp_convert_self_to_deprecated_attention_blocks (line 932) | def _temp_convert_self_to_deprecated_attention_blocks(self): method _undo_temp_convert_self_to_deprecated_attention_blocks (line 959) | def _undo_temp_convert_self_to_deprecated_attention_blocks(self): FILE: 4DoF/diffusers/models/prior_transformer.py class PriorTransformerOutput (line 17) | class PriorTransformerOutput(BaseOutput): class PriorTransformer (line 29) | class PriorTransformer(ModelMixin, ConfigMixin): method __init__ (line 67) | def __init__( method attn_processors (line 164) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 188) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 223) | def set_default_attn_processor(self): method forward (line 229) | def forward( method post_process_latents (line 362) | def post_process_latents(self, prior_latents): FILE: 4DoF/diffusers/models/resnet.py class Upsample1D (line 28) | class Upsample1D(nn.Module): method __init__ (line 42) | def __init__(self, channels, use_conv=False, use_conv_transpose=False,... method forward (line 56) | def forward(self, inputs): class Downsample1D (line 69) | class Downsample1D(nn.Module): method __init__ (line 83) | def __init__(self, channels, use_conv=False, out_channels=None, paddin... method forward (line 98) | def forward(self, inputs): class Upsample2D (line 103) | class Upsample2D(nn.Module): method __init__ (line 117) | def __init__(self, channels, use_conv=False, use_conv_transpose=False,... method forward (line 137) | def forward(self, hidden_states, output_size=None): class Downsample2D (line 175) | class Downsample2D(nn.Module): method __init__ (line 189) | def __init__(self, channels, use_conv=False, out_channels=None, paddin... method forward (line 213) | def forward(self, hidden_states): class FirUpsample2D (line 225) | class FirUpsample2D(nn.Module): method __init__ (line 239) | def __init__(self, channels=None, out_channels=None, use_conv=False, f... method _upsample_2d (line 248) | def _upsample_2d(self, hidden_states, weight=None, kernel=None, factor... method forward (line 328) | def forward(self, hidden_states): class FirDownsample2D (line 338) | class FirDownsample2D(nn.Module): method __init__ (line 352) | def __init__(self, channels=None, out_channels=None, use_conv=False, f... method _downsample_2d (line 361) | def _downsample_2d(self, hidden_states, weight=None, kernel=None, fact... method forward (line 415) | def forward(self, hidden_states): class KDownsample2D (line 426) | class KDownsample2D(nn.Module): method __init__ (line 427) | def __init__(self, pad_mode="reflect"): method forward (line 434) | def forward(self, inputs): class KUpsample2D (line 443) | class KUpsample2D(nn.Module): method __init__ (line 444) | def __init__(self, pad_mode="reflect"): method forward (line 451) | def forward(self, inputs): class ResnetBlock2D (line 460) | class ResnetBlock2D(nn.Module): method __init__ (line 491) | def __init__( method forward (line 590) | def forward(self, input_tensor, temb): function rearrange_dims (line 644) | def rearrange_dims(tensor): class Conv1dBlock (line 655) | class Conv1dBlock(nn.Module): method __init__ (line 660) | def __init__(self, inp_channels, out_channels, kernel_size, n_groups=8): method forward (line 667) | def forward(self, inputs): class ResidualTemporalBlock1D (line 677) | class ResidualTemporalBlock1D(nn.Module): method __init__ (line 678) | def __init__(self, inp_channels, out_channels, embed_dim, kernel_size=5): method forward (line 690) | def forward(self, inputs, t): function upsample_2d (line 706) | def upsample_2d(hidden_states, kernel=None, factor=2, gain=1): function downsample_2d (line 743) | def downsample_2d(hidden_states, kernel=None, factor=2, gain=1): function upfirdn2d_native (line 778) | def upfirdn2d_native(tensor, kernel, up=1, down=1, pad=(0, 0)): class TemporalConvLayer (line 822) | class TemporalConvLayer(nn.Module): method __init__ (line 828) | def __init__(self, in_dim, out_dim=None, dropout=0.0): method forward (line 861) | def forward(self, hidden_states, num_frames=1): FILE: 4DoF/diffusers/models/resnet_flax.py class FlaxUpsample2D (line 19) | class FlaxUpsample2D(nn.Module): method setup (line 23) | def setup(self): method __call__ (line 32) | def __call__(self, hidden_states): class FlaxDownsample2D (line 43) | class FlaxDownsample2D(nn.Module): method setup (line 47) | def setup(self): method __call__ (line 56) | def __call__(self, hidden_states): class FlaxResnetBlock2D (line 63) | class FlaxResnetBlock2D(nn.Module): method setup (line 70) | def setup(self): method __call__ (line 106) | def __call__(self, hidden_states, temb, deterministic=True): FILE: 4DoF/diffusers/models/t5_film_transformer.py class T5FilmDecoder (line 25) | class T5FilmDecoder(ModelMixin, ConfigMixin): method __init__ (line 27) | def __init__( method encoder_decoder_mask (line 66) | def encoder_decoder_mask(self, query_input, key_input): method forward (line 70) | def forward(self, encodings_and_masks, decoder_input_tokens, decoder_n... class DecoderLayer (line 127) | class DecoderLayer(nn.Module): method __init__ (line 128) | def __init__(self, d_model, d_kv, num_heads, d_ff, dropout_rate, layer... method forward (line 153) | def forward( class T5LayerSelfAttentionCond (line 185) | class T5LayerSelfAttentionCond(nn.Module): method __init__ (line 186) | def __init__(self, d_model, d_kv, num_heads, dropout_rate): method forward (line 193) | def forward( class T5LayerCrossAttention (line 213) | class T5LayerCrossAttention(nn.Module): method __init__ (line 214) | def __init__(self, d_model, d_kv, num_heads, dropout_rate, layer_norm_... method forward (line 220) | def forward( class T5LayerFFCond (line 236) | class T5LayerFFCond(nn.Module): method __init__ (line 237) | def __init__(self, d_model, d_ff, dropout_rate, layer_norm_epsilon): method forward (line 244) | def forward(self, hidden_states, conditioning_emb=None): class T5DenseGatedActDense (line 254) | class T5DenseGatedActDense(nn.Module): method __init__ (line 255) | def __init__(self, d_model, d_ff, dropout_rate): method forward (line 263) | def forward(self, hidden_states): class T5LayerNorm (line 273) | class T5LayerNorm(nn.Module): method __init__ (line 274) | def __init__(self, hidden_size, eps=1e-6): method forward (line 282) | def forward(self, hidden_states): class NewGELUActivation (line 298) | class NewGELUActivation(nn.Module): method forward (line 304) | def forward(self, input: torch.Tensor) -> torch.Tensor: class T5FiLMLayer (line 308) | class T5FiLMLayer(nn.Module): method __init__ (line 313) | def __init__(self, in_features, out_features): method forward (line 317) | def forward(self, x, conditioning_emb): FILE: 4DoF/diffusers/models/transformer_2d.py class Transformer2DModelOutput (line 30) | class Transformer2DModelOutput(BaseOutput): class Transformer2DModel (line 43) | class Transformer2DModel(ModelMixin, ConfigMixin): method __init__ (line 72) | def __init__( method forward (line 206) | def forward( FILE: 4DoF/diffusers/models/transformer_temporal.py class TransformerTemporalModelOutput (line 27) | class TransformerTemporalModelOutput(BaseOutput): class TransformerTemporalModel (line 39) | class TransformerTemporalModel(ModelMixin, ConfigMixin): method __init__ (line 61) | def __init__( method forward (line 107) | def forward( FILE: 4DoF/diffusers/models/unet_1d.py class UNet1DOutput (line 29) | class UNet1DOutput(BaseOutput): class UNet1DModel (line 41) | class UNet1DModel(ModelMixin, ConfigMixin): method __init__ (line 75) | def __init__( method forward (line 195) | def forward( FILE: 4DoF/diffusers/models/unet_1d_blocks.py class DownResnetBlock1D (line 24) | class DownResnetBlock1D(nn.Module): method __init__ (line 25) | def __init__( method forward (line 68) | def forward(self, hidden_states, temb=None): class UpResnetBlock1D (line 86) | class UpResnetBlock1D(nn.Module): method __init__ (line 87) | def __init__( method forward (line 128) | def forward(self, hidden_states, res_hidden_states_tuple=None, temb=No... class ValueFunctionMidBlock1D (line 146) | class ValueFunctionMidBlock1D(nn.Module): method __init__ (line 147) | def __init__(self, in_channels, out_channels, embed_dim): method forward (line 158) | def forward(self, x, temb=None): class MidResTemporalBlock1D (line 166) | class MidResTemporalBlock1D(nn.Module): method __init__ (line 167) | def __init__( method forward (line 206) | def forward(self, hidden_states, temb): class OutConv1DBlock (line 219) | class OutConv1DBlock(nn.Module): method __init__ (line 220) | def __init__(self, num_groups_out, out_channels, embed_dim, act_fn): method forward (line 227) | def forward(self, hidden_states, temb=None): class OutValueFunctionBlock (line 237) | class OutValueFunctionBlock(nn.Module): method __init__ (line 238) | def __init__(self, fc_dim, embed_dim): method forward (line 248) | def forward(self, hidden_states, temb): class Downsample1d (line 277) | class Downsample1d(nn.Module): method __init__ (line 278) | def __init__(self, kernel="linear", pad_mode="reflect"): method forward (line 285) | def forward(self, hidden_states): class Upsample1d (line 294) | class Upsample1d(nn.Module): method __init__ (line 295) | def __init__(self, kernel="linear", pad_mode="reflect"): method forward (line 302) | def forward(self, hidden_states, temb=None): class SelfAttention1d (line 311) | class SelfAttention1d(nn.Module): method __init__ (line 312) | def __init__(self, in_channels, n_head=1, dropout_rate=0.0): method transpose_for_scores (line 326) | def transpose_for_scores(self, projection: torch.Tensor) -> torch.Tensor: method forward (line 332) | def forward(self, hidden_states): class ResConvBlock (line 369) | class ResConvBlock(nn.Module): method __init__ (line 370) | def __init__(self, in_channels, mid_channels, out_channels, is_last=Fa... method forward (line 387) | def forward(self, hidden_states): class UNetMidBlock1D (line 403) | class UNetMidBlock1D(nn.Module): method __init__ (line 404) | def __init__(self, mid_channels, in_channels, out_channels=None): method forward (line 432) | def forward(self, hidden_states, temb=None): class AttnDownBlock1D (line 443) | class AttnDownBlock1D(nn.Module): method __init__ (line 444) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 463) | def forward(self, hidden_states, temb=None): class DownBlock1D (line 473) | class DownBlock1D(nn.Module): method __init__ (line 474) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 487) | def forward(self, hidden_states, temb=None): class DownBlock1DNoSkip (line 496) | class DownBlock1DNoSkip(nn.Module): method __init__ (line 497) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 509) | def forward(self, hidden_states, temb=None): class AttnUpBlock1D (line 517) | class AttnUpBlock1D(nn.Module): method __init__ (line 518) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 537) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): class UpBlock1D (line 550) | class UpBlock1D(nn.Module): method __init__ (line 551) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 564) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): class UpBlock1DNoSkip (line 576) | class UpBlock1DNoSkip(nn.Module): method __init__ (line 577) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 589) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): function get_down_block (line 599) | def get_down_block(down_block_type, num_layers, in_channels, out_channel... function get_up_block (line 617) | def get_up_block(up_block_type, num_layers, in_channels, out_channels, t... function get_mid_block (line 635) | def get_mid_block(mid_block_type, num_layers, in_channels, mid_channels,... function get_out_block (line 651) | def get_out_block(*, out_block_type, num_groups_out, embed_dim, out_chan... FILE: 4DoF/diffusers/models/unet_2d.py class UNet2DOutput (line 28) | class UNet2DOutput(BaseOutput): class UNet2DModel (line 40) | class UNet2DModel(ModelMixin, ConfigMixin): method __init__ (line 88) | def __init__( method forward (line 228) | def forward( FILE: 4DoF/diffusers/models/unet_2d_blocks.py function get_down_block (line 32) | def get_down_block( function get_up_block (line 227) | def get_up_block( class UNetMidBlock2D (line 426) | class UNetMidBlock2D(nn.Module): method __init__ (line 427) | def __init__( method forward (line 507) | def forward(self, hidden_states, temb=None): class UNetMidBlock2DCrossAttn (line 517) | class UNetMidBlock2DCrossAttn(nn.Module): method __init__ (line 518) | def __init__( method forward (line 603) | def forward( class UNetMidBlock2DSimpleCrossAttn (line 629) | class UNetMidBlock2DSimpleCrossAttn(nn.Module): method __init__ (line 630) | def __init__( method forward (line 714) | def forward( class AttnDownBlock2D (line 752) | class AttnDownBlock2D(nn.Module): method __init__ (line 753) | def __init__( method forward (line 844) | def forward(self, hidden_states, temb=None, upsample_size=None): class CrossAttnDownBlock2D (line 864) | class CrossAttnDownBlock2D(nn.Module): method __init__ (line 865) | def __init__( method forward (line 952) | def forward( class DownBlock2D (line 1018) | class DownBlock2D(nn.Module): method __init__ (line 1019) | def __init__( method forward (line 1070) | def forward(self, hidden_states, temb=None): class DownEncoderBlock2D (line 1104) | class DownEncoderBlock2D(nn.Module): method __init__ (line 1105) | def __init__( method forward (line 1153) | def forward(self, hidden_states): class AttnDownEncoderBlock2D (line 1164) | class AttnDownEncoderBlock2D(nn.Module): method __init__ (line 1165) | def __init__( method forward (line 1236) | def forward(self, hidden_states): class AttnSkipDownBlock2D (line 1248) | class AttnSkipDownBlock2D(nn.Module): method __init__ (line 1249) | def __init__( method forward (line 1329) | def forward(self, hidden_states, temb=None, skip_sample=None): class SkipDownBlock2D (line 1349) | class SkipDownBlock2D(nn.Module): method __init__ (line 1350) | def __init__( method forward (line 1409) | def forward(self, hidden_states, temb=None, skip_sample=None): class ResnetDownsampleBlock2D (line 1428) | class ResnetDownsampleBlock2D(nn.Module): method __init__ (line 1429) | def __init__( method forward (line 1492) | def forward(self, hidden_states, temb=None): class SimpleCrossAttnDownBlock2D (line 1526) | class SimpleCrossAttnDownBlock2D(nn.Module): method __init__ (line 1527) | def __init__( method forward (line 1621) | def forward( class KDownBlock2D (line 1685) | class KDownBlock2D(nn.Module): method __init__ (line 1686) | def __init__( method forward (line 1731) | def forward(self, hidden_states, temb=None): class KCrossAttnDownBlock2D (line 1763) | class KCrossAttnDownBlock2D(nn.Module): method __init__ (line 1764) | def __init__( method forward (line 1828) | def forward( class AttnUpBlock2D (line 1891) | class AttnUpBlock2D(nn.Module): method __init__ (line 1892) | def __init__( method forward (line 1980) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class CrossAttnUpBlock2D (line 2000) | class CrossAttnUpBlock2D(nn.Module): method __init__ (line 2001) | def __init__( method forward (line 2084) | def forward( class UpBlock2D (line 2151) | class UpBlock2D(nn.Module): method __init__ (line 2152) | def __init__( method forward (line 2199) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class UpDecoderBlock2D (line 2232) | class UpDecoderBlock2D(nn.Module): method __init__ (line 2233) | def __init__( method forward (line 2276) | def forward(self, hidden_states, temb=None): class AttnUpDecoderBlock2D (line 2287) | class AttnUpDecoderBlock2D(nn.Module): method __init__ (line 2288) | def __init__( method forward (line 2355) | def forward(self, hidden_states, temb=None): class AttnSkipUpBlock2D (line 2367) | class AttnSkipUpBlock2D(nn.Module): method __init__ (line 2368) | def __init__( method forward (line 2458) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, s... class SkipUpBlock2D (line 2486) | class SkipUpBlock2D(nn.Module): method __init__ (line 2487) | def __init__( method forward (line 2555) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, s... class ResnetUpsampleBlock2D (line 2581) | class ResnetUpsampleBlock2D(nn.Module): method __init__ (line 2582) | def __init__( method forward (line 2648) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class SimpleCrossAttnUpBlock2D (line 2681) | class SimpleCrossAttnUpBlock2D(nn.Module): method __init__ (line 2682) | def __init__( method forward (line 2778) | def forward( class KUpBlock2D (line 2845) | class KUpBlock2D(nn.Module): method __init__ (line 2846) | def __init__( method forward (line 2893) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class KCrossAttnUpBlock2D (line 2925) | class KCrossAttnUpBlock2D(nn.Module): method __init__ (line 2926) | def __init__( method forward (line 3009) | def forward( class KAttentionBlock (line 3072) | class KAttentionBlock(nn.Module): method __init__ (line 3089) | def __init__( method _to_3d (line 3132) | def _to_3d(self, hidden_states, height, weight): method _to_4d (line 3135) | def _to_4d(self, hidden_states, height, weight): method forward (line 3138) | def forward( FILE: 4DoF/diffusers/models/unet_2d_blocks_flax.py class FlaxCrossAttnDownBlock2D (line 22) | class FlaxCrossAttnDownBlock2D(nn.Module): method setup (line 56) | def setup(self): method __call__ (line 89) | def __call__(self, hidden_states, temb, encoder_hidden_states, determi... class FlaxDownBlock2D (line 104) | class FlaxDownBlock2D(nn.Module): method setup (line 129) | def setup(self): method __call__ (line 147) | def __call__(self, hidden_states, temb, deterministic=True): class FlaxCrossAttnUpBlock2D (line 161) | class FlaxCrossAttnUpBlock2D(nn.Module): method setup (line 196) | def setup(self): method __call__ (line 230) | def __call__(self, hidden_states, res_hidden_states_tuple, temb, encod... class FlaxUpBlock2D (line 246) | class FlaxUpBlock2D(nn.Module): method setup (line 274) | def setup(self): method __call__ (line 294) | def __call__(self, hidden_states, res_hidden_states_tuple, temb, deter... class FlaxUNetMidBlock2DCrossAttn (line 309) | class FlaxUNetMidBlock2DCrossAttn(nn.Module): method setup (line 335) | def setup(self): method __call__ (line 371) | def __call__(self, hidden_states, temb, encoder_hidden_states, determi... FILE: 4DoF/diffusers/models/unet_2d_condition.py class UNet2DConditionOutput (line 54) | class UNet2DConditionOutput(BaseOutput): class UNet2DConditionModel (line 66) | class UNet2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoade... method __init__ (line 157) | def __init__( method attn_processors (line 568) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 591) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 625) | def set_default_attn_processor(self): method set_attention_slice (line 631) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 696) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 700) | def forward( FILE: 4DoF/diffusers/models/unet_2d_condition_flax.py class FlaxUNet2DConditionOutput (line 36) | class FlaxUNet2DConditionOutput(BaseOutput): class FlaxUNet2DConditionModel (line 49) | class FlaxUNet2DConditionModel(nn.Module, FlaxModelMixin, ConfigMixin): method init_weights (line 120) | def init_weights(self, rng: jax.random.KeyArray) -> FrozenDict: method setup (line 132) | def setup(self): method __call__ (line 267) | def __call__( FILE: 4DoF/diffusers/models/unet_3d_blocks.py function get_down_block (line 23) | def get_down_block( function get_up_block (line 79) | def get_up_block( class UNetMidBlock3DCrossAttn (line 135) | class UNetMidBlock3DCrossAttn(nn.Module): method __init__ (line 136) | def __init__( method forward (line 235) | def forward( class CrossAttnDownBlock3D (line 264) | class CrossAttnDownBlock3D(nn.Module): method __init__ (line 265) | def __init__( method forward (line 360) | def forward( class DownBlock3D (line 398) | class DownBlock3D(nn.Module): method __init__ (line 399) | def __init__( method forward (line 459) | def forward(self, hidden_states, temb=None, num_frames=1): class CrossAttnUpBlock3D (line 477) | class CrossAttnUpBlock3D(nn.Module): method __init__ (line 478) | def __init__( method forward (line 569) | def forward( class UpBlock3D (line 608) | class UpBlock3D(nn.Module): method __init__ (line 609) | def __init__( method forward (line 665) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... FILE: 4DoF/diffusers/models/unet_3d_condition.py class UNet3DConditionOutput (line 44) | class UNet3DConditionOutput(BaseOutput): class UNet3DConditionModel (line 56) | class UNet3DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoade... method __init__ (line 90) | def __init__( method attn_processors (line 273) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attention_slice (line 297) | def set_attention_slice(self, slice_size): method set_attn_processor (line 363) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method enable_forward_chunking (line 397) | def enable_forward_chunking(self, chunk_size=None, dim=0): method disable_forward_chunking (line 426) | def disable_forward_chunking(self): method set_default_attn_processor (line 438) | def set_default_attn_processor(self): method _set_gradient_checkpointing (line 444) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 448) | def forward( FILE: 4DoF/diffusers/models/vae.py class DecoderOutput (line 27) | class DecoderOutput(BaseOutput): class Encoder (line 39) | class Encoder(nn.Module): method __init__ (line 40) | def __init__( method forward (line 108) | def forward(self, x): class Decoder (line 152) | class Decoder(nn.Module): method __init__ (line 153) | def __init__( method forward (line 228) | def forward(self, z, latent_embeds=None): class VectorQuantizer (line 283) | class VectorQuantizer(nn.Module): method __init__ (line 292) | def __init__( method remap_to_used (line 321) | def remap_to_used(self, inds): method unmap_to_all (line 335) | def unmap_to_all(self, inds): method forward (line 345) | def forward(self, z): method get_codebook_entry (line 379) | def get_codebook_entry(self, indices, shape): class DiagonalGaussianDistribution (line 397) | class DiagonalGaussianDistribution(object): method __init__ (line 398) | def __init__(self, parameters, deterministic=False): method sample (line 410) | def sample(self, generator: Optional[torch.Generator] = None) -> torch... method kl (line 418) | def kl(self, other=None): method nll (line 434) | def nll(self, sample, dims=[1, 2, 3]): method mode (line 440) | def mode(self): FILE: 4DoF/diffusers/models/vae_flax.py class FlaxDecoderOutput (line 33) | class FlaxDecoderOutput(BaseOutput): class FlaxAutoencoderKLOutput (line 48) | class FlaxAutoencoderKLOutput(BaseOutput): class FlaxUpsample2D (line 61) | class FlaxUpsample2D(nn.Module): method setup (line 75) | def setup(self): method __call__ (line 84) | def __call__(self, hidden_states): class FlaxDownsample2D (line 95) | class FlaxDownsample2D(nn.Module): method setup (line 109) | def setup(self): method __call__ (line 118) | def __call__(self, hidden_states): class FlaxResnetBlock2D (line 125) | class FlaxResnetBlock2D(nn.Module): method setup (line 151) | def setup(self): method __call__ (line 185) | def __call__(self, hidden_states, deterministic=True): class FlaxAttentionBlock (line 202) | class FlaxAttentionBlock(nn.Module): method setup (line 222) | def setup(self): method transpose_for_scores (line 231) | def transpose_for_scores(self, projection): method __call__ (line 239) | def __call__(self, hidden_states): class FlaxDownEncoderBlock2D (line 274) | class FlaxDownEncoderBlock2D(nn.Module): method setup (line 302) | def setup(self): method __call__ (line 320) | def __call__(self, hidden_states, deterministic=True): class FlaxUpDecoderBlock2D (line 330) | class FlaxUpDecoderBlock2D(nn.Module): method setup (line 358) | def setup(self): method __call__ (line 376) | def __call__(self, hidden_states, deterministic=True): class FlaxUNetMidBlock2D (line 386) | class FlaxUNetMidBlock2D(nn.Module): method setup (line 411) | def setup(self): method __call__ (line 448) | def __call__(self, hidden_states, deterministic=True): class FlaxEncoder (line 457) | class FlaxEncoder(nn.Module): method setup (line 501) | def setup(self): method __call__ (line 550) | def __call__(self, sample, deterministic: bool = True): class FlaxDecoder (line 569) | class FlaxDecoder(nn.Module): method setup (line 612) | def setup(self): method __call__ (line 665) | def __call__(self, sample, deterministic: bool = True): class FlaxDiagonalGaussianDistribution (line 683) | class FlaxDiagonalGaussianDistribution(object): method __init__ (line 684) | def __init__(self, parameters, deterministic=False): method sample (line 694) | def sample(self, key): method kl (line 697) | def kl(self, other=None): method nll (line 709) | def nll(self, sample, axis=[1, 2, 3]): method mode (line 716) | def mode(self): class FlaxAutoencoderKL (line 721) | class FlaxAutoencoderKL(nn.Module, FlaxModelMixin, ConfigMixin): method setup (line 783) | def setup(self): method init_weights (line 820) | def init_weights(self, rng: jax.random.KeyArray) -> FrozenDict: method encode (line 830) | def encode(self, sample, deterministic: bool = True, return_dict: bool... method decode (line 842) | def decode(self, latents, deterministic: bool = True, return_dict: boo... method __call__ (line 856) | def __call__(self, sample, sample_posterior=False, deterministic: bool... FILE: 4DoF/diffusers/models/vq_model.py class VQEncoderOutput (line 27) | class VQEncoderOutput(BaseOutput): class VQModel (line 39) | class VQModel(ModelMixin, ConfigMixin): method __init__ (line 70) | def __init__( method encode (line 120) | def encode(self, x: torch.FloatTensor, return_dict: bool = True) -> VQ... method decode (line 130) | def decode( method forward (line 146) | def forward(self, sample: torch.FloatTensor, return_dict: bool = True)... FILE: 4DoF/diffusers/optimization.py class SchedulerType (line 30) | class SchedulerType(Enum): function get_constant_schedule (line 40) | def get_constant_schedule(optimizer: Optimizer, last_epoch: int = -1): function get_constant_schedule_with_warmup (line 56) | def get_constant_schedule_with_warmup(optimizer: Optimizer, num_warmup_s... function get_piecewise_constant_schedule (line 81) | def get_piecewise_constant_schedule(optimizer: Optimizer, step_rules: st... function get_linear_schedule_with_warmup (line 123) | def get_linear_schedule_with_warmup(optimizer, num_warmup_steps, num_tra... function get_cosine_schedule_with_warmup (line 152) | def get_cosine_schedule_with_warmup( function get_cosine_with_hard_restarts_schedule_with_warmup (line 186) | def get_cosine_with_hard_restarts_schedule_with_warmup( function get_polynomial_decay_schedule_with_warmup (line 221) | def get_polynomial_decay_schedule_with_warmup( function get_scheduler (line 282) | def get_scheduler( FILE: 4DoF/diffusers/pipelines/alt_diffusion/__init__.py class AltDiffusionPipelineOutput (line 13) | class AltDiffusionPipelineOutput(BaseOutput): FILE: 4DoF/diffusers/pipelines/alt_diffusion/modeling_roberta_series.py class TransformationModelOutput (line 11) | class TransformationModelOutput(ModelOutput): class RobertaSeriesConfig (line 39) | class RobertaSeriesConfig(XLMRobertaConfig): method __init__ (line 40) | def __init__( class RobertaSeriesModelWithTransformation (line 58) | class RobertaSeriesModelWithTransformation(RobertaPreTrainedModel): method __init__ (line 64) | def __init__(self, config): method forward (line 74) | def forward( FILE: 4DoF/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py function rescale_noise_cfg (line 55) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0): class AltDiffusionPipeline (line 70) | class AltDiffusionPipeline(DiffusionPipeline, TextualInversionLoaderMixi... method __init__ (line 107) | def __init__( method enable_vae_slicing (line 197) | def enable_vae_slicing(self): method disable_vae_slicing (line 206) | def disable_vae_slicing(self): method enable_vae_tiling (line 213) | def enable_vae_tiling(self): method disable_vae_tiling (line 222) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 229) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 254) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 283) | def _execution_device(self): method _encode_prompt (line 300) | def _encode_prompt( method run_safety_checker (line 454) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 468) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 483) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 500) | def check_inputs( method prepare_latents (line 547) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 566) | def __call__( FILE: 4DoF/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py function preprocess (line 71) | def preprocess(image): class AltDiffusionImg2ImgPipeline (line 98) | class AltDiffusionImg2ImgPipeline( method __init__ (line 137) | def __init__( method enable_sequential_cpu_offload (line 227) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 252) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 281) | def _execution_device(self): method _encode_prompt (line 298) | def _encode_prompt( method run_safety_checker (line 452) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 466) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 481) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 498) | def check_inputs( method get_timesteps (line 538) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 547) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 606) | def __call__( FILE: 4DoF/diffusers/pipelines/audio_diffusion/mel.py class Mel (line 37) | class Mel(ConfigMixin, SchedulerMixin): method __init__ (line 52) | def __init__( method set_resolution (line 73) | def set_resolution(self, x_res: int, y_res: int): method load_audio (line 85) | def load_audio(self, audio_file: str = None, raw_audio: np.ndarray = N... method get_number_of_slices (line 101) | def get_number_of_slices(self) -> int: method get_audio_slice (line 109) | def get_audio_slice(self, slice: int = 0) -> np.ndarray: method get_sample_rate (line 120) | def get_sample_rate(self) -> int: method audio_slice_to_image (line 128) | def audio_slice_to_image(self, slice: int) -> Image.Image: method image_to_audio (line 145) | def image_to_audio(self, image: Image.Image) -> np.ndarray: FILE: 4DoF/diffusers/pipelines/audio_diffusion/pipeline_audio_diffusion.py class AudioDiffusionPipeline (line 30) | class AudioDiffusionPipeline(DiffusionPipeline): method __init__ (line 44) | def __init__( method get_default_steps (line 54) | def get_default_steps(self) -> int: method __call__ (line 63) | def __call__( method encode (line 199) | def encode(self, images: List[Image.Image], steps: int = 50) -> np.nda... method slerp (line 236) | def slerp(x0: torch.Tensor, x1: torch.Tensor, alpha: float) -> torch.T... FILE: 4DoF/diffusers/pipelines/audioldm/pipeline_audioldm.py class AudioLDMPipeline (line 46) | class AudioLDMPipeline(DiffusionPipeline): method __init__ (line 72) | def __init__( method enable_vae_slicing (line 94) | def enable_vae_slicing(self): method disable_vae_slicing (line 104) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 111) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 129) | def _execution_device(self): method _encode_prompt (line 146) | def _encode_prompt( method decode_latents (line 285) | def decode_latents(self, latents): method mel_spectrogram_to_waveform (line 290) | def mel_spectrogram_to_waveform(self, mel_spectrogram): method prepare_extra_step_kwargs (line 300) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 317) | def check_inputs( method prepare_latents (line 376) | def prepare_latents(self, batch_size, num_channels_latents, height, dt... method __call__ (line 400) | def __call__( FILE: 4DoF/diffusers/pipelines/consistency_models/pipeline_consistency_models.py class ConsistencyModelPipeline (line 51) | class ConsistencyModelPipeline(DiffusionPipeline): method __init__ (line 69) | def __init__(self, unet: UNet2DModel, scheduler: CMStochasticIterative... method enable_sequential_cpu_offload (line 79) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 104) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 134) | def _execution_device(self): method prepare_latents (line 151) | def prepare_latents(self, batch_size, num_channels, height, width, dty... method postprocess_image (line 169) | def postprocess_image(self, sample: torch.FloatTensor, output_type: st... method prepare_class_labels (line 189) | def prepare_class_labels(self, batch_size, device, class_labels=None): method check_inputs (line 205) | def check_inputs(self, num_inference_steps, timesteps, latents, batch_... method __call__ (line 230) | def __call__( FILE: 4DoF/diffusers/pipelines/controlnet/multicontrolnet.py class MultiControlNetModel (line 15) | class MultiControlNetModel(ModelMixin): method __init__ (line 28) | def __init__(self, controlnets: Union[List[ControlNetModel], Tuple[Con... method forward (line 32) | def forward( method save_pretrained (line 73) | def save_pretrained( method from_pretrained (line 116) | def from_pretrained(cls, pretrained_model_path: Optional[Union[str, os... FILE: 4DoF/diffusers/pipelines/controlnet/pipeline_controlnet.py class StableDiffusionControlNetPipeline (line 93) | class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInvers... method __init__ (line 129) | def __init__( method enable_vae_slicing (line 180) | def enable_vae_slicing(self): method disable_vae_slicing (line 190) | def disable_vae_slicing(self): method enable_vae_tiling (line 198) | def enable_vae_tiling(self): method disable_vae_tiling (line 208) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 215) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 236) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 266) | def _execution_device(self): method _encode_prompt (line 284) | def _encode_prompt( method run_safety_checker (line 439) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 454) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 468) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 485) | def check_inputs( method check_image (line 619) | def check_image(self, image, prompt, prompt_embeds): method prepare_image (line 656) | def prepare_image( method prepare_latents (line 687) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 706) | def __call__( FILE: 4DoF/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py function prepare_image (line 95) | def prepare_image(image): class StableDiffusionControlNetImg2ImgPipeline (line 119) | class StableDiffusionControlNetImg2ImgPipeline(DiffusionPipeline, Textua... method __init__ (line 155) | def __init__( method enable_vae_slicing (line 206) | def enable_vae_slicing(self): method disable_vae_slicing (line 216) | def disable_vae_slicing(self): method enable_vae_tiling (line 224) | def enable_vae_tiling(self): method disable_vae_tiling (line 234) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 241) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 262) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 292) | def _execution_device(self): method _encode_prompt (line 310) | def _encode_prompt( method run_safety_checker (line 465) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 480) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 494) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 511) | def check_inputs( method check_image (line 646) | def check_image(self, image, prompt, prompt_embeds): method prepare_control_image (line 684) | def prepare_control_image( method get_timesteps (line 715) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 725) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 784) | def __call__( FILE: 4DoF/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py function prepare_mask_and_masked_image (line 108) | def prepare_mask_and_masked_image(image, mask, height, width, return_ima... class StableDiffusionControlNetInpaintPipeline (line 225) | class StableDiffusionControlNetInpaintPipeline(DiffusionPipeline, Textua... method __init__ (line 272) | def __init__( method enable_vae_slicing (line 323) | def enable_vae_slicing(self): method disable_vae_slicing (line 333) | def disable_vae_slicing(self): method enable_vae_tiling (line 341) | def enable_vae_tiling(self): method disable_vae_tiling (line 351) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 358) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 379) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 409) | def _execution_device(self): method _encode_prompt (line 427) | def _encode_prompt( method run_safety_checker (line 582) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 597) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 611) | def prepare_extra_step_kwargs(self, generator, eta): method get_timesteps (line 629) | def get_timesteps(self, num_inference_steps, strength, device): method check_inputs (line 638) | def check_inputs( method check_image (line 778) | def check_image(self, image, prompt, prompt_embeds): method prepare_control_image (line 816) | def prepare_control_image( method prepare_latents (line 847) | def prepare_latents( method _default_height_width (line 900) | def _default_height_width(self, height, width, image): method prepare_mask_latents (line 926) | def prepare_mask_latents( method _encode_vae_image (line 968) | def _encode_vae_image(self, image: torch.Tensor, generator: torch.Gene... method __call__ (line 984) | def __call__( FILE: 4DoF/diffusers/pipelines/controlnet/pipeline_flax_controlnet.py class FlaxStableDiffusionControlNetPipeline (line 119) | class FlaxStableDiffusionControlNetPipeline(FlaxDiffusionPipeline): method __init__ (line 150) | def __init__( method prepare_text_inputs (line 189) | def prepare_text_inputs(self, prompt: Union[str, List[str]]): method prepare_image_inputs (line 203) | def prepare_image_inputs(self, image: Union[Image.Image, List[Image.Im... method _get_has_nsfw_concepts (line 214) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 218) | def _run_safety_checker(self, images, safety_model_params, jit=False): method _generate (line 248) | def _generate( method __call__ (line 358) | def __call__( function _p_generate (line 493) | def _p_generate( function _p_get_has_nsfw_concepts (line 519) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 523) | def unshard(x: jnp.ndarray): function preprocess (line 530) | def preprocess(image, dtype): FILE: 4DoF/diffusers/pipelines/dance_diffusion/pipeline_dance_diffusion.py class DanceDiffusionPipeline (line 27) | class DanceDiffusionPipeline(DiffusionPipeline): method __init__ (line 39) | def __init__(self, unet, scheduler): method __call__ (line 44) | def __call__( FILE: 4DoF/diffusers/pipelines/ddim/pipeline_ddim.py class DDIMPipeline (line 24) | class DDIMPipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet, scheduler): method __call__ (line 45) | def __call__( FILE: 4DoF/diffusers/pipelines/ddpm/pipeline_ddpm.py class DDPMPipeline (line 24) | class DDPMPipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet, scheduler): method __call__ (line 41) | def __call__( FILE: 4DoF/diffusers/pipelines/deepfloyd_if/__init__.py class IFPipelineOutput (line 21) | class IFPipelineOutput(BaseOutput): FILE: 4DoF/diffusers/pipelines/deepfloyd_if/pipeline_if.py class IFPipeline (line 89) | class IFPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 107) | def __init__( method enable_sequential_cpu_offload (line 147) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 171) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 212) | def remove_all_hooks(self): method _execution_device (line 228) | def _execution_device(self): method encode_prompt (line 246) | def encode_prompt( method run_safety_checker (line 397) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 414) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 431) | def check_inputs( method prepare_intermediate_images (line 473) | def prepare_intermediate_images(self, batch_size, num_channels, height... method _text_preprocessing (line 487) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 511) | def _clean_caption(self, caption): method __call__ (line 627) | def __call__( FILE: 4DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py function resize (line 41) | def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image: class IFImg2ImgPipeline (line 113) | class IFImg2ImgPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 131) | def __init__( method enable_sequential_cpu_offload (line 172) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 197) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 239) | def remove_all_hooks(self): method _execution_device (line 255) | def _execution_device(self): method encode_prompt (line 274) | def encode_prompt( method run_safety_checker (line 426) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 443) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 460) | def check_inputs( method _text_preprocessing (line 534) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 559) | def _clean_caption(self, caption): method preprocess_image (line 673) | def preprocess_image(self, image: PIL.Image.Image) -> torch.Tensor: method get_timesteps (line 709) | def get_timesteps(self, num_inference_steps, strength): method prepare_intermediate_images (line 718) | def prepare_intermediate_images( method __call__ (line 742) | def __call__( FILE: 4DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py function resize (line 44) | def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image: class IFImg2ImgSuperResolutionPipeline (line 116) | class IFImg2ImgSuperResolutionPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 135) | def __init__( method enable_sequential_cpu_offload (line 183) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 208) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 250) | def remove_all_hooks(self): method _text_preprocessing (line 265) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 290) | def _clean_caption(self, caption): method _execution_device (line 406) | def _execution_device(self): method encode_prompt (line 425) | def encode_prompt( method run_safety_checker (line 577) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 594) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 611) | def check_inputs( method preprocess_original_image (line 721) | def preprocess_original_image(self, image: PIL.Image.Image) -> torch.T... method preprocess_image (line 758) | def preprocess_image(self, image: PIL.Image.Image, num_images_per_prom... method get_timesteps (line 790) | def get_timesteps(self, num_inference_steps, strength): method prepare_intermediate_images (line 800) | def prepare_intermediate_images( method __call__ (line 824) | def __call__( FILE: 4DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py function resize (line 42) | def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image: class IFInpaintingPipeline (line 116) | class IFInpaintingPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 134) | def __init__( method enable_sequential_cpu_offload (line 175) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 200) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 242) | def remove_all_hooks(self): method _execution_device (line 258) | def _execution_device(self): method encode_prompt (line 277) | def encode_prompt( method run_safety_checker (line 429) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 446) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 463) | def check_inputs( method _text_preprocessing (line 573) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 598) | def _clean_caption(self, caption): method preprocess_image (line 713) | def preprocess_image(self, image: PIL.Image.Image) -> torch.Tensor: method preprocess_mask_image (line 749) | def preprocess_mask_image(self, mask_image) -> torch.Tensor: method get_timesteps (line 799) | def get_timesteps(self, num_inference_steps, strength): method prepare_intermediate_images (line 808) | def prepare_intermediate_images( method __call__ (line 834) | def __call__( FILE: 4DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py function resize (line 44) | def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image: class IFInpaintingSuperResolutionPipeline (line 118) | class IFInpaintingSuperResolutionPipeline(DiffusionPipeline, LoraLoaderM... method __init__ (line 137) | def __init__( method enable_sequential_cpu_offload (line 185) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 210) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 252) | def remove_all_hooks(self): method _text_preprocessing (line 267) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 292) | def _clean_caption(self, caption): method _execution_device (line 408) | def _execution_device(self): method encode_prompt (line 427) | def encode_prompt( method run_safety_checker (line 579) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 596) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 613) | def check_inputs( method preprocess_original_image (line 757) | def preprocess_original_image(self, image: PIL.Image.Image) -> torch.T... method preprocess_image (line 794) | def preprocess_image(self, image: PIL.Image.Image, num_images_per_prom... method preprocess_mask_image (line 826) | def preprocess_mask_image(self, mask_image) -> torch.Tensor: method get_timesteps (line 876) | def get_timesteps(self, num_inference_steps, strength): method prepare_intermediate_images (line 886) | def prepare_intermediate_images( method __call__ (line 912) | def __call__( FILE: 4DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py class IFSuperResolutionPipeline (line 74) | class IFSuperResolutionPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 93) | def __init__( method enable_sequential_cpu_offload (line 141) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 166) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 208) | def remove_all_hooks(self): method _text_preprocessing (line 223) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 248) | def _clean_caption(self, caption): method _execution_device (line 364) | def _execution_device(self): method encode_prompt (line 383) | def encode_prompt( method run_safety_checker (line 535) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 552) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 569) | def check_inputs( method prepare_intermediate_images (line 649) | def prepare_intermediate_images(self, batch_size, num_channels, height... method preprocess_image (line 663) | def preprocess_image(self, image, num_images_per_prompt, device): method __call__ (line 696) | def __call__( FILE: 4DoF/diffusers/pipelines/deepfloyd_if/safety_checker.py class IFSafetyChecker (line 12) | class IFSafetyChecker(PreTrainedModel): method __init__ (line 17) | def __init__(self, config: CLIPConfig): method forward (line 26) | def forward(self, clip_input, images, p_threshold=0.5, w_threshold=0.5): FILE: 4DoF/diffusers/pipelines/deepfloyd_if/watermark.py class IFWatermarker (line 12) | class IFWatermarker(ModelMixin, ConfigMixin): method __init__ (line 13) | def __init__(self): method apply_watermark (line 19) | def apply_watermark(self, images: List[PIL.Image.Image], sample_size=N... FILE: 4DoF/diffusers/pipelines/dit/pipeline_dit.py class DiTPipeline (line 31) | class DiTPipeline(DiffusionPipeline): method __init__ (line 45) | def __init__( method get_label_ids (line 63) | def get_label_ids(self, label: Union[str, List[str]]) -> List[int]: method __call__ (line 87) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky/pipeline_kandinsky.py function get_new_h_w (line 69) | def get_new_h_w(h, w, scale_factor=8): class KandinskyPipeline (line 79) | class KandinskyPipeline(DiffusionPipeline): method __init__ (line 99) | def __init__( method prepare_latents (line 119) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 130) | def _encode_prompt( method enable_sequential_cpu_offload (line 232) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 254) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 284) | def _execution_device(self): method __call__ (line 303) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky/pipeline_kandinsky_img2img.py function get_new_h_w (line 82) | def get_new_h_w(h, w, scale_factor=8): function prepare_image (line 92) | def prepare_image(pil_image, w=512, h=512): class KandinskyImg2ImgPipeline (line 101) | class KandinskyImg2ImgPipeline(DiffusionPipeline): method __init__ (line 121) | def __init__( method get_timesteps (line 140) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 149) | def prepare_latents(self, latents, latent_timestep, shape, dtype, devi... method _encode_prompt (line 165) | def _encode_prompt( method enable_sequential_cpu_offload (line 267) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 289) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 319) | def _execution_device(self): method add_noise (line 337) | def add_noise( method __call__ (line 365) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky/pipeline_kandinsky_inpaint.py function get_new_h_w (line 89) | def get_new_h_w(h, w, scale_factor=8): function prepare_mask (line 99) | def prepare_mask(masks): function prepare_mask_and_masked_image (line 123) | def prepare_mask_and_masked_image(image, mask, height, width): class KandinskyInpaintPipeline (line 239) | class KandinskyInpaintPipeline(DiffusionPipeline): method __init__ (line 259) | def __init__( method prepare_latents (line 279) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 290) | def _encode_prompt( method enable_sequential_cpu_offload (line 392) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 414) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 444) | def _execution_device(self): method __call__ (line 463) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky/pipeline_kandinsky_prior.py class KandinskyPriorPipelineOutput (line 114) | class KandinskyPriorPipelineOutput(BaseOutput): class KandinskyPriorPipeline (line 129) | class KandinskyPriorPipeline(DiffusionPipeline): method __init__ (line 150) | def __init__( method interpolate (line 172) | def interpolate( method prepare_latents (line 278) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method get_zero_embed (line 289) | def get_zero_embed(self, batch_size=1, device=None): method enable_sequential_cpu_offload (line 298) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 320) | def _execution_device(self): method _encode_prompt (line 337) | def _encode_prompt( method __call__ (line 436) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky/text_encoder.py class MCLIPConfig (line 5) | class MCLIPConfig(XLMRobertaConfig): method __init__ (line 8) | def __init__(self, transformerDimSize=1024, imageDimSize=768, **kwargs): class MultilingualCLIP (line 14) | class MultilingualCLIP(PreTrainedModel): method __init__ (line 17) | def __init__(self, config, *args, **kwargs): method forward (line 24) | def forward(self, input_ids, attention_mask): FILE: 4DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py function downscale_height_and_width (line 60) | def downscale_height_and_width(height, width, scale_factor=8): class KandinskyV22Pipeline (line 70) | class KandinskyV22Pipeline(DiffusionPipeline): method __init__ (line 86) | def __init__( method prepare_latents (line 102) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 113) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 134) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 161) | def _execution_device(self): method __call__ (line 180) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet.py function downscale_height_and_width (line 100) | def downscale_height_and_width(height, width, scale_factor=8): class KandinskyV22ControlnetPipeline (line 110) | class KandinskyV22ControlnetPipeline(DiffusionPipeline): method __init__ (line 126) | def __init__( method prepare_latents (line 142) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 154) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 176) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 203) | def _execution_device(self): method __call__ (line 222) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet_img2img.py function downscale_height_and_width (line 104) | def downscale_height_and_width(height, width, scale_factor=8): function prepare_image (line 115) | def prepare_image(pil_image, w=512, h=512): class KandinskyV22ControlnetImg2ImgPipeline (line 124) | class KandinskyV22ControlnetImg2ImgPipeline(DiffusionPipeline): method __init__ (line 140) | def __init__( method get_timesteps (line 156) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 166) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method enable_sequential_cpu_offload (line 209) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 231) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 258) | def _execution_device(self): method __call__ (line 277) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py function downscale_height_and_width (line 78) | def downscale_height_and_width(height, width, scale_factor=8): function prepare_image (line 89) | def prepare_image(pil_image, w=512, h=512): class KandinskyV22Img2ImgPipeline (line 98) | class KandinskyV22Img2ImgPipeline(DiffusionPipeline): method __init__ (line 114) | def __init__( method get_timesteps (line 130) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 139) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method enable_sequential_cpu_offload (line 182) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 204) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 231) | def _execution_device(self): method __call__ (line 250) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py function downscale_height_and_width (line 85) | def downscale_height_and_width(height, width, scale_factor=8): function prepare_mask (line 96) | def prepare_mask(masks): function prepare_mask_and_masked_image (line 121) | def prepare_mask_and_masked_image(image, mask, height, width): class KandinskyV22InpaintPipeline (line 237) | class KandinskyV22InpaintPipeline(DiffusionPipeline): method __init__ (line 253) | def __init__( method prepare_latents (line 269) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 281) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 303) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 330) | def _execution_device(self): method __call__ (line 349) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py class KandinskyV22PriorPipeline (line 85) | class KandinskyV22PriorPipeline(DiffusionPipeline): method __init__ (line 108) | def __init__( method interpolate (line 130) | def interpolate( method prepare_latents (line 236) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method get_zero_embed (line 248) | def get_zero_embed(self, batch_size=1, device=None): method enable_sequential_cpu_offload (line 258) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 281) | def _execution_device(self): method _encode_prompt (line 299) | def _encode_prompt( method __call__ (line 398) | def __call__( FILE: 4DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior_emb2emb.py class KandinskyV22PriorEmb2EmbPipeline (line 103) | class KandinskyV22PriorEmb2EmbPipeline(DiffusionPipeline): method __init__ (line 124) | def __init__( method get_timesteps (line 144) | def get_timesteps(self, num_inference_steps, strength, device): method interpolate (line 155) | def interpolate( method _encode_image (line 243) | def _encode_image( method prepare_latents (line 260) | def prepare_latents(self, emb, timestep, batch_size, num_images_per_pr... method get_zero_embed (line 287) | def get_zero_embed(self, batch_size=1, device=None): method enable_sequential_cpu_offload (line 297) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 320) | def _execution_device(self): method _encode_prompt (line 338) | def _encode_prompt( method __call__ (line 437) | def __call__( FILE: 4DoF/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py class LDMTextToImagePipeline (line 32) | class LDMTextToImagePipeline(DiffusionPipeline): method __init__ (line 51) | def __init__( method __call__ (line 64) | def __call__( class LDMBertConfig (line 220) | class LDMBertConfig(PretrainedConfig): method __init__ (line 225) | def __init__( function _expand_mask (line 267) | def _expand_mask(mask: torch.Tensor, dtype: torch.dtype, tgt_len: Option... class LDMBertAttention (line 282) | class LDMBertAttention(nn.Module): method __init__ (line 285) | def __init__( method _shape (line 309) | def _shape(self, tensor: torch.Tensor, seq_len: int, bsz: int): method forward (line 312) | def forward( class LDMBertEncoderLayer (line 426) | class LDMBertEncoderLayer(nn.Module): method __init__ (line 427) | def __init__(self, config: LDMBertConfig): method forward (line 444) | def forward( class LDMBertPreTrainedModel (line 496) | class LDMBertPreTrainedModel(PreTrainedModel): method _init_weights (line 502) | def _init_weights(self, module): method _set_gradient_checkpointing (line 513) | def _set_gradient_checkpointing(self, module, value=False): method dummy_inputs (line 518) | def dummy_inputs(self): class LDMBertEncoder (line 528) | class LDMBertEncoder(LDMBertPreTrainedModel): method __init__ (line 538) | def __init__(self, config: LDMBertConfig): method get_input_embeddings (line 556) | def get_input_embeddings(self): method set_input_embeddings (line 559) | def set_input_embeddings(self, value): method forward (line 562) | def forward( class LDMBertModel (line 695) | class LDMBertModel(LDMBertPreTrainedModel): method __init__ (line 698) | def __init__(self, config: LDMBertConfig): method forward (line 703) | def forward( FILE: 4DoF/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py function preprocess (line 22) | def preprocess(image): class LDMSuperResolutionPipeline (line 32) | class LDMSuperResolutionPipeline(DiffusionPipeline): method __init__ (line 49) | def __init__( method __call__ (line 66) | def __call__( FILE: 4DoF/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py class LDMPipeline (line 26) | class LDMPipeline(DiffusionPipeline): method __init__ (line 39) | def __init__(self, vqvae: VQModel, unet: UNet2DModel, scheduler: DDIMS... method __call__ (line 44) | def __call__( FILE: 4DoF/diffusers/pipelines/onnx_utils.py class OnnxRuntimeModel (line 51) | class OnnxRuntimeModel: method __init__ (line 52) | def __init__(self, model=None, **kwargs): method __call__ (line 58) | def __call__(self, **kwargs): method load_model (line 63) | def load_model(path: Union[str, Path], provider=None, sess_options=None): method _save_pretrained (line 79) | def _save_pretrained(self, save_directory: Union[str, Path], file_name... method save_pretrained (line 110) | def save_pretrained( method _from_pretrained (line 133) | def _from_pretrained( method from_pretrained (line 193) | def from_pretrained( FILE: 4DoF/diffusers/pipelines/paint_by_example/image_encoder.py class PaintByExampleImageEncoder (line 25) | class PaintByExampleImageEncoder(CLIPPreTrainedModel): method __init__ (line 26) | def __init__(self, config, proj_size=768): method forward (line 38) | def forward(self, pixel_values, return_uncond_vector=False): class PaintByExampleMapper (line 50) | class PaintByExampleMapper(nn.Module): method __init__ (line 51) | def __init__(self, config): method forward (line 63) | def forward(self, hidden_states): FILE: 4DoF/diffusers/pipelines/paint_by_example/pipeline_paint_by_example.py function prepare_mask_and_masked_image (line 39) | def prepare_mask_and_masked_image(image, mask): class PaintByExamplePipeline (line 139) | class PaintByExamplePipeline(DiffusionPipeline): method __init__ (line 168) | def __init__( method enable_sequential_cpu_offload (line 192) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 213) | def _execution_device(self): method run_safety_checker (line 231) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 246) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 264) | def decode_latents(self, latents): method check_inputs (line 278) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 301) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_mask_latents (line 319) | def prepare_mask_latents( method _encode_vae_image (line 361) | def _encode_vae_image(self, image: torch.Tensor, generator: torch.Gene... method _encode_image (line 375) | def _encode_image(self, image, device, num_images_per_prompt, do_class... method __call__ (line 401) | def __call__( FILE: 4DoF/diffusers/pipelines/pipeline_flax_utils.py function import_flax_or_no_model (line 66) | def import_flax_or_no_model(module, class_name): class FlaxImagePipelineOutput (line 80) | class FlaxImagePipelineOutput(BaseOutput): class FlaxDiffusionPipeline (line 93) | class FlaxDiffusionPipeline(ConfigMixin): method register_modules (line 110) | def register_modules(self, **kwargs): method save_pretrained (line 143) | def save_pretrained(self, save_directory: Union[str, os.PathLike], par... method from_pretrained (line 194) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method _get_signature_keys (line 496) | def _get_signature_keys(obj): method components (line 504) | def components(self) -> Dict[str, Any]: method numpy_to_pil (line 541) | def numpy_to_pil(images): method progress_bar (line 557) | def progress_bar(self, iterable): method set_progress_bar_config (line 567) | def set_progress_bar_config(self, **kwargs): FILE: 4DoF/diffusers/pipelines/pipeline_utils.py class ImagePipelineOutput (line 112) | class ImagePipelineOutput(BaseOutput): class AudioPipelineOutput (line 126) | class AudioPipelineOutput(BaseOutput): function is_safetensors_compatible (line 138) | def is_safetensors_compatible(filenames, variant=None, passed_components... function variant_compatible_siblings (line 187) | def variant_compatible_siblings(filenames, variant=None) -> Union[List[o... function warn_deprecated_model_variant (line 254) | def warn_deprecated_model_variant(pretrained_model_name_or_path, use_aut... function maybe_raise_or_warn (line 276) | def maybe_raise_or_warn( function get_class_obj_and_candidates (line 308) | def get_class_obj_and_candidates(library_name, class_name, importable_cl... function _get_pipeline_class (line 325) | def _get_pipeline_class(class_obj, config, custom_pipeline=None, cache_d... function load_sub_model (line 346) | def load_sub_model( class DiffusionPipeline (line 458) | class DiffusionPipeline(ConfigMixin): method register_modules (line 478) | def register_modules(self, **kwargs): method __setattr__ (line 521) | def __setattr__(self, name: str, value: Any): method save_pretrained (line 536) | def save_pretrained( method to (line 623) | def to( method device (line 699) | def device(self) -> torch.device: method from_pretrained (line 714) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method download (line 1079) | def download(cls, pretrained_model_name, **kwargs) -> Union[str, os.Pa... method _get_signature_keys (line 1340) | def _get_signature_keys(obj): method components (line 1348) | def components(self) -> Dict[str, Any]: method numpy_to_pil (line 1384) | def numpy_to_pil(images): method progress_bar (line 1390) | def progress_bar(self, iterable=None, total=None): method set_progress_bar_config (line 1405) | def set_progress_bar_config(self, **kwargs): method enable_xformers_memory_efficient_attention (line 1408) | def enable_xformers_memory_efficient_attention(self, attention_op: Opt... method disable_xformers_memory_efficient_attention (line 1444) | def disable_xformers_memory_efficient_attention(self): method set_use_memory_efficient_attention_xformers (line 1450) | def set_use_memory_efficient_attention_xformers( method enable_attention_slicing (line 1470) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 1486) | def disable_attention_slicing(self): method set_attention_slice (line 1494) | def set_attention_slice(self, slice_size: Optional[int]): FILE: 4DoF/diffusers/pipelines/pndm/pipeline_pndm.py class PNDMPipeline (line 26) | class PNDMPipeline(DiffusionPipeline): method __init__ (line 40) | def __init__(self, unet: UNet2DModel, scheduler: PNDMScheduler): method __call__ (line 48) | def __call__( FILE: 4DoF/diffusers/pipelines/repaint/pipeline_repaint.py function _preprocess_image (line 33) | def _preprocess_image(image: Union[List, PIL.Image.Image, torch.Tensor]): function _preprocess_mask (line 59) | def _preprocess_mask(mask: Union[List, PIL.Image.Image, torch.Tensor]): class RePaintPipeline (line 79) | class RePaintPipeline(DiffusionPipeline): method __init__ (line 83) | def __init__(self, unet, scheduler): method __call__ (line 88) | def __call__( FILE: 4DoF/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py class ScoreSdeVePipeline (line 25) | class ScoreSdeVePipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet: UNet2DModel, scheduler: ScoreSdeVeScheduler): method __call__ (line 41) | def __call__( FILE: 4DoF/diffusers/pipelines/semantic_stable_diffusion/__init__.py class SemanticStableDiffusionPipelineOutput (line 13) | class SemanticStableDiffusionPipelineOutput(BaseOutput): FILE: 4DoF/diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py class SemanticStableDiffusionPipeline (line 63) | class SemanticStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 95) | def __init__( method run_safety_checker (line 138) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 153) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 167) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 185) | def check_inputs( method prepare_latents (line 233) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 251) | def __call__( FILE: 4DoF/diffusers/pipelines/shap_e/camera.py class DifferentiableProjectiveCamera (line 23) | class DifferentiableProjectiveCamera: method __post_init__ (line 38) | def __post_init__(self): method resolution (line 43) | def resolution(self): method fov (line 46) | def fov(self): method get_image_coords (line 49) | def get_image_coords(self) -> torch.Tensor: method camera_rays (line 64) | def camera_rays(self): method get_camera_rays (line 76) | def get_camera_rays(self, coords: torch.Tensor) -> torch.Tensor: method resize_image (line 105) | def resize_image(self, width: int, height: int) -> "DifferentiableProj... function create_pan_cameras (line 122) | def create_pan_cameras(size: int) -> DifferentiableProjectiveCamera: FILE: 4DoF/diffusers/pipelines/shap_e/pipeline_shap_e.py class ShapEPipelineOutput (line 69) | class ShapEPipelineOutput(BaseOutput): class ShapEPipeline (line 81) | class ShapEPipeline(DiffusionPipeline): method __init__ (line 103) | def __init__( method prepare_latents (line 122) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 133) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 151) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 180) | def _execution_device(self): method _encode_prompt (line 197) | def _encode_prompt( method __call__ (line 248) | def __call__( FILE: 4DoF/diffusers/pipelines/shap_e/pipeline_shap_e_img2img.py class ShapEPipelineOutput (line 69) | class ShapEPipelineOutput(BaseOutput): class ShapEImg2ImgPipeline (line 81) | class ShapEImg2ImgPipeline(DiffusionPipeline): method __init__ (line 103) | def __init__( method prepare_latents (line 122) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 133) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 152) | def _execution_device(self): method _encode_image (line 169) | def _encode_image( method __call__ (line 201) | def __call__( FILE: 4DoF/diffusers/pipelines/shap_e/renderer.py function sample_pmf (line 30) | def sample_pmf(pmf: torch.Tensor, n_samples: int) -> torch.Tensor: function posenc_nerf (line 53) | def posenc_nerf(x: torch.Tensor, min_deg: int = 0, max_deg: int = 15) ->... function encode_position (line 70) | def encode_position(position): function encode_direction (line 74) | def encode_direction(position, direction=None): function _sanitize_name (line 81) | def _sanitize_name(x: str) -> str: function integrate_samples (line 85) | def integrate_samples(volume_range, ts, density, channels): class VoidNeRFModel (line 119) | class VoidNeRFModel(nn.Module): method __init__ (line 124) | def __init__(self, background, channel_scale=255.0): method forward (line 130) | def forward(self, position): class VolumeRange (line 142) | class VolumeRange: method __post_init__ (line 147) | def __post_init__(self): method partition (line 150) | def partition(self, ts): class BoundingBoxVolume (line 174) | class BoundingBoxVolume(nn.Module): method __init__ (line 179) | def __init__( method intersect (line 205) | def intersect( class StratifiedRaySampler (line 258) | class StratifiedRaySampler(nn.Module): method __init__ (line 263) | def __init__(self, depth_mode: str = "linear"): method sample (line 271) | def sample( class ImportanceRaySampler (line 310) | class ImportanceRaySampler(nn.Module): method __init__ (line 315) | def __init__( method sample (line 338) | def sample(self, t0: torch.Tensor, t1: torch.Tensor, n_samples: int) -... class MLPNeRFModelOutput (line 372) | class MLPNeRFModelOutput(BaseOutput): class MLPNeRSTFModel (line 379) | class MLPNeRSTFModel(ModelMixin, ConfigMixin): method __init__ (line 381) | def __init__( method map_indices_to_keys (line 418) | def map_indices_to_keys(self, output): method forward (line 432) | def forward(self, *, position, direction, ts, nerf_level="coarse"): class ChannelsProj (line 471) | class ChannelsProj(nn.Module): method __init__ (line 472) | def __init__( method forward (line 486) | def forward(self, x: torch.Tensor) -> torch.Tensor: class ShapEParamsProjModel (line 497) | class ShapEParamsProjModel(ModelMixin, ConfigMixin): method __init__ (line 505) | def __init__( method forward (line 535) | def forward(self, x: torch.Tensor): class ShapERenderer (line 547) | class ShapERenderer(ModelMixin, ConfigMixin): method __init__ (line 549) | def __init__( method render_rays (line 588) | def render_rays(self, rays, sampler, n_samples, prev_model_out=None, r... method decode (line 667) | def decode( FILE: 4DoF/diffusers/pipelines/spectrogram_diffusion/continous_encoder.py class SpectrogramContEncoder (line 29) | class SpectrogramContEncoder(ModelMixin, ConfigMixin, ModuleUtilsMixin): method __init__ (line 31) | def __init__( method forward (line 71) | def forward(self, encoder_inputs, encoder_inputs_mask): FILE: 4DoF/diffusers/pipelines/spectrogram_diffusion/midi_utils.py class NoteRepresentationConfig (line 84) | class NoteRepresentationConfig: class NoteEventData (line 92) | class NoteEventData: class NoteEncodingState (line 101) | class NoteEncodingState: class EventRange (line 109) | class EventRange: class Event (line 116) | class Event: class Tokenizer (line 121) | class Tokenizer: method __init__ (line 122) | def __init__(self, regular_ids: int): method encode (line 127) | def encode(self, token_ids): class Codec (line 145) | class Codec: method __init__ (line 156) | def __init__(self, max_shift_steps: int, steps_per_second: float, even... method num_classes (line 172) | def num_classes(self) -> int: method is_shift_event_index (line 178) | def is_shift_event_index(self, index: int) -> bool: method max_shift_steps (line 182) | def max_shift_steps(self) -> int: method encode_event (line 185) | def encode_event(self, event: Event) -> int: method event_type_range (line 200) | def event_type_range(self, event_type: str) -> Tuple[int, int]: method decode_event_index (line 210) | def decode_event_index(self, index: int) -> Event: class ProgramGranularity (line 222) | class ProgramGranularity: function drop_programs (line 228) | def drop_programs(tokens, codec: Codec): function programs_to_midi_classes (line 234) | def programs_to_midi_classes(tokens, codec): function frame (line 254) | def frame(signal, frame_length, frame_step, pad_end=False, pad_value=0, ... function program_to_slakh_program (line 272) | def program_to_slakh_program(program): function audio_to_frames (line 279) | def audio_to_frames( function note_sequence_to_onsets_and_offsets_and_programs (line 302) | def note_sequence_to_onsets_and_offsets_and_programs( function num_velocity_bins_from_codec (line 332) | def num_velocity_bins_from_codec(codec: Codec): function segment (line 339) | def segment(a, n): function velocity_to_bin (line 343) | def velocity_to_bin(velocity, num_velocity_bins): function note_event_data_to_events (line 350) | def note_event_data_to_events( function note_encoding_state_to_events (line 382) | def note_encoding_state_to_events(state: NoteEncodingState) -> Sequence[... function encode_and_index_events (line 392) | def encode_and_index_events( function extract_sequence_with_indices (line 498) | def extract_sequence_with_indices(features, state_events_end_token=None,... function map_midi_programs (line 524) | def map_midi_programs( function run_length_encode_shifts_fn (line 534) | def run_length_encode_shifts_fn( function note_representation_processor_chain (line 604) | def note_representation_processor_chain(features, codec: Codec, note_rep... class MidiProcessor (line 619) | class MidiProcessor: method __init__ (line 620) | def __init__(self): method __call__ (line 635) | def __call__(self, midi: Union[bytes, os.PathLike, str]): FILE: 4DoF/diffusers/pipelines/spectrogram_diffusion/notes_encoder.py class SpectrogramNotesEncoder (line 25) | class SpectrogramNotesEncoder(ModelMixin, ConfigMixin, ModuleUtilsMixin): method __init__ (line 27) | def __init__( method forward (line 69) | def forward(self, encoder_input_tokens, encoder_inputs_mask): FILE: 4DoF/diffusers/pipelines/spectrogram_diffusion/pipeline_spectrogram_diffusion.py class SpectrogramDiffusionPipeline (line 40) | class SpectrogramDiffusionPipeline(DiffusionPipeline): method __init__ (line 43) | def __init__( method scale_features (line 66) | def scale_features(self, features, output_range=(-1.0, 1.0), clip=False): method scale_to_features (line 76) | def scale_to_features(self, outputs, input_range=(-1.0, 1.0), clip=Fal... method encode (line 85) | def encode(self, input_tokens, continuous_inputs, continuous_mask): method decode (line 97) | def decode(self, encodings_and_masks, input_tokens, noise_time): method __call__ (line 113) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/__init__.py class StableDiffusionPipelineOutput (line 22) | class StableDiffusionPipelineOutput(BaseOutput): class FlaxStableDiffusionPipelineOutput (line 117) | class FlaxStableDiffusionPipelineOutput(BaseOutput): FILE: 4DoF/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py function shave_segments (line 68) | def shave_segments(path, n_shave_prefix_segments=1): function renew_resnet_paths (line 78) | def renew_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_vae_resnet_paths (line 100) | def renew_vae_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_attention_paths (line 116) | def renew_attention_paths(old_list, n_shave_prefix_segments=0): function renew_vae_attention_paths (line 137) | def renew_vae_attention_paths(old_list, n_shave_prefix_segments=0): function assign_to_checkpoint (line 167) | def assign_to_checkpoint( function conv_attn_to_linear (line 222) | def conv_attn_to_linear(checkpoint): function create_unet_diffusers_config (line 234) | def create_unet_diffusers_config(original_config, image_size: int, contr... function create_vae_diffusers_config (line 334) | def create_vae_diffusers_config(original_config, image_size: int): function create_diffusers_schedular (line 358) | def create_diffusers_schedular(original_config): function create_ldm_bert_config (line 368) | def create_ldm_bert_config(original_config): function convert_ldm_unet_checkpoint (line 378) | def convert_ldm_unet_checkpoint( function convert_ldm_vae_checkpoint (line 620) | def convert_ldm_vae_checkpoint(checkpoint, config): function convert_ldm_bert_checkpoint (line 727) | def convert_ldm_bert_checkpoint(checkpoint, config): function convert_ldm_clip_checkpoint (line 777) | def convert_ldm_clip_checkpoint(checkpoint, local_files_only=False, text... function convert_paint_by_example_checkpoint (line 827) | def convert_paint_by_example_checkpoint(checkpoint): function convert_open_clip_checkpoint (line 894) | def convert_open_clip_checkpoint( function stable_unclip_image_encoder (line 959) | def stable_unclip_image_encoder(original_config): function stable_unclip_image_noising_components (line 992) | def stable_unclip_image_noising_components( function convert_controlnet_checkpoint (line 1037) | def convert_controlnet_checkpoint( function download_from_original_stable_diffusion_ckpt (line 1081) | def download_from_original_stable_diffusion_ckpt( function download_controlnet_from_original_ckpt (line 1563) | def download_controlnet_from_original_ckpt( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py function preprocess (line 42) | def preprocess(image): function posterior_sample (line 68) | def posterior_sample(scheduler, latents, timestep, clean_latents, genera... function compute_noise (line 95) | def compute_noise(scheduler, prev_latents, latents, timestep, noise_pred... class CycleDiffusionPipeline (line 129) | class CycleDiffusionPipeline(DiffusionPipeline, TextualInversionLoaderMi... method __init__ (line 158) | def __init__( method enable_sequential_cpu_offload (line 235) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 261) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 291) | def _execution_device(self): method _encode_prompt (line 309) | def _encode_prompt( method check_inputs (line 464) | def check_inputs( method prepare_extra_step_kwargs (line 505) | def prepare_extra_step_kwargs(self, generator, eta): method run_safety_checker (line 523) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 538) | def decode_latents(self, latents): method get_timesteps (line 552) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 561) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 616) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py class FlaxStableDiffusionPipeline (line 81) | class FlaxStableDiffusionPipeline(FlaxDiffusionPipeline): method __init__ (line 110) | def __init__( method prepare_inputs (line 168) | def prepare_inputs(self, prompt: Union[str, List[str]]): method _get_has_nsfw_concepts (line 181) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 185) | def _run_safety_checker(self, images, safety_model_params, jit=False): method _generate (line 215) | def _generate( method __call__ (line 312) | def __call__( function _p_generate (line 436) | def _p_generate( function _p_get_has_nsfw_concepts (line 462) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 466) | def unshard(x: jnp.ndarray): FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py class FlaxStableDiffusionImg2ImgPipeline (line 105) | class FlaxStableDiffusionImg2ImgPipeline(FlaxDiffusionPipeline): method __init__ (line 134) | def __init__( method prepare_inputs (line 171) | def prepare_inputs(self, prompt: Union[str, List[str]], image: Union[I... method _get_has_nsfw_concepts (line 192) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 196) | def _run_safety_checker(self, images, safety_model_params, jit=False): method get_timestep_start (line 226) | def get_timestep_start(self, num_inference_steps, strength): method _generate (line 234) | def _generate( method __call__ (line 339) | def __call__( function _p_generate (line 480) | def _p_generate( function _p_get_has_nsfw_concepts (line 510) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 514) | def unshard(x: jnp.ndarray): function preprocess (line 521) | def preprocess(image, dtype): FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_inpaint.py class FlaxStableDiffusionInpaintPipeline (line 102) | class FlaxStableDiffusionInpaintPipeline(FlaxDiffusionPipeline): method __init__ (line 131) | def __init__( method prepare_inputs (line 189) | def prepare_inputs( method _get_has_nsfw_concepts (line 228) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 232) | def _run_safety_checker(self, images, safety_model_params, jit=False): method _generate (line 262) | def _generate( method __call__ (line 390) | def __call__( function _p_generate (line 523) | def _p_generate( function _p_get_has_nsfw_concepts (line 553) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 557) | def unshard(x: jnp.ndarray): function preprocess_image (line 564) | def preprocess_image(image, dtype): function preprocess_mask (line 573) | def preprocess_mask(mask, dtype): FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py class OnnxStableDiffusionPipeline (line 33) | class OnnxStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 45) | def __init__( method _encode_prompt (line 114) | def _encode_prompt( method check_inputs (line 217) | def check_inputs( method __call__ (line 264) | def __call__( class StableDiffusionOnnxPipeline (line 462) | class StableDiffusionOnnxPipeline(OnnxStableDiffusionPipeline): method __init__ (line 463) | def __init__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py function preprocess (line 36) | def preprocess(image): class OnnxStableDiffusionImg2ImgPipeline (line 64) | class OnnxStableDiffusionImg2ImgPipeline(DiffusionPipeline): method __init__ (line 102) | def __init__( method _encode_prompt (line 172) | def _encode_prompt( method check_inputs (line 275) | def check_inputs( method __call__ (line 317) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py function prepare_mask_and_masked_image (line 38) | def prepare_mask_and_masked_image(image, mask, latents_shape): class OnnxStableDiffusionInpaintPipeline (line 56) | class OnnxStableDiffusionInpaintPipeline(DiffusionPipeline): method __init__ (line 94) | def __init__( method _encode_prompt (line 165) | def _encode_prompt( method check_inputs (line 269) | def check_inputs( method __call__ (line 317) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint_legacy.py function preprocess (line 20) | def preprocess(image): function preprocess_mask (line 29) | def preprocess_mask(mask, scale_factor=8): class OnnxStableDiffusionInpaintPipelineLegacy (line 41) | class OnnxStableDiffusionInpaintPipelineLegacy(DiffusionPipeline): method __init__ (line 80) | def __init__( method _encode_prompt (line 150) | def _encode_prompt( method check_inputs (line 253) | def check_inputs( method __call__ (line 295) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_upscale.py function preprocess (line 26) | def preprocess(image): class OnnxStableDiffusionUpscalePipeline (line 48) | class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline): method __init__ (line 49) | def __init__( method __call__ (line 72) | def __call__( method decode_latents (line 281) | def decode_latents(self, latents): method _encode_prompt (line 288) | def _encode_prompt( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py function rescale_noise_cfg (line 58) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0): class StableDiffusionPipeline (line 72) | class StableDiffusionPipeline(DiffusionPipeline, TextualInversionLoaderM... method __init__ (line 109) | def __init__( method enable_vae_slicing (line 199) | def enable_vae_slicing(self): method disable_vae_slicing (line 208) | def disable_vae_slicing(self): method enable_vae_tiling (line 215) | def enable_vae_tiling(self): method disable_vae_tiling (line 224) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 231) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 256) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 285) | def _execution_device(self): method _encode_prompt (line 302) | def _encode_prompt( method run_safety_checker (line 456) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 470) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 483) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 500) | def check_inputs( method prepare_latents (line 547) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 566) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py class AttentionStore (line 74) | class AttentionStore: method get_empty_store (line 76) | def get_empty_store(): method __call__ (line 79) | def __call__(self, attn, is_cross: bool, place_in_unet: str): method between_steps (line 89) | def between_steps(self): method get_average_attention (line 93) | def get_average_attention(self): method aggregate_attention (line 97) | def aggregate_attention(self, from_where: List[str]) -> torch.Tensor: method reset (line 109) | def reset(self): method __init__ (line 114) | def __init__(self, attn_res): class AttendExciteAttnProcessor (line 127) | class AttendExciteAttnProcessor: method __init__ (line 128) | def __init__(self, attnstore, place_in_unet): method __call__ (line 133) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class StableDiffusionAttendAndExcitePipeline (line 165) | class StableDiffusionAttendAndExcitePipeline(DiffusionPipeline, TextualI... method __init__ (line 194) | def __init__( method enable_vae_slicing (line 237) | def enable_vae_slicing(self): method disable_vae_slicing (line 247) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 255) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 282) | def _execution_device(self): method _encode_prompt (line 300) | def _encode_prompt( method run_safety_checker (line 455) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 470) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 484) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 501) | def check_inputs( method prepare_latents (line 575) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method _compute_max_attention_per_index (line 593) | def _compute_max_attention_per_index( method _aggregate_and_get_max_attention_per_token (line 615) | def _aggregate_and_get_max_attention_per_token( method _compute_loss (line 630) | def _compute_loss(max_attention_per_index: List[torch.Tensor]) -> torc... method _update_latent (line 637) | def _update_latent(latents: torch.Tensor, loss: torch.Tensor, step_siz... method _perform_iterative_refinement_step (line 643) | def _perform_iterative_refinement_step( method register_attention_control (line 697) | def register_attention_control(self): method get_indices (line 716) | def get_indices(self, prompt: str) -> Dict[str, int]: method __call__ (line 724) | def __call__( class GaussianSmoothing (line 1015) | class GaussianSmoothing(torch.nn.Module): method __init__ (line 1028) | def __init__( method forward (line 1069) | def forward(self, input): FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py function preprocess (line 39) | def preprocess(image): class StableDiffusionDepth2ImgPipeline (line 65) | class StableDiffusionDepth2ImgPipeline(DiffusionPipeline, TextualInversi... method __init__ (line 95) | def __init__( method enable_sequential_cpu_offload (line 140) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 159) | def _execution_device(self): method _encode_prompt (line 177) | def _encode_prompt( method run_safety_checker (line 332) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 347) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 361) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 379) | def check_inputs( method get_timesteps (line 420) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 430) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method prepare_depth_map (line 487) | def prepare_depth_map(self, image, depth_map, batch_size, do_classifie... method __call__ (line 532) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_diffedit.py class DiffEditInversionPipelineOutput (line 50) | class DiffEditInversionPipelineOutput(BaseOutput): function auto_corr_loss (line 140) | def auto_corr_loss(hidden_states, generator=None): function kl_divergence (line 156) | def kl_divergence(hidden_states): function preprocess (line 161) | def preprocess(image): function preprocess_mask (line 187) | def preprocess_mask(mask, batch_size: int = 1): class StableDiffusionDiffEditPipeline (line 240) | class StableDiffusionDiffEditPipeline(DiffusionPipeline, TextualInversio... method __init__ (line 277) | def __init__( method enable_vae_slicing (line 371) | def enable_vae_slicing(self): method disable_vae_slicing (line 381) | def disable_vae_slicing(self): method enable_vae_tiling (line 389) | def enable_vae_tiling(self): method disable_vae_tiling (line 399) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 407) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 433) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 463) | def _execution_device(self): method _encode_prompt (line 481) | def _encode_prompt( method run_safety_checker (line 636) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 651) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 669) | def decode_latents(self, latents): method check_inputs (line 682) | def check_inputs( method check_source_inputs (line 730) | def check_source_inputs( method get_timesteps (line 766) | def get_timesteps(self, num_inference_steps, strength, device): method get_inverse_timesteps (line 775) | def get_inverse_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 789) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_image_latents (line 807) | def prepare_image_latents(self, image, batch_size, dtype, device, gene... method get_epsilon (line 856) | def get_epsilon(self, model_output: torch.Tensor, sample: torch.Tensor... method generate_mask (line 874) | def generate_mask( method invert (line 1097) | def invert( method __call__ (line 1335) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py class StableDiffusionImageVariationPipeline (line 37) | class StableDiffusionImageVariationPipeline(DiffusionPipeline): method __init__ (line 65) | def __init__( method enable_sequential_cpu_offload (line 126) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 145) | def _execution_device(self): method _encode_image (line 162) | def _encode_image(self, image, device, num_images_per_prompt, do_class... method run_safety_checker (line 188) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 203) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 217) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 234) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 257) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 275) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py function preprocess (line 75) | def preprocess(image): class StableDiffusionImg2ImgPipeline (line 101) | class StableDiffusionImg2ImgPipeline( method __init__ (line 140) | def __init__( method enable_sequential_cpu_offload (line 231) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 257) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 287) | def _execution_device(self): method _encode_prompt (line 305) | def _encode_prompt( method run_safety_checker (line 460) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 475) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 489) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 506) | def check_inputs( method get_timesteps (line 546) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 555) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 614) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py function prepare_mask_and_masked_image (line 39) | def prepare_mask_and_masked_image(image, mask, height, width, return_ima... class StableDiffusionInpaintPipeline (line 156) | class StableDiffusionInpaintPipeline( method __init__ (line 204) | def __init__( method enable_sequential_cpu_offload (line 300) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 326) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 356) | def _execution_device(self): method _encode_prompt (line 374) | def _encode_prompt( method run_safety_checker (line 529) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 544) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 562) | def decode_latents(self, latents): method check_inputs (line 575) | def check_inputs( method prepare_latents (line 626) | def prepare_latents( method _encode_vae_image (line 679) | def _encode_vae_image(self, image: torch.Tensor, generator: torch.Gene... method prepare_mask_latents (line 693) | def prepare_mask_latents( method get_timesteps (line 735) | def get_timesteps(self, num_inference_steps, strength, device): method __call__ (line 745) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py function preprocess_image (line 46) | def preprocess_image(image, batch_size): function preprocess_mask (line 56) | def preprocess_mask(mask, batch_size, scale_factor=8): class StableDiffusionInpaintPipelineLegacy (line 87) | class StableDiffusionInpaintPipelineLegacy( method __init__ (line 126) | def __init__( method enable_sequential_cpu_offload (line 224) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 250) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 280) | def _execution_device(self): method _encode_prompt (line 298) | def _encode_prompt( method run_safety_checker (line 453) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 468) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 482) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 500) | def check_inputs( method get_timesteps (line 541) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 550) | def prepare_latents(self, image, timestep, num_images_per_prompt, dtyp... method __call__ (line 567) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py function preprocess (line 45) | def preprocess(image): class StableDiffusionInstructPix2PixPipeline (line 71) | class StableDiffusionInstructPix2PixPipeline(DiffusionPipeline, TextualI... method __init__ (line 107) | def __init__( method __call__ (line 150) | def __call__( method enable_sequential_cpu_offload (line 432) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 458) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 488) | def _execution_device(self): method _encode_prompt (line 505) | def _encode_prompt( method run_safety_checker (line 653) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 668) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 686) | def decode_latents(self, latents): method check_inputs (line 699) | def check_inputs( method prepare_latents (line 737) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_image_latents (line 754) | def prepare_image_latents( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py class ModelWrapper (line 34) | class ModelWrapper: method __init__ (line 35) | def __init__(self, model, alphas_cumprod): method apply_model (line 39) | def apply_model(self, *args, **kwargs): class StableDiffusionKDiffusionPipeline (line 48) | class StableDiffusionKDiffusionPipeline(DiffusionPipeline, TextualInvers... method __init__ (line 83) | def __init__( method set_scheduler (line 124) | def set_scheduler(self, scheduler_type: str): method enable_sequential_cpu_offload (line 130) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 156) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 186) | def _execution_device(self): method _encode_prompt (line 204) | def _encode_prompt( method run_safety_checker (line 359) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 374) | def decode_latents(self, latents): method check_inputs (line 388) | def check_inputs( method prepare_latents (line 435) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 448) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py function preprocess (line 35) | def preprocess(image): class StableDiffusionLatentUpscalePipeline (line 61) | class StableDiffusionLatentUpscalePipeline(DiffusionPipeline): method __init__ (line 84) | def __init__( method enable_sequential_cpu_offload (line 104) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 123) | def _execution_device(self): method _encode_prompt (line 140) | def _encode_prompt(self, prompt, device, do_classifier_free_guidance, ... method decode_latents (line 231) | def decode_latents(self, latents): method check_inputs (line 244) | def check_inputs(self, prompt, image, callback_steps): method prepare_latents (line 282) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 296) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py class LDM3DPipelineOutput (line 59) | class LDM3DPipelineOutput(BaseOutput): class StableDiffusionLDM3DPipeline (line 77) | class StableDiffusionLDM3DPipeline( method __init__ (line 118) | def __init__( method enable_vae_slicing (line 161) | def enable_vae_slicing(self): method disable_vae_slicing (line 171) | def disable_vae_slicing(self): method enable_vae_tiling (line 179) | def enable_vae_tiling(self): method disable_vae_tiling (line 189) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 196) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 222) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 252) | def _execution_device(self): method _encode_prompt (line 270) | def _encode_prompt( method run_safety_checker (line 424) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 440) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 458) | def check_inputs( method prepare_latents (line 505) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 524) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_model_editing.py class StableDiffusionModelEditingPipeline (line 58) | class StableDiffusionModelEditingPipeline(DiffusionPipeline, TextualInve... method __init__ (line 90) | def __init__( method enable_vae_slicing (line 168) | def enable_vae_slicing(self): method disable_vae_slicing (line 178) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 186) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 213) | def _execution_device(self): method _encode_prompt (line 231) | def _encode_prompt( method run_safety_checker (line 386) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 401) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 415) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 433) | def check_inputs( method prepare_latents (line 481) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method edit_model (line 499) | def edit_model( method __call__ (line 617) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_panorama.py class StableDiffusionPanoramaPipeline (line 54) | class StableDiffusionPanoramaPipeline(DiffusionPipeline, TextualInversio... method __init__ (line 87) | def __init__( method enable_vae_slicing (line 130) | def enable_vae_slicing(self): method disable_vae_slicing (line 140) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 148) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 175) | def _execution_device(self): method _encode_prompt (line 193) | def _encode_prompt( method run_safety_checker (line 348) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 363) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 377) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 395) | def check_inputs( method prepare_latents (line 443) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method get_views (line 460) | def get_views(self, panorama_height, panorama_width, window_size=64, s... method __call__ (line 479) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_paradigms.py class StableDiffusionParadigmsPipeline (line 62) | class StableDiffusionParadigmsPipeline( method __init__ (line 104) | def __init__( method enable_vae_slicing (line 150) | def enable_vae_slicing(self): method disable_vae_slicing (line 160) | def disable_vae_slicing(self): method enable_vae_tiling (line 168) | def enable_vae_tiling(self): method disable_vae_tiling (line 178) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 186) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 212) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 242) | def _execution_device(self): method _encode_prompt (line 260) | def _encode_prompt( method run_safety_checker (line 415) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 430) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 448) | def check_inputs( method prepare_latents (line 496) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method _cumsum (line 513) | def _cumsum(self, input, dim, debug=False): method __call__ (line 523) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py class Pix2PixInversionPipelineOutput (line 57) | class Pix2PixInversionPipelineOutput(BaseOutput, TextualInversionLoaderM... function preprocess (line 179) | def preprocess(image): function prepare_unet (line 205) | def prepare_unet(unet: UNet2DConditionModel): class Pix2PixZeroL2Loss (line 222) | class Pix2PixZeroL2Loss: method __init__ (line 223) | def __init__(self): method compute_loss (line 226) | def compute_loss(self, predictions, targets): class Pix2PixZeroAttnProcessor (line 230) | class Pix2PixZeroAttnProcessor: method __init__ (line 234) | def __init__(self, is_pix2pix_zero=False): method __call__ (line 239) | def __call__( class StableDiffusionPix2PixZeroPipeline (line 285) | class StableDiffusionPix2PixZeroPipeline(DiffusionPipeline): method __init__ (line 323) | def __init__( method enable_sequential_cpu_offload (line 372) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 397) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 423) | def _execution_device(self): method _encode_prompt (line 441) | def _encode_prompt( method run_safety_checker (line 596) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 611) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 625) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 642) | def check_inputs( method prepare_latents (line 673) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method generate_caption (line 691) | def generate_caption(self, images): method construct_direction (line 710) | def construct_direction(self, embs_source: torch.Tensor, embs_target: ... method get_embeds (line 715) | def get_embeds(self, prompt: List[str], batch_size: int = 16) -> torch... method prepare_image_latents (line 734) | def prepare_image_latents(self, image, batch_size, dtype, device, gene... method get_epsilon (line 783) | def get_epsilon(self, model_output: torch.Tensor, sample: torch.Tensor... method auto_corr_loss (line 800) | def auto_corr_loss(self, hidden_states, generator=None): method kl_divergence (line 815) | def kl_divergence(self, hidden_states): method __call__ (line 822) | def __call__( method invert (line 1101) | def invert( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_sag.py class CrossAttnStoreProcessor (line 53) | class CrossAttnStoreProcessor: method __init__ (line 54) | def __init__(self): method __call__ (line 57) | def __call__( class StableDiffusionSAGPipeline (line 93) | class StableDiffusionSAGPipeline(DiffusionPipeline, TextualInversionLoad... method __init__ (line 122) | def __init__( method enable_vae_slicing (line 149) | def enable_vae_slicing(self): method disable_vae_slicing (line 159) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 167) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 194) | def _execution_device(self): method _encode_prompt (line 212) | def _encode_prompt( method run_safety_checker (line 367) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 382) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 396) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 414) | def check_inputs( method prepare_latents (line 462) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 481) | def __call__( method sag_masking (line 724) | def sag_masking(self, original_latents, attn_map, map_size, t, eps): method pred_x0 (line 754) | def pred_x0(self, sample, model_output, timestep): method pred_epsilon (line 774) | def pred_epsilon(self, sample, model_output, timestep): function gaussian_blur_2d (line 794) | def gaussian_blur_2d(img, kernel_size, sigma): FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py function preprocess (line 42) | def preprocess(image): class StableDiffusionUpscalePipeline (line 68) | class StableDiffusionUpscalePipeline(DiffusionPipeline, TextualInversion... method __init__ (line 95) | def __init__( method enable_sequential_cpu_offload (line 143) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 160) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 188) | def _execution_device(self): method run_safety_checker (line 206) | def run_safety_checker(self, image, device, dtype): method _encode_prompt (line 223) | def _encode_prompt( method prepare_extra_step_kwargs (line 378) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 396) | def decode_latents(self, latents): method check_inputs (line 409) | def check_inputs( method prepare_latents (line 491) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 505) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py class StableUnCLIPPipeline (line 53) | class StableUnCLIPPipeline(DiffusionPipeline, TextualInversionLoaderMixi... method __init__ (line 106) | def __init__( method enable_vae_slicing (line 144) | def enable_vae_slicing(self): method disable_vae_slicing (line 154) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 161) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 185) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 212) | def _execution_device(self): method _encode_prior_prompt (line 230) | def _encode_prior_prompt( method _encode_prompt (line 332) | def _encode_prompt( method decode_latents (line 487) | def decode_latents(self, latents): method prepare_prior_extra_step_kwargs (line 501) | def prepare_prior_extra_step_kwargs(self, generator, eta): method prepare_extra_step_kwargs (line 519) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 536) | def check_inputs( method prepare_latents (line 597) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method noise_image_embeddings (line 608) | def noise_image_embeddings( method __call__ (line 656) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py class StableUnCLIPImg2ImgPipeline (line 66) | class StableUnCLIPImg2ImgPipeline(DiffusionPipeline, TextualInversionLoa... method __init__ (line 112) | def __init__( method enable_vae_slicing (line 146) | def enable_vae_slicing(self): method disable_vae_slicing (line 156) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 163) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 187) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 214) | def _execution_device(self): method _encode_prompt (line 232) | def _encode_prompt( method _encode_image (line 386) | def _encode_image( method decode_latents (line 442) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 456) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 473) | def check_inputs( method prepare_latents (line 557) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method noise_image_embeddings (line 575) | def noise_image_embeddings( method __call__ (line 623) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/safety_checker.py function cosine_distance (line 26) | def cosine_distance(image_embeds, text_embeds): class StableDiffusionSafetyChecker (line 32) | class StableDiffusionSafetyChecker(PreTrainedModel): method __init__ (line 37) | def __init__(self, config: CLIPConfig): method forward (line 50) | def forward(self, clip_input, images): method forward_onnx (line 102) | def forward_onnx(self, clip_input: torch.FloatTensor, images: torch.Fl... FILE: 4DoF/diffusers/pipelines/stable_diffusion/safety_checker_flax.py function jax_cosine_distance (line 25) | def jax_cosine_distance(emb_1, emb_2, eps=1e-12): class FlaxStableDiffusionSafetyCheckerModule (line 31) | class FlaxStableDiffusionSafetyCheckerModule(nn.Module): method setup (line 35) | def setup(self): method __call__ (line 47) | def __call__(self, clip_input): class FlaxStableDiffusionSafetyChecker (line 71) | class FlaxStableDiffusionSafetyChecker(FlaxPreTrainedModel): method __init__ (line 76) | def __init__( method init_weights (line 90) | def init_weights(self, rng: jax.random.KeyArray, input_shape: Tuple, p... method __call__ (line 101) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion/stable_unclip_image_normalizer.py class StableUnCLIPImageNormalizer (line 24) | class StableUnCLIPImageNormalizer(ModelMixin, ConfigMixin): method __init__ (line 33) | def __init__( method to (line 42) | def to( method scale (line 51) | def scale(self, embeds): method unscale (line 55) | def unscale(self, embeds): FILE: 4DoF/diffusers/pipelines/stable_diffusion_safe/__init__.py class SafetyConfig (line 13) | class SafetyConfig(object): class StableDiffusionSafePipelineOutput (line 45) | class StableDiffusionSafePipelineOutput(BaseOutput): FILE: 4DoF/diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py class StableDiffusionPipelineSafe (line 22) | class StableDiffusionPipelineSafe(DiffusionPipeline): method __init__ (line 54) | def __init__( method safety_concept (line 150) | def safety_concept(self): method safety_concept (line 160) | def safety_concept(self, concept): method enable_sequential_cpu_offload (line 170) | def enable_sequential_cpu_offload(self): method _execution_device (line 189) | def _execution_device(self): method _encode_prompt (line 206) | def _encode_prompt( method run_safety_checker (line 341) | def run_safety_checker(self, image, device, dtype, enable_safety_guida... method decode_latents (line 365) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 379) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 397) | def check_inputs( method prepare_latents (line 445) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method perform_safety_guidance (line 462) | def perform_safety_guidance( method __call__ (line 505) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion_safe/safety_checker.py function cosine_distance (line 25) | def cosine_distance(image_embeds, text_embeds): class SafeStableDiffusionSafetyChecker (line 31) | class SafeStableDiffusionSafetyChecker(PreTrainedModel): method __init__ (line 36) | def __init__(self, config: CLIPConfig): method forward (line 49) | def forward(self, clip_input, images): method forward_onnx (line 88) | def forward_onnx(self, clip_input: torch.FloatTensor, images: torch.Fl... FILE: 4DoF/diffusers/pipelines/stable_diffusion_xl/__init__.py class StableDiffusionXLPipelineOutput (line 11) | class StableDiffusionXLPipelineOutput(BaseOutput): FILE: 4DoF/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py function rescale_noise_cfg (line 62) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0): class StableDiffusionXLPipeline (line 76) | class StableDiffusionXLPipeline(DiffusionPipeline, FromSingleFileMixin): method __init__ (line 107) | def __init__( method enable_vae_slicing (line 137) | def enable_vae_slicing(self): method disable_vae_slicing (line 147) | def disable_vae_slicing(self): method enable_vae_tiling (line 155) | def enable_vae_tiling(self): method disable_vae_tiling (line 165) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 172) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 194) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 226) | def _execution_device(self): method encode_prompt (line 243) | def encode_prompt( method prepare_extra_step_kwargs (line 430) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 447) | def check_inputs( method prepare_latents (line 507) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method _get_add_time_ids (line 524) | def _get_add_time_ids(self, original_size, crops_coords_top_left, targ... method __call__ (line 542) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py function rescale_noise_cfg (line 67) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0): class StableDiffusionXLImg2ImgPipeline (line 81) | class StableDiffusionXLImg2ImgPipeline(DiffusionPipeline, FromSingleFile... method __init__ (line 113) | def __init__( method enable_vae_slicing (line 144) | def enable_vae_slicing(self): method disable_vae_slicing (line 154) | def disable_vae_slicing(self): method enable_vae_tiling (line 162) | def enable_vae_tiling(self): method disable_vae_tiling (line 172) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 180) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 203) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 235) | def _execution_device(self): method encode_prompt (line 253) | def encode_prompt( method prepare_extra_step_kwargs (line 440) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 457) | def check_inputs( method get_timesteps (line 497) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 506) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method _get_add_time_ids (line 568) | def _get_add_time_ids( method __call__ (line 609) | def __call__( FILE: 4DoF/diffusers/pipelines/stable_diffusion_xl/watermark.py class StableDiffusionXLWatermarker (line 12) | class StableDiffusionXLWatermarker: method __init__ (line 13) | def __init__(self): method apply_watermark (line 19) | def apply_watermark(self, images: torch.FloatTensor): FILE: 4DoF/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py class KarrasVePipeline (line 25) | class KarrasVePipeline(DiffusionPipeline): method __init__ (line 44) | def __init__(self, unet: UNet2DModel, scheduler: KarrasVeScheduler): method __call__ (line 49) | def __call__( FILE: 4DoF/diffusers/pipelines/text_to_video_synthesis/__init__.py class TextToVideoSDPipelineOutput (line 11) | class TextToVideoSDPipelineOutput(BaseOutput): FILE: 4DoF/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py function tensor2vid (line 58) | def tensor2vid(video: torch.Tensor, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5,... class TextToVideoSDPipeline (line 76) | class TextToVideoSDPipeline(DiffusionPipeline, TextualInversionLoaderMix... method __init__ (line 97) | def __init__( method enable_vae_slicing (line 117) | def enable_vae_slicing(self): method disable_vae_slicing (line 127) | def disable_vae_slicing(self): method enable_vae_tiling (line 135) | def enable_vae_tiling(self): method disable_vae_tiling (line 145) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 152) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 173) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 200) | def _execution_device(self): method _encode_prompt (line 218) | def _encode_prompt( method decode_latents (line 372) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 396) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 414) | def check_inputs( method prepare_latents (line 461) | def prepare_latents( method __call__ (line 488) | def __call__( FILE: 4DoF/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py function tensor2vid (line 79) | def tensor2vid(video: torch.Tensor, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5,... function preprocess_video (line 97) | def preprocess_video(video): class VideoToVideoSDPipeline (line 138) | class VideoToVideoSDPipeline(DiffusionPipeline, TextualInversionLoaderMi... method __init__ (line 159) | def __init__( method enable_vae_slicing (line 179) | def enable_vae_slicing(self): method disable_vae_slicing (line 189) | def disable_vae_slicing(self): method enable_vae_tiling (line 197) | def enable_vae_tiling(self): method disable_vae_tiling (line 207) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 214) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 235) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 262) | def _execution_device(self): method _encode_prompt (line 280) | def _encode_prompt( method decode_latents (line 435) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 459) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 477) | def check_inputs( method get_timesteps (line 518) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 527) | def prepare_latents(self, video, timestep, batch_size, dtype, device, ... method __call__ (line 573) | def __call__( FILE: 4DoF/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py function rearrange_0 (line 18) | def rearrange_0(tensor, f): function rearrange_1 (line 24) | def rearrange_1(tensor): function rearrange_3 (line 29) | def rearrange_3(tensor, f): function rearrange_4 (line 34) | def rearrange_4(tensor): class CrossFrameAttnProcessor (line 39) | class CrossFrameAttnProcessor: method __init__ (line 49) | def __init__(self, batch_size=2): method __call__ (line 52) | def __call__(self, attn, hidden_states, encoder_hidden_states=None, at... class CrossFrameAttnProcessor2_0 (line 98) | class CrossFrameAttnProcessor2_0: method __init__ (line 108) | def __init__(self, batch_size=2): method __call__ (line 113) | def __call__(self, attn, hidden_states, encoder_hidden_states=None, at... class TextToVideoPipelineOutput (line 174) | class TextToVideoPipelineOutput(BaseOutput): function coords_grid (line 179) | def coords_grid(batch, ht, wd, device): function warp_single_latent (line 186) | def warp_single_latent(latent, reference_flow): function create_motion_field (line 213) | def create_motion_field(motion_field_strength_x, motion_field_strength_y... function create_motion_field_and_warp_latents (line 236) | def create_motion_field_and_warp_latents(motion_field_strength_x, motion... class TextToVideoZeroPipeline (line 263) | class TextToVideoZeroPipeline(StableDiffusionPipeline): method __init__ (line 291) | def __init__( method forward_loop (line 312) | def forward_loop(self, x_t0, t0, t1, generator): method backward_loop (line 330) | def backward_loop( method __call__ (line 400) | def __call__( FILE: 4DoF/diffusers/pipelines/unclip/pipeline_unclip.py class UnCLIPPipeline (line 34) | class UnCLIPPipeline(DiffusionPipeline): method __init__ (line 78) | def __init__( method prepare_latents (line 106) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 117) | def _encode_prompt( method enable_sequential_cpu_offload (line 208) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 234) | def _execution_device(self): method __call__ (line 252) | def __call__( FILE: 4DoF/diffusers/pipelines/unclip/pipeline_unclip_image_variation.py class UnCLIPImageVariationPipeline (line 38) | class UnCLIPImageVariationPipeline(DiffusionPipeline): method __init__ (line 84) | def __init__( method prepare_latents (line 113) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 124) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method _encode_image (line 187) | def _encode_image(self, image, device, num_images_per_prompt, image_em... method enable_sequential_cpu_offload (line 201) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 227) | def _execution_device(self): method __call__ (line 245) | def __call__( FILE: 4DoF/diffusers/pipelines/unclip/text_proj.py class UnCLIPTextProjModel (line 22) | class UnCLIPTextProjModel(ModelMixin, ConfigMixin): method __init__ (line 31) | def __init__( method forward (line 55) | def forward(self, *, image_embeddings, prompt_embeds, text_encoder_hid... FILE: 4DoF/diffusers/pipelines/unidiffuser/modeling_text_decoder.py class UniDiffuserTextDecoder (line 14) | class UniDiffuserTextDecoder(ModelMixin, ConfigMixin, ModuleUtilsMixin): method __init__ (line 67) | def __init__( method forward (line 131) | def forward( method get_dummy_token (line 163) | def get_dummy_token(self, batch_size: int, device: torch.device) -> to... method encode (line 166) | def encode(self, prefix): method generate_captions (line 170) | def generate_captions(self, features, eos_token_id, device): method generate_beam (line 202) | def generate_beam( FILE: 4DoF/diffusers/pipelines/unidiffuser/modeling_uvit.py function _no_grad_trunc_normal_ (line 19) | def _no_grad_trunc_normal_(tensor, mean, std, a, b): function trunc_normal_ (line 56) | def trunc_normal_(tensor, mean=0.0, std=1.0, a=-2.0, b=2.0): class PatchEmbed (line 75) | class PatchEmbed(nn.Module): method __init__ (line 78) | def __init__( method forward (line 109) | def forward(self, latent): class SkipBlock (line 121) | class SkipBlock(nn.Module): method __init__ (line 122) | def __init__(self, dim: int): method forward (line 130) | def forward(self, x, skip): class UTransformerBlock (line 140) | class UTransformerBlock(nn.Module): method __init__ (line 174) | def __init__( method forward (line 251) | def forward( class UniDiffuserBlock (line 333) | class UniDiffuserBlock(nn.Module): method __init__ (line 369) | def __init__( method forward (line 446) | def forward( class UTransformer2DModel (line 530) | class UTransformer2DModel(ModelMixin, ConfigMixin): method __init__ (line 589) | def __init__( method forward (line 727) | def forward( class UniDiffuserModel (line 832) | class UniDiffuserModel(ModelMixin, ConfigMixin): method __init__ (line 897) | def __init__( method no_weight_decay (line 1034) | def no_weight_decay(self): method forward (line 1037) | def forward( FILE: 4DoF/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py function preprocess (line 37) | def preprocess(image): class ImageTextPipelineOutput (line 65) | class ImageTextPipelineOutput(BaseOutput): class UniDiffuserPipeline (line 82) | class UniDiffuserPipeline(DiffusionPipeline): method __init__ (line 126) | def __init__( method enable_sequential_cpu_offload (line 177) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 204) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 234) | def _execution_device(self): method prepare_extra_step_kwargs (line 252) | def prepare_extra_step_kwargs(self, generator, eta): method _infer_mode (line 269) | def _infer_mode(self, prompt, prompt_embeds, image, latents, prompt_la... method set_text_mode (line 328) | def set_text_mode(self): method set_image_mode (line 332) | def set_image_mode(self): method set_text_to_image_mode (line 336) | def set_text_to_image_mode(self): method set_image_to_text_mode (line 340) | def set_image_to_text_mode(self): method set_joint_mode (line 344) | def set_joint_mode(self): method reset_mode (line 348) | def reset_mode(self): method _infer_batch_size (line 352) | def _infer_batch_size( method _encode_prompt (line 430) | def _encode_prompt( method encode_image_vae_latents (line 570) | def encode_image_vae_latents( method encode_image_clip_latents (line 630) | def encode_image_clip_latents( method decode_image_latents (line 689) | def decode_image_latents(self, latents): method prepare_text_latents (line 697) | def prepare_text_latents( method prepare_image_vae_latents (line 721) | def prepare_image_vae_latents( method prepare_image_clip_latents (line 756) | def prepare_image_clip_latents( method _split (line 778) | def _split(self, x, height, width): method _combine (line 794) | def _combine(self, img_vae, img_clip): method _split_joint (line 803) | def _split_joint(self, x, height, width): method _combine_joint (line 822) | def _combine_joint(self, img_vae, img_clip, text): method _get_noise_pred (line 833) | def _get_noise_pred( method check_latents_shape (line 955) | def check_latents_shape(self, latents_name, latents, expected_shape): method check_inputs (line 971) | def check_inputs( method __call__ (line 1085) | def __call__( FILE: 4DoF/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py function get_down_block (line 38) | def get_down_block( function get_up_block (line 97) | def get_up_block( class UNetFlatConditionModel (line 157) | class UNetFlatConditionModel(ModelMixin, ConfigMixin): method __init__ (line 248) | def __init__( method attn_processors (line 674) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 697) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 731) | def set_default_attn_processor(self): method set_attention_slice (line 737) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 802) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 806) | def forward( class LinearMultiDim (line 1096) | class LinearMultiDim(nn.Linear): method __init__ (line 1097) | def __init__(self, in_features, out_features=None, second_dim=4, *args... method forward (line 1106) | def forward(self, input_tensor, *args, **kwargs): class ResnetBlockFlat (line 1115) | class ResnetBlockFlat(nn.Module): method __init__ (line 1116) | def __init__( method forward (line 1176) | def forward(self, input_tensor, temb): class DownBlockFlat (line 1210) | class DownBlockFlat(nn.Module): method __init__ (line 1211) | def __init__( method forward (line 1262) | def forward(self, hidden_states, temb=None): class CrossAttnDownBlockFlat (line 1297) | class CrossAttnDownBlockFlat(nn.Module): method __init__ (line 1298) | def __init__( method forward (line 1385) | def forward( class UpBlockFlat (line 1449) | class UpBlockFlat(nn.Module): method __init__ (line 1450) | def __init__( method forward (line 1497) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class CrossAttnUpBlockFlat (line 1531) | class CrossAttnUpBlockFlat(nn.Module): method __init__ (line 1532) | def __init__( method forward (line 1615) | def forward( class UNetMidBlockFlatCrossAttn (line 1680) | class UNetMidBlockFlatCrossAttn(nn.Module): method __init__ (line 1681) | def __init__( method forward (line 1766) | def forward( class UNetMidBlockFlatSimpleCrossAttn (line 1791) | class UNetMidBlockFlatSimpleCrossAttn(nn.Module): method __init__ (line 1792) | def __init__( method forward (line 1876) | def forward( FILE: 4DoF/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion.py class VersatileDiffusionPipeline (line 20) | class VersatileDiffusionPipeline(DiffusionPipeline): method __init__ (line 57) | def __init__( method image_variation (line 83) | def image_variation( method text_to_image (line 199) | def text_to_image( method dual_guided (line 311) | def dual_guided( FILE: 4DoF/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py class VersatileDiffusionDualGuidedPipeline (line 41) | class VersatileDiffusionDualGuidedPipeline(DiffusionPipeline): method __init__ (line 70) | def __init__( method remove_unused_weights (line 101) | def remove_unused_weights(self): method _convert_to_dual_attention (line 104) | def _convert_to_dual_attention(self): method _revert_dual_attention (line 138) | def _revert_dual_attention(self): method enable_sequential_cpu_offload (line 151) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 170) | def _execution_device(self): method _encode_text_prompt (line 187) | def _encode_text_prompt(self, prompt, device, num_images_per_prompt, d... method _encode_image_prompt (line 278) | def _encode_image_prompt(self, prompt, device, num_images_per_prompt, ... method decode_latents (line 334) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 348) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 365) | def check_inputs(self, prompt, image, height, width, callback_steps): method prepare_latents (line 383) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method set_transformer_params (line 400) | def set_transformer_params(self, mix_ratio: float = 0.5, condition_typ... method __call__ (line 414) | def __call__( FILE: 4DoF/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py class VersatileDiffusionImageVariationPipeline (line 35) | class VersatileDiffusionImageVariationPipeline(DiffusionPipeline): method __init__ (line 59) | def __init__( method enable_sequential_cpu_offload (line 78) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 97) | def _execution_device(self): method _encode_prompt (line 114) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method decode_latents (line 194) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 208) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 226) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 249) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 267) | def __call__( FILE: 4DoF/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py class VersatileDiffusionTextToImagePipeline (line 34) | class VersatileDiffusionTextToImagePipeline(DiffusionPipeline): method __init__ (line 62) | def __init__( method _swap_unet_attention_blocks (line 86) | def _swap_unet_attention_blocks(self): method remove_unused_weights (line 99) | def remove_unused_weights(self): method enable_sequential_cpu_offload (line 102) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 121) | def _execution_device(self): method _encode_prompt (line 138) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method decode_latents (line 251) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 265) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 283) | def check_inputs( method prepare_latents (line 331) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 349) | def __call__( FILE: 4DoF/diffusers/pipelines/vq_diffusion/pipeline_vq_diffusion.py class LearnedClassifierFreeSamplingEmbeddings (line 30) | class LearnedClassifierFreeSamplingEmbeddings(ModelMixin, ConfigMixin): method __init__ (line 36) | def __init__(self, learnable: bool, hidden_size: Optional[int] = None,... class VQDiffusionPipeline (line 52) | class VQDiffusionPipeline(DiffusionPipeline): method __init__ (line 83) | def __init__( method _encode_prompt (line 103) | def _encode_prompt(self, prompt, num_images_per_prompt, do_classifier_... method __call__ (line 167) | def __call__( method truncate (line 310) | def truncate(self, log_p_x_0: torch.FloatTensor, truncation_rate: floa... FILE: 4DoF/diffusers/schedulers/scheduling_consistency_models.py class CMStochasticIterativeSchedulerOutput (line 30) | class CMStochasticIterativeSchedulerOutput(BaseOutput): class CMStochasticIterativeScheduler (line 43) | class CMStochasticIterativeScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 82) | def __init__( method index_for_timestep (line 106) | def index_for_timestep(self, timestep, schedule_timesteps=None): method scale_model_input (line 113) | def scale_model_input( method sigma_to_t (line 136) | def sigma_to_t(self, sigmas: Union[float, np.ndarray]): method set_timesteps (line 152) | def set_timesteps( method _convert_to_karras (line 223) | def _convert_to_karras(self, ramp): method get_scalings (line 235) | def get_scalings(self, sigma): method get_scalings_for_boundary_condition (line 242) | def get_scalings_for_boundary_condition(self, sigma): method step (line 264) | def step( method add_noise (line 354) | def add_noise( method __len__ (line 379) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_ddim.py class DDIMSchedulerOutput (line 32) | class DDIMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar( function rescale_zero_terminal_snr (line 94) | def rescale_zero_terminal_snr(betas): class DDIMScheduler (line 130) | class DDIMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 190) | def __init__( method scale_model_input (line 243) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method _get_variance (line 257) | def _get_variance(self, timestep, prev_timestep): method _threshold_sample (line 268) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method set_timesteps (line 302) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 347) | def step( method add_noise (line 470) | def add_noise( method get_velocity (line 494) | def get_velocity( method __len__ (line 514) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_ddim_flax.py class DDIMSchedulerState (line 36) | class DDIMSchedulerState: method create (line 46) | def create( class FlaxDDIMSchedulerOutput (line 62) | class FlaxDDIMSchedulerOutput(FlaxSchedulerOutput): class FlaxDDIMScheduler (line 66) | class FlaxDDIMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 109) | def has_state(self): method __init__ (line 113) | def __init__( method create_state (line 127) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 151) | def scale_model_input( method set_timesteps (line 165) | def set_timesteps( method _get_variance (line 187) | def _get_variance(self, state: DDIMSchedulerState, timestep, prev_time... method step (line 199) | def step( method add_noise (line 286) | def add_noise( method get_velocity (line 295) | def get_velocity( method __len__ (line 304) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_ddim_inverse.py class DDIMSchedulerOutput (line 31) | class DDIMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 49) | def betas_for_alpha_bar( class DDIMInverseScheduler (line 93) | class DDIMInverseScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 134) | def __init__( method scale_model_input (line 187) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 201) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 225) | def step( method __len__ (line 282) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_ddim_parallel.py class DDIMParallelSchedulerOutput (line 32) | class DDIMParallelSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar( function rescale_zero_terminal_snr (line 95) | def rescale_zero_terminal_snr(betas): class DDIMParallelScheduler (line 131) | class DDIMParallelScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 193) | def __init__( method scale_model_input (line 247) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method _get_variance (line 261) | def _get_variance(self, timestep, prev_timestep=None): method _batch_get_variance (line 274) | def _batch_get_variance(self, t, prev_t): method _threshold_sample (line 286) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method set_timesteps (line 321) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 366) | def step( method batch_step_no_noise (line 488) | def batch_step_no_noise( method add_noise (line 597) | def add_noise( method get_velocity (line 621) | def get_velocity( method __len__ (line 641) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_ddpm.py class DDPMSchedulerOutput (line 30) | class DDPMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 47) | def betas_for_alpha_bar( class DDPMScheduler (line 91) | class DDPMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 145) | def __init__( method scale_model_input (line 195) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 209) | def set_timesteps( method _get_variance (line 284) | def _get_variance(self, t, predicted_variance=None, variance_type=None): method _threshold_sample (line 324) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method step (line 358) | def step( method add_noise (line 454) | def add_noise( method get_velocity (line 477) | def get_velocity( method __len__ (line 497) | def __len__(self): method previous_timestep (line 500) | def previous_timestep(self, timestep): FILE: 4DoF/diffusers/schedulers/scheduling_ddpm_flax.py class DDPMSchedulerState (line 36) | class DDPMSchedulerState: method create (line 45) | def create(cls, common: CommonSchedulerState, init_noise_sigma: jnp.nd... class FlaxDDPMSchedulerOutput (line 50) | class FlaxDDPMSchedulerOutput(FlaxSchedulerOutput): class FlaxDDPMScheduler (line 54) | class FlaxDDPMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 92) | def has_state(self): method __init__ (line 96) | def __init__( method create_state (line 110) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 125) | def scale_model_input( method set_timesteps (line 139) | def set_timesteps( method _get_variance (line 162) | def _get_variance(self, state: DDPMSchedulerState, t, predicted_varian... method step (line 195) | def step( method add_noise (line 280) | def add_noise( method get_velocity (line 289) | def get_velocity( method __len__ (line 298) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_ddpm_parallel.py class DDPMParallelSchedulerOutput (line 31) | class DDPMParallelSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 49) | def betas_for_alpha_bar( class DDPMParallelScheduler (line 93) | class DDPMParallelScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 149) | def __init__( method scale_model_input (line 200) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 215) | def set_timesteps( method _get_variance (line 291) | def _get_variance(self, t, predicted_variance=None, variance_type=None): method _threshold_sample (line 332) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method step (line 366) | def step( method batch_step_no_noise (line 462) | def batch_step_no_noise( method add_noise (line 543) | def add_noise( method get_velocity (line 567) | def get_velocity( method __len__ (line 587) | def __len__(self): method previous_timestep (line 591) | def previous_timestep(self, timestep): FILE: 4DoF/diffusers/schedulers/scheduling_deis_multistep.py function betas_for_alpha_bar (line 29) | def betas_for_alpha_bar( class DEISMultistepScheduler (line 73) | class DEISMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 138) | def __init__( method set_timesteps (line 202) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _threshold_sample (line 261) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method convert_model_output (line 295) | def convert_model_output( method deis_first_order_update (line 333) | def deis_first_order_update( method multistep_deis_second_order_update (line 363) | def multistep_deis_second_order_update( method multistep_deis_third_order_update (line 405) | def multistep_deis_third_order_update( method step (line 462) | def step( method scale_model_input (line 530) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 544) | def add_noise( method __len__ (line 567) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_dpmsolver_multistep.py function betas_for_alpha_bar (line 29) | def betas_for_alpha_bar( class DPMSolverMultistepScheduler (line 73) | class DPMSolverMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 165) | def __init__( method set_timesteps (line 231) | def set_timesteps(self, num_inference_steps: int = None, device: Union... method _threshold_sample (line 292) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method _sigma_to_t (line 327) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 351) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method convert_model_output (line 364) | def convert_model_output( method dpm_solver_first_order_update (line 439) | def dpm_solver_first_order_update( method multistep_dpm_solver_second_order_update (line 486) | def multistep_dpm_solver_second_order_update( method multistep_dpm_solver_third_order_update (line 578) | def multistep_dpm_solver_third_order_update( method step (line 633) | def step( method scale_model_input (line 711) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 725) | def add_noise( method __len__ (line 748) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py class DPMSolverMultistepSchedulerState (line 35) | class DPMSolverMultistepSchedulerState: method create (line 53) | def create( class FlaxDPMSolverMultistepSchedulerOutput (line 73) | class FlaxDPMSolverMultistepSchedulerOutput(FlaxSchedulerOutput): class FlaxDPMSolverMultistepScheduler (line 77) | class FlaxDPMSolverMultistepScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 147) | def has_state(self): method __init__ (line 151) | def __init__( method create_state (line 170) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method set_timesteps (line 199) | def set_timesteps( method convert_model_output (line 236) | def convert_model_output( method dpm_solver_first_order_update (line 306) | def dpm_solver_first_order_update( method multistep_dpm_solver_second_order_update (line 341) | def multistep_dpm_solver_second_order_update( method multistep_dpm_solver_third_order_update (line 401) | def multistep_dpm_solver_third_order_update( method step (line 457) | def step( method scale_model_input (line 594) | def scale_model_input( method add_noise (line 612) | def add_noise( method __len__ (line 621) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py function betas_for_alpha_bar (line 29) | def betas_for_alpha_bar( class DPMSolverMultistepInverseScheduler (line 73) | class DPMSolverMultistepInverseScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 147) | def __init__( method set_timesteps (line 212) | def set_timesteps(self, num_inference_steps: int = None, device: Union... method _threshold_sample (line 252) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method _sigma_to_t (line 287) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 311) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method convert_model_output (line 325) | def convert_model_output( method dpm_solver_first_order_update (line 401) | def dpm_solver_first_order_update( method multistep_dpm_solver_second_order_update (line 449) | def multistep_dpm_solver_second_order_update( method multistep_dpm_solver_third_order_update (line 542) | def multistep_dpm_solver_third_order_update( method step (line 597) | def step( method scale_model_input (line 678) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 692) | def add_noise( method __len__ (line 715) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_dpmsolver_sde.py class BatchedBrownianTree (line 27) | class BatchedBrownianTree: method __init__ (line 30) | def __init__(self, x, t0, t1, seed=None, **kwargs): method sort (line 45) | def sort(a, b): method __call__ (line 48) | def __call__(self, t0, t1): class BrownianTreeNoiseSampler (line 54) | class BrownianTreeNoiseSampler: method __init__ (line 69) | def __init__(self, x, sigma_min, sigma_max, seed=None, transform=lambd... method __call__ (line 74) | def __call__(self, sigma, sigma_next): function betas_for_alpha_bar (line 80) | def betas_for_alpha_bar( class DPMSolverSDEScheduler (line 124) | class DPMSolverSDEScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 165) | def __init__( method index_for_timestep (line 203) | def index_for_timestep(self, timestep, schedule_timesteps=None): method init_noise_sigma (line 222) | def init_noise_sigma(self): method scale_model_input (line 229) | def scale_model_input( method set_timesteps (line 249) | def set_timesteps( method _second_order_timesteps (line 321) | def _second_order_timesteps(self, sigmas, log_sigmas): method _sigma_to_t (line 337) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 361) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor) -> torch.Fl... method state_in_first_order (line 375) | def state_in_first_order(self): method step (line 378) | def step( method add_noise (line 483) | def add_noise( method __len__ (line 508) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_dpmsolver_singlestep.py function betas_for_alpha_bar (line 32) | def betas_for_alpha_bar( class DPMSolverSinglestepScheduler (line 76) | class DPMSolverSinglestepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 157) | def __init__( method get_order_list (line 221) | def get_order_list(self, num_inference_steps: int) -> List[int]: method set_timesteps (line 255) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _threshold_sample (line 298) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method _sigma_to_t (line 333) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 357) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method convert_model_output (line 370) | def convert_model_output( method dpm_solver_first_order_update (line 436) | def dpm_solver_first_order_update( method singlestep_dpm_solver_second_order_update (line 468) | def singlestep_dpm_solver_second_order_update( method singlestep_dpm_solver_third_order_update (line 529) | def singlestep_dpm_solver_third_order_update( method singlestep_dpm_solver_update (line 600) | def singlestep_dpm_solver_update( method step (line 637) | def step( method scale_model_input (line 699) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 713) | def add_noise( method __len__ (line 736) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_euler_ancestral_discrete.py class EulerAncestralDiscreteSchedulerOutput (line 32) | class EulerAncestralDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar( class EulerAncestralDiscreteScheduler (line 94) | class EulerAncestralDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 130) | def __init__( method init_noise_sigma (line 170) | def init_noise_sigma(self): method scale_model_input (line 177) | def scale_model_input( method set_timesteps (line 198) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 242) | def step( method add_noise (line 332) | def add_noise( method __len__ (line 357) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_euler_discrete.py class EulerDiscreteSchedulerOutput (line 32) | class EulerDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar( class EulerDiscreteScheduler (line 94) | class EulerDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 138) | def __init__( method init_noise_sigma (line 181) | def init_noise_sigma(self): method scale_model_input (line 188) | def scale_model_input( method set_timesteps (line 211) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _sigma_to_t (line 270) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 294) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method step (line 307) | def step( method add_noise (line 406) | def add_noise( method __len__ (line 431) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_heun_discrete.py function betas_for_alpha_bar (line 27) | def betas_for_alpha_bar( class HeunDiscreteScheduler (line 71) | class HeunDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 114) | def __init__( method index_for_timestep (line 152) | def index_for_timestep(self, timestep, schedule_timesteps=None): method init_noise_sigma (line 171) | def init_noise_sigma(self): method scale_model_input (line 178) | def scale_model_input( method set_timesteps (line 197) | def set_timesteps( method _sigma_to_t (line 266) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 290) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method state_in_first_order (line 304) | def state_in_first_order(self): method step (line 307) | def step( method add_noise (line 400) | def add_noise( method __len__ (line 425) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_ipndm.py class IPNDMScheduler (line 25) | class IPNDMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 44) | def __init__( method set_timesteps (line 61) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 85) | def step( method scale_model_input (line 135) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method _get_prev_sample (line 148) | def _get_prev_sample(self, sample, timestep_index, prev_timestep_index... method __len__ (line 160) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar( class KDPM2AncestralDiscreteScheduler (line 72) | class KDPM2AncestralDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 110) | def __init__( method index_for_timestep (line 143) | def index_for_timestep(self, timestep, schedule_timesteps=None): method init_noise_sigma (line 162) | def init_noise_sigma(self): method scale_model_input (line 169) | def scale_model_input( method set_timesteps (line 192) | def set_timesteps( method sigma_to_t (line 274) | def sigma_to_t(self, sigma): method state_in_first_order (line 298) | def state_in_first_order(self): method step (line 301) | def step( method add_noise (line 394) | def add_noise( method __len__ (line 419) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_k_dpm_2_discrete.py function betas_for_alpha_bar (line 27) | def betas_for_alpha_bar( class KDPM2DiscreteScheduler (line 71) | class KDPM2DiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 109) | def __init__( method index_for_timestep (line 142) | def index_for_timestep(self, timestep, schedule_timesteps=None): method init_noise_sigma (line 161) | def init_noise_sigma(self): method scale_model_input (line 168) | def scale_model_input( method set_timesteps (line 191) | def set_timesteps( method sigma_to_t (line 263) | def sigma_to_t(self, sigma): method state_in_first_order (line 287) | def state_in_first_order(self): method step (line 290) | def step( method add_noise (line 375) | def add_noise( method __len__ (line 400) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_karras_ve.py class KarrasVeOutput (line 28) | class KarrasVeOutput(BaseOutput): class KarrasVeScheduler (line 48) | class KarrasVeScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 83) | def __init__( method scale_model_input (line 100) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 114) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method add_noise_to_input (line 135) | def add_noise_to_input( method step (line 156) | def step( method step_correct (line 194) | def step_correct( method add_noise (line 231) | def add_noise(self, original_samples, noise, timesteps): FILE: 4DoF/diffusers/schedulers/scheduling_karras_ve_flax.py class KarrasVeSchedulerState (line 29) | class KarrasVeSchedulerState: method create (line 36) | def create(cls): class FlaxKarrasVeOutput (line 41) | class FlaxKarrasVeOutput(BaseOutput): class FlaxKarrasVeScheduler (line 59) | class FlaxKarrasVeScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 91) | def has_state(self): method __init__ (line 95) | def __init__( method create_state (line 106) | def create_state(self): method set_timesteps (line 109) | def set_timesteps( method add_noise_to_input (line 137) | def add_noise_to_input( method step (line 163) | def step( method step_correct (line 199) | def step_correct( method add_noise (line 236) | def add_noise(self, state: KarrasVeSchedulerState, original_samples, n... FILE: 4DoF/diffusers/schedulers/scheduling_lms_discrete.py class LMSDiscreteSchedulerOutput (line 30) | class LMSDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 48) | def betas_for_alpha_bar( class LMSDiscreteScheduler (line 92) | class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 133) | def __init__( method init_noise_sigma (line 175) | def init_noise_sigma(self): method scale_model_input (line 182) | def scale_model_input( method get_lms_coefficient (line 203) | def get_lms_coefficient(self, order, t, current_order): method set_timesteps (line 225) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _sigma_to_t (line 279) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 303) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor) -> torch.Fl... method step (line 316) | def step( method add_noise (line 387) | def add_noise( method __len__ (line 412) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_lms_discrete_flax.py class LMSDiscreteSchedulerState (line 33) | class LMSDiscreteSchedulerState: method create (line 46) | def create( class FlaxLMSSchedulerOutput (line 53) | class FlaxLMSSchedulerOutput(FlaxSchedulerOutput): class FlaxLMSDiscreteScheduler (line 57) | class FlaxLMSDiscreteScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 90) | def has_state(self): method __init__ (line 94) | def __init__( method create_state (line 106) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 123) | def scale_model_input(self, state: LMSDiscreteSchedulerState, sample: ... method get_lms_coefficient (line 145) | def get_lms_coefficient(self, state: LMSDiscreteSchedulerState, order,... method set_timesteps (line 167) | def set_timesteps( method step (line 203) | def step( method add_noise (line 268) | def add_noise( method __len__ (line 282) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_pndm.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar( class PNDMScheduler (line 72) | class PNDMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 116) | def __init__( method set_timesteps (line 170) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 228) | def step( method step_prk (line 259) | def step_prk( method step_plms (line 314) | def step_plms( method scale_model_input (line 381) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method _get_prev_sample (line 394) | def _get_prev_sample(self, sample, timestep, prev_timestep, model_outp... method add_noise (line 438) | def add_noise( method __len__ (line 461) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_pndm_flax.py class PNDMSchedulerState (line 35) | class PNDMSchedulerState: method create (line 53) | def create( class FlaxPNDMSchedulerOutput (line 69) | class FlaxPNDMSchedulerOutput(FlaxSchedulerOutput): class FlaxPNDMScheduler (line 73) | class FlaxPNDMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 119) | def has_state(self): method __init__ (line 123) | def __init__( method create_state (line 143) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method set_timesteps (line 167) | def set_timesteps(self, state: PNDMSchedulerState, num_inference_steps... method scale_model_input (line 222) | def scale_model_input( method step (line 239) | def step( method step_prk (line 294) | def step_prk( method step_plms (line 362) | def step_plms( method _get_prev_sample (line 456) | def _get_prev_sample(self, state: PNDMSchedulerState, sample, timestep... method add_noise (line 501) | def add_noise( method __len__ (line 510) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_repaint.py class RePaintSchedulerOutput (line 28) | class RePaintSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 46) | def betas_for_alpha_bar( class RePaintScheduler (line 90) | class RePaintScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 124) | def __init__( method scale_model_input (line 168) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 182) | def set_timesteps( method _get_variance (line 212) | def _get_variance(self, t): method step (line 231) | def step( method undo_step (line 318) | def undo_step(self, sample, timestep, generator=None): method add_noise (line 335) | def add_noise( method __len__ (line 343) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_sde_ve.py class SdeVeOutput (line 29) | class SdeVeOutput(BaseOutput): class ScoreSdeVeScheduler (line 45) | class ScoreSdeVeScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 72) | def __init__( method scale_model_input (line 89) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 103) | def set_timesteps( method set_sigmas (line 120) | def set_sigmas( method get_adjacent_sigma (line 149) | def get_adjacent_sigma(self, timesteps, t): method step_pred (line 156) | def step_pred( method step_correct (line 219) | def step_correct( method add_noise (line 270) | def add_noise( method __len__ (line 287) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_sde_ve_flax.py class ScoreSdeVeSchedulerState (line 29) | class ScoreSdeVeSchedulerState: method create (line 36) | def create(cls): class FlaxSdeVeOutput (line 41) | class FlaxSdeVeOutput(FlaxSchedulerOutput): class FlaxScoreSdeVeScheduler (line 59) | class FlaxScoreSdeVeScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 84) | def has_state(self): method __init__ (line 88) | def __init__( method create_state (line 99) | def create_state(self): method set_timesteps (line 109) | def set_timesteps( method set_sigmas (line 128) | def set_sigmas( method get_adjacent_sigma (line 163) | def get_adjacent_sigma(self, state, timesteps, t): method step_pred (line 166) | def step_pred( method step_correct (line 226) | def step_correct( method __len__ (line 278) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_sde_vp.py class ScoreSdeVpScheduler (line 27) | class ScoreSdeVpScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 45) | def __init__(self, num_train_timesteps=2000, beta_min=0.1, beta_max=20... method set_timesteps (line 50) | def set_timesteps(self, num_inference_steps, device: Union[str, torch.... method step_pred (line 53) | def step_pred(self, score, x, t, generator=None): method __len__ (line 89) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_unclip.py class UnCLIPSchedulerOutput (line 29) | class UnCLIPSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 47) | def betas_for_alpha_bar( class UnCLIPScheduler (line 91) | class UnCLIPScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 121) | def __init__( method scale_model_input (line 148) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 162) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _get_variance (line 179) | def _get_variance(self, t, prev_timestep=None, predicted_variance=None... method step (line 215) | def step( method add_noise (line 327) | def add_noise( FILE: 4DoF/diffusers/schedulers/scheduling_unipc_multistep.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class UniPCMultistepScheduler (line 57) | class UniPCMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 137) | def __init__( method set_timesteps (line 201) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _threshold_sample (line 263) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method convert_model_output (line 297) | def convert_model_output( method multistep_uni_p_bh_update (line 348) | def multistep_uni_p_bh_update( method multistep_uni_c_bh_update (line 453) | def multistep_uni_c_bh_update( method step (line 559) | def step( method scale_model_input (line 643) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 657) | def add_noise( method __len__ (line 680) | def __len__(self): FILE: 4DoF/diffusers/schedulers/scheduling_utils.py class KarrasDiffusionSchedulers (line 32) | class KarrasDiffusionSchedulers(Enum): class SchedulerOutput (line 50) | class SchedulerOutput(BaseOutput): class SchedulerMixin (line 63) | class SchedulerMixin: method from_pretrained (line 78) | def from_pretrained( method save_pretrained (line 149) | def save_pretrained(self, save_directory: Union[str, os.PathLike], pus... method compatibles (line 161) | def compatibles(self): method _get_compatibles (line 171) | def _get_compatibles(cls): FILE: 4DoF/diffusers/schedulers/scheduling_utils_flax.py class FlaxKarrasDiffusionSchedulers (line 34) | class FlaxKarrasDiffusionSchedulers(Enum): class FlaxSchedulerOutput (line 43) | class FlaxSchedulerOutput(BaseOutput): class FlaxSchedulerMixin (line 56) | class FlaxSchedulerMixin: method from_pretrained (line 72) | def from_pretrained( method save_pretrained (line 151) | def save_pretrained(self, save_directory: Union[str, os.PathLike], pus... method compatibles (line 163) | def compatibles(self): method _get_compatibles (line 173) | def _get_compatibles(cls): function broadcast_to_shape_from_left (line 182) | def broadcast_to_shape_from_left(x: jnp.ndarray, shape: Tuple[int]) -> j... function betas_for_alpha_bar (line 187) | def betas_for_alpha_bar(num_diffusion_timesteps: int, max_beta=0.999, dt... class CommonSchedulerState (line 217) | class CommonSchedulerState: method create (line 223) | def create(cls, scheduler): function get_sqrt_alpha_prod (line 257) | def get_sqrt_alpha_prod( function add_noise_common (line 273) | def add_noise_common( function get_velocity_common (line 281) | def get_velocity_common(state: CommonSchedulerState, sample: jnp.ndarray... FILE: 4DoF/diffusers/schedulers/scheduling_vq_diffusion.py class VQDiffusionSchedulerOutput (line 28) | class VQDiffusionSchedulerOutput(BaseOutput): function index_to_log_onehot (line 41) | def index_to_log_onehot(x: torch.LongTensor, num_classes: int) -> torch.... function gumbel_noised (line 62) | def gumbel_noised(logits: torch.FloatTensor, generator: Optional[torch.G... function alpha_schedules (line 72) | def alpha_schedules(num_diffusion_timesteps: int, alpha_cum_start=0.9999... function gamma_schedules (line 88) | def gamma_schedules(num_diffusion_timesteps: int, gamma_cum_start=0.0000... class VQDiffusionScheduler (line 106) | class VQDiffusionScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 144) | def __init__( method set_timesteps (line 190) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 212) | def step( method q_posterior (line 260) | def q_posterior(self, log_p_x_0, x_t, t): method log_Q_t_transitioning_to_known_class (line 379) | def log_Q_t_transitioning_to_known_class( method apply_cumulative_transitions (line 484) | def apply_cumulative_transitions(self, q, t): FILE: 4DoF/diffusers/training_utils.py function set_seed (line 16) | def set_seed(seed: int): class EMAModel (line 30) | class EMAModel: method __init__ (line 35) | def __init__( method from_pretrained (line 117) | def from_pretrained(cls, path, model_cls) -> "EMAModel": method save_pretrained (line 126) | def save_pretrained(self, path): method get_decay (line 141) | def get_decay(self, optimization_step: int) -> float: method step (line 161) | def step(self, parameters: Iterable[torch.nn.Parameter]): method copy_to (line 198) | def copy_to(self, parameters: Iterable[torch.nn.Parameter]) -> None: method to (line 211) | def to(self, device=None, dtype=None) -> None: method state_dict (line 223) | def state_dict(self) -> dict: method store (line 242) | def store(self, parameters: Iterable[torch.nn.Parameter]) -> None: method restore (line 251) | def restore(self, parameters: Iterable[torch.nn.Parameter]) -> None: method load_state_dict (line 269) | def load_state_dict(self, state_dict: dict) -> None: FILE: 4DoF/diffusers/utils/__init__.py function check_min_version (line 113) | def check_min_version(min_version): FILE: 4DoF/diffusers/utils/accelerate_utils.py function apply_forward_hook (line 27) | def apply_forward_hook(method): FILE: 4DoF/diffusers/utils/deprecation_utils.py function deprecate (line 8) | def deprecate(*args, take_from: Optional[Union[Dict, Any]] = None, stand... FILE: 4DoF/diffusers/utils/doc_utils.py function replace_example_docstring (line 20) | def replace_example_docstring(example_docstring): FILE: 4DoF/diffusers/utils/dummy_flax_and_transformers_objects.py class FlaxStableDiffusionControlNetPipeline (line 5) | class FlaxStableDiffusionControlNetPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class FlaxStableDiffusionImg2ImgPipeline (line 20) | class FlaxStableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class FlaxStableDiffusionInpaintPipeline (line 35) | class FlaxStableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class FlaxStableDiffusionPipeline (line 50) | class FlaxStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_flax_objects.py class FlaxControlNetModel (line 5) | class FlaxControlNetModel(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class FlaxModelMixin (line 20) | class FlaxModelMixin(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class FlaxUNet2DConditionModel (line 35) | class FlaxUNet2DConditionModel(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class FlaxAutoencoderKL (line 50) | class FlaxAutoencoderKL(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class FlaxDiffusionPipeline (line 65) | class FlaxDiffusionPipeline(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class FlaxDDIMScheduler (line 80) | class FlaxDDIMScheduler(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class FlaxDDPMScheduler (line 95) | class FlaxDDPMScheduler(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class FlaxDPMSolverMultistepScheduler (line 110) | class FlaxDPMSolverMultistepScheduler(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class FlaxKarrasVeScheduler (line 125) | class FlaxKarrasVeScheduler(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): class FlaxLMSDiscreteScheduler (line 140) | class FlaxLMSDiscreteScheduler(metaclass=DummyObject): method __init__ (line 143) | def __init__(self, *args, **kwargs): method from_config (line 147) | def from_config(cls, *args, **kwargs): method from_pretrained (line 151) | def from_pretrained(cls, *args, **kwargs): class FlaxPNDMScheduler (line 155) | class FlaxPNDMScheduler(metaclass=DummyObject): method __init__ (line 158) | def __init__(self, *args, **kwargs): method from_config (line 162) | def from_config(cls, *args, **kwargs): method from_pretrained (line 166) | def from_pretrained(cls, *args, **kwargs): class FlaxSchedulerMixin (line 170) | class FlaxSchedulerMixin(metaclass=DummyObject): method __init__ (line 173) | def __init__(self, *args, **kwargs): method from_config (line 177) | def from_config(cls, *args, **kwargs): method from_pretrained (line 181) | def from_pretrained(cls, *args, **kwargs): class FlaxScoreSdeVeScheduler (line 185) | class FlaxScoreSdeVeScheduler(metaclass=DummyObject): method __init__ (line 188) | def __init__(self, *args, **kwargs): method from_config (line 192) | def from_config(cls, *args, **kwargs): method from_pretrained (line 196) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_note_seq_objects.py class MidiProcessor (line 5) | class MidiProcessor(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_onnx_objects.py class OnnxRuntimeModel (line 5) | class OnnxRuntimeModel(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_pt_objects.py class AutoencoderKL (line 5) | class AutoencoderKL(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class ControlNetModel (line 20) | class ControlNetModel(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class ModelMixin (line 35) | class ModelMixin(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class PriorTransformer (line 50) | class PriorTransformer(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class T5FilmDecoder (line 65) | class T5FilmDecoder(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class Transformer2DModel (line 80) | class Transformer2DModel(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class UNet1DModel (line 95) | class UNet1DModel(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class UNet2DConditionModel (line 110) | class UNet2DConditionModel(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class UNet2DModel (line 125) | class UNet2DModel(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): class UNet3DConditionModel (line 140) | class UNet3DConditionModel(metaclass=DummyObject): method __init__ (line 143) | def __init__(self, *args, **kwargs): method from_config (line 147) | def from_config(cls, *args, **kwargs): method from_pretrained (line 151) | def from_pretrained(cls, *args, **kwargs): class VQModel (line 155) | class VQModel(metaclass=DummyObject): method __init__ (line 158) | def __init__(self, *args, **kwargs): method from_config (line 162) | def from_config(cls, *args, **kwargs): method from_pretrained (line 166) | def from_pretrained(cls, *args, **kwargs): function get_constant_schedule (line 170) | def get_constant_schedule(*args, **kwargs): function get_constant_schedule_with_warmup (line 174) | def get_constant_schedule_with_warmup(*args, **kwargs): function get_cosine_schedule_with_warmup (line 178) | def get_cosine_schedule_with_warmup(*args, **kwargs): function get_cosine_with_hard_restarts_schedule_with_warmup (line 182) | def get_cosine_with_hard_restarts_schedule_with_warmup(*args, **kwargs): function get_linear_schedule_with_warmup (line 186) | def get_linear_schedule_with_warmup(*args, **kwargs): function get_polynomial_decay_schedule_with_warmup (line 190) | def get_polynomial_decay_schedule_with_warmup(*args, **kwargs): function get_scheduler (line 194) | def get_scheduler(*args, **kwargs): class AudioPipelineOutput (line 198) | class AudioPipelineOutput(metaclass=DummyObject): method __init__ (line 201) | def __init__(self, *args, **kwargs): method from_config (line 205) | def from_config(cls, *args, **kwargs): method from_pretrained (line 209) | def from_pretrained(cls, *args, **kwargs): class ConsistencyModelPipeline (line 213) | class ConsistencyModelPipeline(metaclass=DummyObject): method __init__ (line 216) | def __init__(self, *args, **kwargs): method from_config (line 220) | def from_config(cls, *args, **kwargs): method from_pretrained (line 224) | def from_pretrained(cls, *args, **kwargs): class DanceDiffusionPipeline (line 228) | class DanceDiffusionPipeline(metaclass=DummyObject): method __init__ (line 231) | def __init__(self, *args, **kwargs): method from_config (line 235) | def from_config(cls, *args, **kwargs): method from_pretrained (line 239) | def from_pretrained(cls, *args, **kwargs): class DDIMPipeline (line 243) | class DDIMPipeline(metaclass=DummyObject): method __init__ (line 246) | def __init__(self, *args, **kwargs): method from_config (line 250) | def from_config(cls, *args, **kwargs): method from_pretrained (line 254) | def from_pretrained(cls, *args, **kwargs): class DDPMPipeline (line 258) | class DDPMPipeline(metaclass=DummyObject): method __init__ (line 261) | def __init__(self, *args, **kwargs): method from_config (line 265) | def from_config(cls, *args, **kwargs): method from_pretrained (line 269) | def from_pretrained(cls, *args, **kwargs): class DiffusionPipeline (line 273) | class DiffusionPipeline(metaclass=DummyObject): method __init__ (line 276) | def __init__(self, *args, **kwargs): method from_config (line 280) | def from_config(cls, *args, **kwargs): method from_pretrained (line 284) | def from_pretrained(cls, *args, **kwargs): class DiTPipeline (line 288) | class DiTPipeline(metaclass=DummyObject): method __init__ (line 291) | def __init__(self, *args, **kwargs): method from_config (line 295) | def from_config(cls, *args, **kwargs): method from_pretrained (line 299) | def from_pretrained(cls, *args, **kwargs): class ImagePipelineOutput (line 303) | class ImagePipelineOutput(metaclass=DummyObject): method __init__ (line 306) | def __init__(self, *args, **kwargs): method from_config (line 310) | def from_config(cls, *args, **kwargs): method from_pretrained (line 314) | def from_pretrained(cls, *args, **kwargs): class KarrasVePipeline (line 318) | class KarrasVePipeline(metaclass=DummyObject): method __init__ (line 321) | def __init__(self, *args, **kwargs): method from_config (line 325) | def from_config(cls, *args, **kwargs): method from_pretrained (line 329) | def from_pretrained(cls, *args, **kwargs): class LDMPipeline (line 333) | class LDMPipeline(metaclass=DummyObject): method __init__ (line 336) | def __init__(self, *args, **kwargs): method from_config (line 340) | def from_config(cls, *args, **kwargs): method from_pretrained (line 344) | def from_pretrained(cls, *args, **kwargs): class LDMSuperResolutionPipeline (line 348) | class LDMSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 351) | def __init__(self, *args, **kwargs): method from_config (line 355) | def from_config(cls, *args, **kwargs): method from_pretrained (line 359) | def from_pretrained(cls, *args, **kwargs): class PNDMPipeline (line 363) | class PNDMPipeline(metaclass=DummyObject): method __init__ (line 366) | def __init__(self, *args, **kwargs): method from_config (line 370) | def from_config(cls, *args, **kwargs): method from_pretrained (line 374) | def from_pretrained(cls, *args, **kwargs): class RePaintPipeline (line 378) | class RePaintPipeline(metaclass=DummyObject): method __init__ (line 381) | def __init__(self, *args, **kwargs): method from_config (line 385) | def from_config(cls, *args, **kwargs): method from_pretrained (line 389) | def from_pretrained(cls, *args, **kwargs): class ScoreSdeVePipeline (line 393) | class ScoreSdeVePipeline(metaclass=DummyObject): method __init__ (line 396) | def __init__(self, *args, **kwargs): method from_config (line 400) | def from_config(cls, *args, **kwargs): method from_pretrained (line 404) | def from_pretrained(cls, *args, **kwargs): class CMStochasticIterativeScheduler (line 408) | class CMStochasticIterativeScheduler(metaclass=DummyObject): method __init__ (line 411) | def __init__(self, *args, **kwargs): method from_config (line 415) | def from_config(cls, *args, **kwargs): method from_pretrained (line 419) | def from_pretrained(cls, *args, **kwargs): class DDIMInverseScheduler (line 423) | class DDIMInverseScheduler(metaclass=DummyObject): method __init__ (line 426) | def __init__(self, *args, **kwargs): method from_config (line 430) | def from_config(cls, *args, **kwargs): method from_pretrained (line 434) | def from_pretrained(cls, *args, **kwargs): class DDIMParallelScheduler (line 438) | class DDIMParallelScheduler(metaclass=DummyObject): method __init__ (line 441) | def __init__(self, *args, **kwargs): method from_config (line 445) | def from_config(cls, *args, **kwargs): method from_pretrained (line 449) | def from_pretrained(cls, *args, **kwargs): class DDIMScheduler (line 453) | class DDIMScheduler(metaclass=DummyObject): method __init__ (line 456) | def __init__(self, *args, **kwargs): method from_config (line 460) | def from_config(cls, *args, **kwargs): method from_pretrained (line 464) | def from_pretrained(cls, *args, **kwargs): class DDPMParallelScheduler (line 468) | class DDPMParallelScheduler(metaclass=DummyObject): method __init__ (line 471) | def __init__(self, *args, **kwargs): method from_config (line 475) | def from_config(cls, *args, **kwargs): method from_pretrained (line 479) | def from_pretrained(cls, *args, **kwargs): class DDPMScheduler (line 483) | class DDPMScheduler(metaclass=DummyObject): method __init__ (line 486) | def __init__(self, *args, **kwargs): method from_config (line 490) | def from_config(cls, *args, **kwargs): method from_pretrained (line 494) | def from_pretrained(cls, *args, **kwargs): class DEISMultistepScheduler (line 498) | class DEISMultistepScheduler(metaclass=DummyObject): method __init__ (line 501) | def __init__(self, *args, **kwargs): method from_config (line 505) | def from_config(cls, *args, **kwargs): method from_pretrained (line 509) | def from_pretrained(cls, *args, **kwargs): class DPMSolverMultistepInverseScheduler (line 513) | class DPMSolverMultistepInverseScheduler(metaclass=DummyObject): method __init__ (line 516) | def __init__(self, *args, **kwargs): method from_config (line 520) | def from_config(cls, *args, **kwargs): method from_pretrained (line 524) | def from_pretrained(cls, *args, **kwargs): class DPMSolverMultistepScheduler (line 528) | class DPMSolverMultistepScheduler(metaclass=DummyObject): method __init__ (line 531) | def __init__(self, *args, **kwargs): method from_config (line 535) | def from_config(cls, *args, **kwargs): method from_pretrained (line 539) | def from_pretrained(cls, *args, **kwargs): class DPMSolverSinglestepScheduler (line 543) | class DPMSolverSinglestepScheduler(metaclass=DummyObject): method __init__ (line 546) | def __init__(self, *args, **kwargs): method from_config (line 550) | def from_config(cls, *args, **kwargs): method from_pretrained (line 554) | def from_pretrained(cls, *args, **kwargs): class EulerAncestralDiscreteScheduler (line 558) | class EulerAncestralDiscreteScheduler(metaclass=DummyObject): method __init__ (line 561) | def __init__(self, *args, **kwargs): method from_config (line 565) | def from_config(cls, *args, **kwargs): method from_pretrained (line 569) | def from_pretrained(cls, *args, **kwargs): class EulerDiscreteScheduler (line 573) | class EulerDiscreteScheduler(metaclass=DummyObject): method __init__ (line 576) | def __init__(self, *args, **kwargs): method from_config (line 580) | def from_config(cls, *args, **kwargs): method from_pretrained (line 584) | def from_pretrained(cls, *args, **kwargs): class HeunDiscreteScheduler (line 588) | class HeunDiscreteScheduler(metaclass=DummyObject): method __init__ (line 591) | def __init__(self, *args, **kwargs): method from_config (line 595) | def from_config(cls, *args, **kwargs): method from_pretrained (line 599) | def from_pretrained(cls, *args, **kwargs): class IPNDMScheduler (line 603) | class IPNDMScheduler(metaclass=DummyObject): method __init__ (line 606) | def __init__(self, *args, **kwargs): method from_config (line 610) | def from_config(cls, *args, **kwargs): method from_pretrained (line 614) | def from_pretrained(cls, *args, **kwargs): class KarrasVeScheduler (line 618) | class KarrasVeScheduler(metaclass=DummyObject): method __init__ (line 621) | def __init__(self, *args, **kwargs): method from_config (line 625) | def from_config(cls, *args, **kwargs): method from_pretrained (line 629) | def from_pretrained(cls, *args, **kwargs): class KDPM2AncestralDiscreteScheduler (line 633) | class KDPM2AncestralDiscreteScheduler(metaclass=DummyObject): method __init__ (line 636) | def __init__(self, *args, **kwargs): method from_config (line 640) | def from_config(cls, *args, **kwargs): method from_pretrained (line 644) | def from_pretrained(cls, *args, **kwargs): class KDPM2DiscreteScheduler (line 648) | class KDPM2DiscreteScheduler(metaclass=DummyObject): method __init__ (line 651) | def __init__(self, *args, **kwargs): method from_config (line 655) | def from_config(cls, *args, **kwargs): method from_pretrained (line 659) | def from_pretrained(cls, *args, **kwargs): class PNDMScheduler (line 663) | class PNDMScheduler(metaclass=DummyObject): method __init__ (line 666) | def __init__(self, *args, **kwargs): method from_config (line 670) | def from_config(cls, *args, **kwargs): method from_pretrained (line 674) | def from_pretrained(cls, *args, **kwargs): class RePaintScheduler (line 678) | class RePaintScheduler(metaclass=DummyObject): method __init__ (line 681) | def __init__(self, *args, **kwargs): method from_config (line 685) | def from_config(cls, *args, **kwargs): method from_pretrained (line 689) | def from_pretrained(cls, *args, **kwargs): class SchedulerMixin (line 693) | class SchedulerMixin(metaclass=DummyObject): method __init__ (line 696) | def __init__(self, *args, **kwargs): method from_config (line 700) | def from_config(cls, *args, **kwargs): method from_pretrained (line 704) | def from_pretrained(cls, *args, **kwargs): class ScoreSdeVeScheduler (line 708) | class ScoreSdeVeScheduler(metaclass=DummyObject): method __init__ (line 711) | def __init__(self, *args, **kwargs): method from_config (line 715) | def from_config(cls, *args, **kwargs): method from_pretrained (line 719) | def from_pretrained(cls, *args, **kwargs): class UnCLIPScheduler (line 723) | class UnCLIPScheduler(metaclass=DummyObject): method __init__ (line 726) | def __init__(self, *args, **kwargs): method from_config (line 730) | def from_config(cls, *args, **kwargs): method from_pretrained (line 734) | def from_pretrained(cls, *args, **kwargs): class UniPCMultistepScheduler (line 738) | class UniPCMultistepScheduler(metaclass=DummyObject): method __init__ (line 741) | def __init__(self, *args, **kwargs): method from_config (line 745) | def from_config(cls, *args, **kwargs): method from_pretrained (line 749) | def from_pretrained(cls, *args, **kwargs): class VQDiffusionScheduler (line 753) | class VQDiffusionScheduler(metaclass=DummyObject): method __init__ (line 756) | def __init__(self, *args, **kwargs): method from_config (line 760) | def from_config(cls, *args, **kwargs): method from_pretrained (line 764) | def from_pretrained(cls, *args, **kwargs): class EMAModel (line 768) | class EMAModel(metaclass=DummyObject): method __init__ (line 771) | def __init__(self, *args, **kwargs): method from_config (line 775) | def from_config(cls, *args, **kwargs): method from_pretrained (line 779) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_torch_and_librosa_objects.py class AudioDiffusionPipeline (line 5) | class AudioDiffusionPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class Mel (line 20) | class Mel(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_torch_and_scipy_objects.py class LMSDiscreteScheduler (line 5) | class LMSDiscreteScheduler(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_torch_and_torchsde_objects.py class DPMSolverSDEScheduler (line 5) | class DPMSolverSDEScheduler(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_torch_and_transformers_and_invisible_watermark_objects.py class StableDiffusionXLImg2ImgPipeline (line 5) | class StableDiffusionXLImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionXLPipeline (line 20) | class StableDiffusionXLPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_torch_and_transformers_and_k_diffusion_objects.py class StableDiffusionKDiffusionPipeline (line 5) | class StableDiffusionKDiffusionPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_torch_and_transformers_and_onnx_objects.py class OnnxStableDiffusionImg2ImgPipeline (line 5) | class OnnxStableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionInpaintPipeline (line 20) | class OnnxStableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionInpaintPipelineLegacy (line 35) | class OnnxStableDiffusionInpaintPipelineLegacy(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionPipeline (line 50) | class OnnxStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionUpscalePipeline (line 65) | class OnnxStableDiffusionUpscalePipeline(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionOnnxPipeline (line 80) | class StableDiffusionOnnxPipeline(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_torch_and_transformers_objects.py class AltDiffusionImg2ImgPipeline (line 5) | class AltDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class AltDiffusionPipeline (line 20) | class AltDiffusionPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class AudioLDMPipeline (line 35) | class AudioLDMPipeline(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class CycleDiffusionPipeline (line 50) | class CycleDiffusionPipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class IFImg2ImgPipeline (line 65) | class IFImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class IFImg2ImgSuperResolutionPipeline (line 80) | class IFImg2ImgSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class IFInpaintingPipeline (line 95) | class IFInpaintingPipeline(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class IFInpaintingSuperResolutionPipeline (line 110) | class IFInpaintingSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class IFPipeline (line 125) | class IFPipeline(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): class IFSuperResolutionPipeline (line 140) | class IFSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 143) | def __init__(self, *args, **kwargs): method from_config (line 147) | def from_config(cls, *args, **kwargs): method from_pretrained (line 151) | def from_pretrained(cls, *args, **kwargs): class ImageTextPipelineOutput (line 155) | class ImageTextPipelineOutput(metaclass=DummyObject): method __init__ (line 158) | def __init__(self, *args, **kwargs): method from_config (line 162) | def from_config(cls, *args, **kwargs): method from_pretrained (line 166) | def from_pretrained(cls, *args, **kwargs): class KandinskyImg2ImgPipeline (line 170) | class KandinskyImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 173) | def __init__(self, *args, **kwargs): method from_config (line 177) | def from_config(cls, *args, **kwargs): method from_pretrained (line 181) | def from_pretrained(cls, *args, **kwargs): class KandinskyInpaintPipeline (line 185) | class KandinskyInpaintPipeline(metaclass=DummyObject): method __init__ (line 188) | def __init__(self, *args, **kwargs): method from_config (line 192) | def from_config(cls, *args, **kwargs): method from_pretrained (line 196) | def from_pretrained(cls, *args, **kwargs): class KandinskyPipeline (line 200) | class KandinskyPipeline(metaclass=DummyObject): method __init__ (line 203) | def __init__(self, *args, **kwargs): method from_config (line 207) | def from_config(cls, *args, **kwargs): method from_pretrained (line 211) | def from_pretrained(cls, *args, **kwargs): class KandinskyPriorPipeline (line 215) | class KandinskyPriorPipeline(metaclass=DummyObject): method __init__ (line 218) | def __init__(self, *args, **kwargs): method from_config (line 222) | def from_config(cls, *args, **kwargs): method from_pretrained (line 226) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22ControlnetImg2ImgPipeline (line 230) | class KandinskyV22ControlnetImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 233) | def __init__(self, *args, **kwargs): method from_config (line 237) | def from_config(cls, *args, **kwargs): method from_pretrained (line 241) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22ControlnetPipeline (line 245) | class KandinskyV22ControlnetPipeline(metaclass=DummyObject): method __init__ (line 248) | def __init__(self, *args, **kwargs): method from_config (line 252) | def from_config(cls, *args, **kwargs): method from_pretrained (line 256) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22Img2ImgPipeline (line 260) | class KandinskyV22Img2ImgPipeline(metaclass=DummyObject): method __init__ (line 263) | def __init__(self, *args, **kwargs): method from_config (line 267) | def from_config(cls, *args, **kwargs): method from_pretrained (line 271) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22InpaintPipeline (line 275) | class KandinskyV22InpaintPipeline(metaclass=DummyObject): method __init__ (line 278) | def __init__(self, *args, **kwargs): method from_config (line 282) | def from_config(cls, *args, **kwargs): method from_pretrained (line 286) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22Pipeline (line 290) | class KandinskyV22Pipeline(metaclass=DummyObject): method __init__ (line 293) | def __init__(self, *args, **kwargs): method from_config (line 297) | def from_config(cls, *args, **kwargs): method from_pretrained (line 301) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22PriorEmb2EmbPipeline (line 305) | class KandinskyV22PriorEmb2EmbPipeline(metaclass=DummyObject): method __init__ (line 308) | def __init__(self, *args, **kwargs): method from_config (line 312) | def from_config(cls, *args, **kwargs): method from_pretrained (line 316) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22PriorPipeline (line 320) | class KandinskyV22PriorPipeline(metaclass=DummyObject): method __init__ (line 323) | def __init__(self, *args, **kwargs): method from_config (line 327) | def from_config(cls, *args, **kwargs): method from_pretrained (line 331) | def from_pretrained(cls, *args, **kwargs): class LDMTextToImagePipeline (line 335) | class LDMTextToImagePipeline(metaclass=DummyObject): method __init__ (line 338) | def __init__(self, *args, **kwargs): method from_config (line 342) | def from_config(cls, *args, **kwargs): method from_pretrained (line 346) | def from_pretrained(cls, *args, **kwargs): class PaintByExamplePipeline (line 350) | class PaintByExamplePipeline(metaclass=DummyObject): method __init__ (line 353) | def __init__(self, *args, **kwargs): method from_config (line 357) | def from_config(cls, *args, **kwargs): method from_pretrained (line 361) | def from_pretrained(cls, *args, **kwargs): class SemanticStableDiffusionPipeline (line 365) | class SemanticStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 368) | def __init__(self, *args, **kwargs): method from_config (line 372) | def from_config(cls, *args, **kwargs): method from_pretrained (line 376) | def from_pretrained(cls, *args, **kwargs): class ShapEImg2ImgPipeline (line 380) | class ShapEImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 383) | def __init__(self, *args, **kwargs): method from_config (line 387) | def from_config(cls, *args, **kwargs): method from_pretrained (line 391) | def from_pretrained(cls, *args, **kwargs): class ShapEPipeline (line 395) | class ShapEPipeline(metaclass=DummyObject): method __init__ (line 398) | def __init__(self, *args, **kwargs): method from_config (line 402) | def from_config(cls, *args, **kwargs): method from_pretrained (line 406) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionAttendAndExcitePipeline (line 410) | class StableDiffusionAttendAndExcitePipeline(metaclass=DummyObject): method __init__ (line 413) | def __init__(self, *args, **kwargs): method from_config (line 417) | def from_config(cls, *args, **kwargs): method from_pretrained (line 421) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionControlNetImg2ImgPipeline (line 425) | class StableDiffusionControlNetImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 428) | def __init__(self, *args, **kwargs): method from_config (line 432) | def from_config(cls, *args, **kwargs): method from_pretrained (line 436) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionControlNetInpaintPipeline (line 440) | class StableDiffusionControlNetInpaintPipeline(metaclass=DummyObject): method __init__ (line 443) | def __init__(self, *args, **kwargs): method from_config (line 447) | def from_config(cls, *args, **kwargs): method from_pretrained (line 451) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionControlNetPipeline (line 455) | class StableDiffusionControlNetPipeline(metaclass=DummyObject): method __init__ (line 458) | def __init__(self, *args, **kwargs): method from_config (line 462) | def from_config(cls, *args, **kwargs): method from_pretrained (line 466) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionDepth2ImgPipeline (line 470) | class StableDiffusionDepth2ImgPipeline(metaclass=DummyObject): method __init__ (line 473) | def __init__(self, *args, **kwargs): method from_config (line 477) | def from_config(cls, *args, **kwargs): method from_pretrained (line 481) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionDiffEditPipeline (line 485) | class StableDiffusionDiffEditPipeline(metaclass=DummyObject): method __init__ (line 488) | def __init__(self, *args, **kwargs): method from_config (line 492) | def from_config(cls, *args, **kwargs): method from_pretrained (line 496) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionImageVariationPipeline (line 500) | class StableDiffusionImageVariationPipeline(metaclass=DummyObject): method __init__ (line 503) | def __init__(self, *args, **kwargs): method from_config (line 507) | def from_config(cls, *args, **kwargs): method from_pretrained (line 511) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionImg2ImgPipeline (line 515) | class StableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 518) | def __init__(self, *args, **kwargs): method from_config (line 522) | def from_config(cls, *args, **kwargs): method from_pretrained (line 526) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInpaintPipeline (line 530) | class StableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 533) | def __init__(self, *args, **kwargs): method from_config (line 537) | def from_config(cls, *args, **kwargs): method from_pretrained (line 541) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInpaintPipelineLegacy (line 545) | class StableDiffusionInpaintPipelineLegacy(metaclass=DummyObject): method __init__ (line 548) | def __init__(self, *args, **kwargs): method from_config (line 552) | def from_config(cls, *args, **kwargs): method from_pretrained (line 556) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInstructPix2PixPipeline (line 560) | class StableDiffusionInstructPix2PixPipeline(metaclass=DummyObject): method __init__ (line 563) | def __init__(self, *args, **kwargs): method from_config (line 567) | def from_config(cls, *args, **kwargs): method from_pretrained (line 571) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionLatentUpscalePipeline (line 575) | class StableDiffusionLatentUpscalePipeline(metaclass=DummyObject): method __init__ (line 578) | def __init__(self, *args, **kwargs): method from_config (line 582) | def from_config(cls, *args, **kwargs): method from_pretrained (line 586) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionLDM3DPipeline (line 590) | class StableDiffusionLDM3DPipeline(metaclass=DummyObject): method __init__ (line 593) | def __init__(self, *args, **kwargs): method from_config (line 597) | def from_config(cls, *args, **kwargs): method from_pretrained (line 601) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionModelEditingPipeline (line 605) | class StableDiffusionModelEditingPipeline(metaclass=DummyObject): method __init__ (line 608) | def __init__(self, *args, **kwargs): method from_config (line 612) | def from_config(cls, *args, **kwargs): method from_pretrained (line 616) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPanoramaPipeline (line 620) | class StableDiffusionPanoramaPipeline(metaclass=DummyObject): method __init__ (line 623) | def __init__(self, *args, **kwargs): method from_config (line 627) | def from_config(cls, *args, **kwargs): method from_pretrained (line 631) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionParadigmsPipeline (line 635) | class StableDiffusionParadigmsPipeline(metaclass=DummyObject): method __init__ (line 638) | def __init__(self, *args, **kwargs): method from_config (line 642) | def from_config(cls, *args, **kwargs): method from_pretrained (line 646) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPipeline (line 650) | class StableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 653) | def __init__(self, *args, **kwargs): method from_config (line 657) | def from_config(cls, *args, **kwargs): method from_pretrained (line 661) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPipelineSafe (line 665) | class StableDiffusionPipelineSafe(metaclass=DummyObject): method __init__ (line 668) | def __init__(self, *args, **kwargs): method from_config (line 672) | def from_config(cls, *args, **kwargs): method from_pretrained (line 676) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPix2PixZeroPipeline (line 680) | class StableDiffusionPix2PixZeroPipeline(metaclass=DummyObject): method __init__ (line 683) | def __init__(self, *args, **kwargs): method from_config (line 687) | def from_config(cls, *args, **kwargs): method from_pretrained (line 691) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionSAGPipeline (line 695) | class StableDiffusionSAGPipeline(metaclass=DummyObject): method __init__ (line 698) | def __init__(self, *args, **kwargs): method from_config (line 702) | def from_config(cls, *args, **kwargs): method from_pretrained (line 706) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionUpscalePipeline (line 710) | class StableDiffusionUpscalePipeline(metaclass=DummyObject): method __init__ (line 713) | def __init__(self, *args, **kwargs): method from_config (line 717) | def from_config(cls, *args, **kwargs): method from_pretrained (line 721) | def from_pretrained(cls, *args, **kwargs): class StableUnCLIPImg2ImgPipeline (line 725) | class StableUnCLIPImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 728) | def __init__(self, *args, **kwargs): method from_config (line 732) | def from_config(cls, *args, **kwargs): method from_pretrained (line 736) | def from_pretrained(cls, *args, **kwargs): class StableUnCLIPPipeline (line 740) | class StableUnCLIPPipeline(metaclass=DummyObject): method __init__ (line 743) | def __init__(self, *args, **kwargs): method from_config (line 747) | def from_config(cls, *args, **kwargs): method from_pretrained (line 751) | def from_pretrained(cls, *args, **kwargs): class TextToVideoSDPipeline (line 755) | class TextToVideoSDPipeline(metaclass=DummyObject): method __init__ (line 758) | def __init__(self, *args, **kwargs): method from_config (line 762) | def from_config(cls, *args, **kwargs): method from_pretrained (line 766) | def from_pretrained(cls, *args, **kwargs): class TextToVideoZeroPipeline (line 770) | class TextToVideoZeroPipeline(metaclass=DummyObject): method __init__ (line 773) | def __init__(self, *args, **kwargs): method from_config (line 777) | def from_config(cls, *args, **kwargs): method from_pretrained (line 781) | def from_pretrained(cls, *args, **kwargs): class UnCLIPImageVariationPipeline (line 785) | class UnCLIPImageVariationPipeline(metaclass=DummyObject): method __init__ (line 788) | def __init__(self, *args, **kwargs): method from_config (line 792) | def from_config(cls, *args, **kwargs): method from_pretrained (line 796) | def from_pretrained(cls, *args, **kwargs): class UnCLIPPipeline (line 800) | class UnCLIPPipeline(metaclass=DummyObject): method __init__ (line 803) | def __init__(self, *args, **kwargs): method from_config (line 807) | def from_config(cls, *args, **kwargs): method from_pretrained (line 811) | def from_pretrained(cls, *args, **kwargs): class UniDiffuserModel (line 815) | class UniDiffuserModel(metaclass=DummyObject): method __init__ (line 818) | def __init__(self, *args, **kwargs): method from_config (line 822) | def from_config(cls, *args, **kwargs): method from_pretrained (line 826) | def from_pretrained(cls, *args, **kwargs): class UniDiffuserPipeline (line 830) | class UniDiffuserPipeline(metaclass=DummyObject): method __init__ (line 833) | def __init__(self, *args, **kwargs): method from_config (line 837) | def from_config(cls, *args, **kwargs): method from_pretrained (line 841) | def from_pretrained(cls, *args, **kwargs): class UniDiffuserTextDecoder (line 845) | class UniDiffuserTextDecoder(metaclass=DummyObject): method __init__ (line 848) | def __init__(self, *args, **kwargs): method from_config (line 852) | def from_config(cls, *args, **kwargs): method from_pretrained (line 856) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionDualGuidedPipeline (line 860) | class VersatileDiffusionDualGuidedPipeline(metaclass=DummyObject): method __init__ (line 863) | def __init__(self, *args, **kwargs): method from_config (line 867) | def from_config(cls, *args, **kwargs): method from_pretrained (line 871) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionImageVariationPipeline (line 875) | class VersatileDiffusionImageVariationPipeline(metaclass=DummyObject): method __init__ (line 878) | def __init__(self, *args, **kwargs): method from_config (line 882) | def from_config(cls, *args, **kwargs): method from_pretrained (line 886) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionPipeline (line 890) | class VersatileDiffusionPipeline(metaclass=DummyObject): method __init__ (line 893) | def __init__(self, *args, **kwargs): method from_config (line 897) | def from_config(cls, *args, **kwargs): method from_pretrained (line 901) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionTextToImagePipeline (line 905) | class VersatileDiffusionTextToImagePipeline(metaclass=DummyObject): method __init__ (line 908) | def __init__(self, *args, **kwargs): method from_config (line 912) | def from_config(cls, *args, **kwargs): method from_pretrained (line 916) | def from_pretrained(cls, *args, **kwargs): class VideoToVideoSDPipeline (line 920) | class VideoToVideoSDPipeline(metaclass=DummyObject): method __init__ (line 923) | def __init__(self, *args, **kwargs): method from_config (line 927) | def from_config(cls, *args, **kwargs): method from_pretrained (line 931) | def from_pretrained(cls, *args, **kwargs): class VQDiffusionPipeline (line 935) | class VQDiffusionPipeline(metaclass=DummyObject): method __init__ (line 938) | def __init__(self, *args, **kwargs): method from_config (line 942) | def from_config(cls, *args, **kwargs): method from_pretrained (line 946) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dummy_transformers_and_torch_and_note_seq_objects.py class SpectrogramDiffusionPipeline (line 5) | class SpectrogramDiffusionPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 4DoF/diffusers/utils/dynamic_modules_utils.py function get_diffusers_versions (line 43) | def get_diffusers_versions(): function init_hf_modules (line 49) | def init_hf_modules(): function create_dynamic_module (line 64) | def create_dynamic_module(name: Union[str, os.PathLike]): function get_relative_imports (line 79) | def get_relative_imports(module_file): function get_relative_import_files (line 97) | def get_relative_import_files(module_file): function check_imports (line 126) | def check_imports(filename): function get_class_in_module (line 158) | def get_class_in_module(class_name, module_path): function find_pipeline_class (line 170) | def find_pipeline_class(loaded_module): function get_cached_module_file (line 197) | def get_cached_module_file( function get_class_from_dynamic_module (line 369) | def get_class_from_dynamic_module( FILE: 4DoF/diffusers/utils/hub_utils.py function http_user_agent (line 68) | def http_user_agent(user_agent: Union[Dict, str, None] = None) -> str: function get_full_repo_name (line 92) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function create_model_card (line 102) | def create_model_card(args, model_name): function extract_commit_hash (line 151) | def extract_commit_hash(resolved_file: Optional[str], commit_hash: Optio... function move_cache (line 175) | def move_cache(old_cache_dir: Optional[str] = None, new_cache_dir: Optio... function _add_variant (line 237) | def _add_variant(weights_name: str, variant: Optional[str] = None) -> str: function _get_model_file (line 246) | def _get_model_file( FILE: 4DoF/diffusers/utils/import_utils.py function is_torch_available (line 305) | def is_torch_available(): function is_safetensors_available (line 309) | def is_safetensors_available(): function is_tf_available (line 313) | def is_tf_available(): function is_flax_available (line 317) | def is_flax_available(): function is_transformers_available (line 321) | def is_transformers_available(): function is_inflect_available (line 325) | def is_inflect_available(): function is_unidecode_available (line 329) | def is_unidecode_available(): function is_onnx_available (line 333) | def is_onnx_available(): function is_opencv_available (line 337) | def is_opencv_available(): function is_scipy_available (line 341) | def is_scipy_available(): function is_librosa_available (line 345) | def is_librosa_available(): function is_xformers_available (line 349) | def is_xformers_available(): function is_accelerate_available (line 353) | def is_accelerate_available(): function is_k_diffusion_available (line 357) | def is_k_diffusion_available(): function is_note_seq_available (line 361) | def is_note_seq_available(): function is_wandb_available (line 365) | def is_wandb_available(): function is_omegaconf_available (line 369) | def is_omegaconf_available(): function is_tensorboard_available (line 373) | def is_tensorboard_available(): function is_compel_available (line 377) | def is_compel_available(): function is_ftfy_available (line 381) | def is_ftfy_available(): function is_bs4_available (line 385) | def is_bs4_available(): function is_torchsde_available (line 389) | def is_torchsde_available(): function is_invisible_watermark_available (line 393) | def is_invisible_watermark_available(): function requires_backends (line 536) | def requires_backends(obj, backends): class DummyObject (line 567) | class DummyObject(type): method __getattr__ (line 573) | def __getattr__(cls, key): function compare_versions (line 580) | def compare_versions(library_or_version: Union[str, Version], operation:... function is_torch_version (line 600) | def is_torch_version(operation: str, version: str): function is_transformers_version (line 612) | def is_transformers_version(operation: str, version: str): function is_accelerate_version (line 626) | def is_accelerate_version(operation: str, version: str): function is_k_diffusion_version (line 640) | def is_k_diffusion_version(operation: str, version: str): class OptionalDependencyNotAvailable (line 654) | class OptionalDependencyNotAvailable(BaseException): FILE: 4DoF/diffusers/utils/logging.py function _get_default_logging_level (line 52) | def _get_default_logging_level(): function _get_library_name (line 69) | def _get_library_name() -> str: function _get_library_root_logger (line 73) | def _get_library_root_logger() -> logging.Logger: function _configure_library_root_logger (line 77) | def _configure_library_root_logger() -> None: function _reset_library_root_logger (line 94) | def _reset_library_root_logger() -> None: function get_log_levels_dict (line 107) | def get_log_levels_dict(): function get_logger (line 111) | def get_logger(name: Optional[str] = None) -> logging.Logger: function get_verbosity (line 125) | def get_verbosity() -> int: function set_verbosity (line 145) | def set_verbosity(verbosity: int) -> None: function set_verbosity_info (line 164) | def set_verbosity_info(): function set_verbosity_warning (line 169) | def set_verbosity_warning(): function set_verbosity_debug (line 174) | def set_verbosity_debug(): function set_verbosity_error (line 179) | def set_verbosity_error(): function disable_default_handler (line 184) | def disable_default_handler() -> None: function enable_default_handler (line 193) | def enable_default_handler() -> None: function add_handler (line 202) | def add_handler(handler: logging.Handler) -> None: function remove_handler (line 211) | def remove_handler(handler: logging.Handler) -> None: function disable_propagation (line 220) | def disable_propagation() -> None: function enable_propagation (line 229) | def enable_propagation() -> None: function enable_explicit_format (line 239) | def enable_explicit_format() -> None: function reset_format (line 254) | def reset_format() -> None: function warning_advice (line 266) | def warning_advice(self, *args, **kwargs): class EmptyTqdm (line 280) | class EmptyTqdm: method __init__ (line 283) | def __init__(self, *args, **kwargs): # pylint: disable=unused-argument method __iter__ (line 286) | def __iter__(self): method __getattr__ (line 289) | def __getattr__(self, _): method __enter__ (line 297) | def __enter__(self): method __exit__ (line 300) | def __exit__(self, type_, value, traceback): class _tqdm_cls (line 304) | class _tqdm_cls: method __call__ (line 305) | def __call__(self, *args, **kwargs): method set_lock (line 311) | def set_lock(self, *args, **kwargs): method get_lock (line 316) | def get_lock(self): function is_progress_bar_enabled (line 324) | def is_progress_bar_enabled() -> bool: function enable_progress_bar (line 330) | def enable_progress_bar(): function disable_progress_bar (line 336) | def disable_progress_bar(): FILE: 4DoF/diffusers/utils/outputs.py function is_tensor (line 27) | def is_tensor(x): class BaseOutput (line 40) | class BaseOutput(OrderedDict): method __post_init__ (line 54) | def __post_init__(self): method __delitem__ (line 73) | def __delitem__(self, *args, **kwargs): method setdefault (line 76) | def setdefault(self, *args, **kwargs): method pop (line 79) | def pop(self, *args, **kwargs): method update (line 82) | def update(self, *args, **kwargs): method __getitem__ (line 85) | def __getitem__(self, k): method __setattr__ (line 92) | def __setattr__(self, name, value): method __setitem__ (line 98) | def __setitem__(self, key, value): method to_tuple (line 104) | def to_tuple(self) -> Tuple[Any]: FILE: 4DoF/diffusers/utils/pil_utils.py function pt_to_pil (line 25) | def pt_to_pil(images): function numpy_to_pil (line 35) | def numpy_to_pil(images): FILE: 4DoF/diffusers/utils/testing_utils.py function torch_all_close (line 64) | def torch_all_close(a, b, *args, **kwargs): function print_tensor_test (line 72) | def print_tensor_test(tensor, filename="test_corrections.txt", expected_... function get_tests_dir (line 87) | def get_tests_dir(append_path=None): function parse_flag_from_env (line 108) | def parse_flag_from_env(key, default=False): function floats_tensor (line 128) | def floats_tensor(shape, scale=1.0, rng=None, name=None): function slow (line 144) | def slow(test_case): function nightly (line 154) | def nightly(test_case): function require_torch (line 164) | def require_torch(test_case): function require_torch_2 (line 171) | def require_torch_2(test_case): function require_torch_gpu (line 180) | def require_torch_gpu(test_case): function skip_mps (line 187) | def skip_mps(test_case): function require_flax (line 192) | def require_flax(test_case): function require_compel (line 199) | def require_compel(test_case): function require_onnxruntime (line 207) | def require_onnxruntime(test_case): function require_note_seq (line 214) | def require_note_seq(test_case): function require_torchsde (line 221) | def require_torchsde(test_case): function load_numpy (line 228) | def load_numpy(arry: Union[str, np.ndarray], local_path: Optional[str] =... function load_pt (line 255) | def load_pt(url: str): function load_image (line 262) | def load_image(image: Union[str, PIL.Image.Image]) -> PIL.Image.Image: function preprocess_image (line 293) | def preprocess_image(image: PIL.Image, batch_size: int): function export_to_gif (line 303) | def export_to_gif(image: List[PIL.Image.Image], output_gif_path: str = N... function export_to_video (line 318) | def export_to_video(video_frames: List[np.ndarray], output_video_path: s... function load_hf_numpy (line 335) | def load_hf_numpy(path) -> np.ndarray: function pytest_addoption_shared (line 350) | def pytest_addoption_shared(parser): function pytest_terminal_summary_main (line 369) | def pytest_terminal_summary_main(tr, id): function run_test_in_subprocess (line 499) | def run_test_in_subprocess(test_case, target_func, inputs=None, timeout=... class CaptureLogger (line 542) | class CaptureLogger: method __init__ (line 563) | def __init__(self, logger): method __enter__ (line 569) | def __enter__(self): method __exit__ (line 573) | def __exit__(self, *exc): method __repr__ (line 577) | def __repr__(self): function enable_full_determinism (line 581) | def enable_full_determinism(): function disable_full_determinism (line 599) | def disable_full_determinism(): FILE: 4DoF/diffusers/utils/torch_utils.py function maybe_allow_in_graph (line 32) | def maybe_allow_in_graph(cls): function randn_tensor (line 36) | def randn_tensor( function is_compiled_module (line 80) | def is_compiled_module(module): FILE: 4DoF/pipeline_zero1to3.py class CCProjection (line 50) | class CCProjection(ModelMixin, ConfigMixin): method __init__ (line 52) | def __init__(self, in_channel=772, out_channel=768): method forward (line 58) | def forward(self, x): class CLIPProjection (line 61) | class CLIPProjection(ModelMixin, ConfigMixin): method __init__ (line 63) | def __init__(self, in_channel, out_channel): method forward (line 70) | def forward(self, x): class CNLayernorm (line 74) | class CNLayernorm(ModelMixin, ConfigMixin): method __init__ (line 76) | def __init__(self, in_channel, eps): method forward (line 81) | def forward(self, x): class Zero1to3StableDiffusionPipeline (line 84) | class Zero1to3StableDiffusionPipeline(DiffusionPipeline): method __init__ (line 113) | def __init__( method enable_vae_slicing (line 204) | def enable_vae_slicing(self): method disable_vae_slicing (line 213) | def disable_vae_slicing(self): method enable_vae_tiling (line 220) | def enable_vae_tiling(self): method disable_vae_tiling (line 229) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 236) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 261) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 290) | def _execution_device(self): method _encode_prompt (line 307) | def _encode_prompt( method _encode_image (line 445) | def _encode_image(self, image, device, num_images_per_prompt, do_class... method run_safety_checker (line 499) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 509) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 517) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 534) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 556) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_img_latents (line 573) | def prepare_img_latents(self, image, batch_size, dtype, device, genera... method __call__ (line 636) | def __call__( FILE: 4DoF/train_eschernet.py function image_grid (line 72) | def image_grid(imgs, rows, cols): function log_validation (line 83) | def log_validation(validation_dataloader, vae, image_encoder, feature_ex... function parse_args (line 210) | def parse_args(input_args=None): function _encode_image (line 494) | def _encode_image(feature_extractor, image_encoder, image, device, dtype... function main (line 507) | def main(args): FILE: 4DoF/unet_2d_condition.py class UNet2DConditionOutput (line 55) | class UNet2DConditionOutput(BaseOutput): class UNet2DConditionModel (line 67) | class UNet2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoade... method __init__ (line 158) | def __init__( method attn_processors (line 574) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 597) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 631) | def set_default_attn_processor(self): method set_attention_slice (line 637) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 702) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 706) | def forward( FILE: 6DoF/CN_encoder.py class CN_encoder (line 6) | class CN_encoder(ConvNextV2Model): method __init__ (line 7) | def __init__(self, config): method forward (line 10) | def forward( FILE: 6DoF/dataset.py class ObjaverseDataLoader (line 15) | class ObjaverseDataLoader(): method __init__ (line 16) | def __init__(self, root_dir, batch_size, total_view=12, num_workers=4): method train_dataloader (line 27) | def train_dataloader(self): method val_dataloader (line 34) | def val_dataloader(self): function get_pose (line 40) | def get_pose(transformation): class ObjaverseData (line 44) | class ObjaverseData(Dataset): method __init__ (line 45) | def __init__(self, method __len__ (line 87) | def __len__(self): method get_pose (line 90) | def get_pose(self, transformation): method load_im (line 95) | def load_im(self, path, color): method __getitem__ (line 108) | def __getitem__(self, index): method process_im (line 174) | def process_im(self, im): FILE: 6DoF/diffusers/commands/__init__.py class BaseDiffusersCLICommand (line 19) | class BaseDiffusersCLICommand(ABC): method register_subcommand (line 22) | def register_subcommand(parser: ArgumentParser): method run (line 26) | def run(self): FILE: 6DoF/diffusers/commands/diffusers_cli.py function main (line 21) | def main(): FILE: 6DoF/diffusers/commands/env.py function info_command_factory (line 25) | def info_command_factory(_): class EnvironmentCommand (line 29) | class EnvironmentCommand(BaseDiffusersCLICommand): method register_subcommand (line 31) | def register_subcommand(parser: ArgumentParser): method run (line 35) | def run(self): method format_dict (line 83) | def format_dict(d): FILE: 6DoF/diffusers/configuration_utils.py class FrozenDict (line 50) | class FrozenDict(OrderedDict): method __init__ (line 51) | def __init__(self, *args, **kwargs): method __delitem__ (line 59) | def __delitem__(self, *args, **kwargs): method setdefault (line 62) | def setdefault(self, *args, **kwargs): method pop (line 65) | def pop(self, *args, **kwargs): method update (line 68) | def update(self, *args, **kwargs): method __setattr__ (line 71) | def __setattr__(self, name, value): method __setitem__ (line 76) | def __setitem__(self, name, value): class ConfigMixin (line 82) | class ConfigMixin: method register_to_config (line 104) | def register_to_config(self, **kwargs): method __getattr__ (line 121) | def __getattr__(self, name: str) -> Any: method save_config (line 139) | def save_config(self, save_directory: Union[str, os.PathLike], push_to... method from_config (line 160) | def from_config(cls, config: Union[FrozenDict, Dict[str, Any]] = None,... method get_config_dict (line 246) | def get_config_dict(cls, *args, **kwargs): method load_config (line 255) | def load_config( method _get_init_keys (line 421) | def _get_init_keys(cls): method extract_init_dict (line 425) | def extract_init_dict(cls, config_dict, **kwargs): method _dict_from_json_file (line 513) | def _dict_from_json_file(cls, json_file: Union[str, os.PathLike]): method __repr__ (line 518) | def __repr__(self): method config (line 522) | def config(self) -> Dict[str, Any]: method to_json_string (line 531) | def to_json_string(self) -> str: method to_json_file (line 557) | def to_json_file(self, json_file_path: Union[str, os.PathLike]): function register_to_config (line 569) | def register_to_config(init): function flax_register_to_config (line 619) | def flax_register_to_config(cls): FILE: 6DoF/diffusers/dependency_versions_check.py function dep_version_check (line 46) | def dep_version_check(pkg, hint=None): FILE: 6DoF/diffusers/experimental/rl/value_guided_sampling.py class ValueGuidedRLPipeline (line 25) | class ValueGuidedRLPipeline(DiffusionPipeline): method __init__ (line 42) | def __init__( method normalize (line 70) | def normalize(self, x_in, key): method de_normalize (line 73) | def de_normalize(self, x_in, key): method to_torch (line 76) | def to_torch(self, x_in): method reset_x0 (line 83) | def reset_x0(self, x_in, cond, act_dim): method run_diffusion (line 88) | def run_diffusion(self, x, conditions, n_guide_steps, scale): method __call__ (line 121) | def __call__(self, obs, batch_size=64, planning_horizon=32, n_guide_st... FILE: 6DoF/diffusers/image_processor.py class VaeImageProcessor (line 27) | class VaeImageProcessor(ConfigMixin): method __init__ (line 48) | def __init__( method numpy_to_pil (line 59) | def numpy_to_pil(images: np.ndarray) -> PIL.Image.Image: method pil_to_numpy (line 75) | def pil_to_numpy(images: Union[List[PIL.Image.Image], PIL.Image.Image]... method numpy_to_pt (line 87) | def numpy_to_pt(images: np.ndarray) -> torch.FloatTensor: method pt_to_numpy (line 98) | def pt_to_numpy(images: torch.FloatTensor) -> np.ndarray: method normalize (line 106) | def normalize(images): method denormalize (line 113) | def denormalize(images): method convert_to_rgb (line 120) | def convert_to_rgb(image: PIL.Image.Image) -> PIL.Image.Image: method resize (line 127) | def resize( method preprocess (line 147) | def preprocess( method postprocess (line 215) | def postprocess( class VaeImageProcessorLDM3D (line 255) | class VaeImageProcessorLDM3D(VaeImageProcessor): method __init__ (line 273) | def __init__( method numpy_to_pil (line 283) | def numpy_to_pil(images): method rgblike_to_depthmap (line 299) | def rgblike_to_depthmap(image): method numpy_to_depth (line 309) | def numpy_to_depth(self, images): method postprocess (line 329) | def postprocess( FILE: 6DoF/diffusers/loaders.py class AttnProcsLayers (line 70) | class AttnProcsLayers(torch.nn.Module): method __init__ (line 71) | def __init__(self, state_dict: Dict[str, torch.Tensor]): class UNet2DConditionLoadersMixin (line 112) | class UNet2DConditionLoadersMixin: method load_attn_procs (line 116) | def load_attn_procs(self, pretrained_model_name_or_path_or_dict: Union... method save_attn_procs (line 329) | def save_attn_procs( class TextualInversionLoaderMixin (line 407) | class TextualInversionLoaderMixin: method maybe_convert_prompt (line 412) | def maybe_convert_prompt(self, prompt: Union[str, List[str]], tokenize... method _maybe_convert_prompt (line 439) | def _maybe_convert_prompt(self, prompt: str, tokenizer: "PreTrainedTok... method load_textual_inversion (line 469) | def load_textual_inversion( class LoraLoaderMixin (line 738) | class LoraLoaderMixin: method load_lora_weights (line 746) | def load_lora_weights(self, pretrained_model_name_or_path_or_dict: Uni... method lora_scale (line 909) | def lora_scale(self) -> float: method text_encoder_lora_attn_procs (line 915) | def text_encoder_lora_attn_procs(self): method _remove_text_encoder_monkey_patch (line 920) | def _remove_text_encoder_monkey_patch(self): method _modify_text_encoder (line 933) | def _modify_text_encoder(self, attn_processors: Dict[str, LoRAAttnProc... method _lora_attn_processor_attr_to_text_encoder_attr (line 970) | def _lora_attn_processor_attr_to_text_encoder_attr(self): method _load_text_encoder_attn_procs (line 978) | def _load_text_encoder_attn_procs( method save_lora_weights (line 1151) | def save_lora_weights( method _convert_kohya_lora_to_diffusers (line 1228) | def _convert_kohya_lora_to_diffusers(self, state_dict): class FromSingleFileMixin (line 1279) | class FromSingleFileMixin: method from_ckpt (line 1285) | def from_ckpt(cls, *args, **kwargs): method from_single_file (line 1291) | def from_single_file(cls, pretrained_model_link_or_path, **kwargs): FILE: 6DoF/diffusers/models/activations.py function get_activation (line 4) | def get_activation(act_fn): FILE: 6DoF/diffusers/models/attention.py class BasicTransformerBlock (line 27) | class BasicTransformerBlock(nn.Module): method __init__ (line 48) | def __init__( method set_chunk_feed_forward (line 126) | def set_chunk_feed_forward(self, chunk_size: Optional[int], dim: int): method forward (line 131) | def forward( class FeedForward (line 210) | class FeedForward(nn.Module): method __init__ (line 223) | def __init__( method forward (line 256) | def forward(self, hidden_states): class GELU (line 262) | class GELU(nn.Module): method __init__ (line 267) | def __init__(self, dim_in: int, dim_out: int, approximate: str = "none"): method gelu (line 272) | def gelu(self, gate): method forward (line 278) | def forward(self, hidden_states): class GEGLU (line 284) | class GEGLU(nn.Module): method __init__ (line 293) | def __init__(self, dim_in: int, dim_out: int): method gelu (line 297) | def gelu(self, gate): method forward (line 303) | def forward(self, hidden_states): class ApproximateGELU (line 308) | class ApproximateGELU(nn.Module): method __init__ (line 315) | def __init__(self, dim_in: int, dim_out: int): method forward (line 319) | def forward(self, x): class AdaLayerNorm (line 324) | class AdaLayerNorm(nn.Module): method __init__ (line 329) | def __init__(self, embedding_dim, num_embeddings): method forward (line 336) | def forward(self, x, timestep): class AdaLayerNormZero (line 343) | class AdaLayerNormZero(nn.Module): method __init__ (line 348) | def __init__(self, embedding_dim, num_embeddings): method forward (line 357) | def forward(self, x, timestep, class_labels, hidden_dtype=None): class AdaGroupNorm (line 364) | class AdaGroupNorm(nn.Module): method __init__ (line 369) | def __init__( method forward (line 383) | def forward(self, x, emb): FILE: 6DoF/diffusers/models/attention_flax.py function _query_chunk_attention (line 23) | def _query_chunk_attention(query, key, value, precision, key_chunk_size:... function jax_memory_efficient_attention (line 74) | def jax_memory_efficient_attention( class FlaxAttention (line 119) | class FlaxAttention(nn.Module): method setup (line 145) | def setup(self): method reshape_heads_to_batch_dim (line 157) | def reshape_heads_to_batch_dim(self, tensor): method reshape_batch_dim_to_heads (line 165) | def reshape_batch_dim_to_heads(self, tensor): method __call__ (line 173) | def __call__(self, hidden_states, context=None, deterministic=True): class FlaxBasicTransformerBlock (line 221) | class FlaxBasicTransformerBlock(nn.Module): method setup (line 251) | def setup(self): method __call__ (line 266) | def __call__(self, hidden_states, context, deterministic=True): class FlaxTransformer2DModel (line 288) | class FlaxTransformer2DModel(nn.Module): method setup (line 322) | def setup(self): method __call__ (line 363) | def __call__(self, hidden_states, context, deterministic=True): class FlaxFeedForward (line 388) | class FlaxFeedForward(nn.Module): method setup (line 409) | def setup(self): method __call__ (line 415) | def __call__(self, hidden_states, deterministic=True): class FlaxGEGLU (line 421) | class FlaxGEGLU(nn.Module): method setup (line 438) | def setup(self): method __call__ (line 443) | def __call__(self, hidden_states, deterministic=True): FILE: 6DoF/diffusers/models/attention_processor.py function cape_embed (line 36) | def cape_embed(f, P): class Attention (line 43) | class Attention(nn.Module): method __init__ (line 58) | def __init__( method set_use_memory_efficient_attention_xformers (line 173) | def set_use_memory_efficient_attention_xformers( method set_attention_slice (line 291) | def set_attention_slice(self, slice_size): method set_processor (line 312) | def set_processor(self, processor: "AttnProcessor"): method forward (line 325) | def forward(self, hidden_states, encoder_hidden_states=None, attention... method batch_to_head_dim (line 337) | def batch_to_head_dim(self, tensor): method head_to_batch_dim (line 344) | def head_to_batch_dim(self, tensor, out_dim=3): method get_attention_scores (line 355) | def get_attention_scores(self, query, key, attention_mask=None): method prepare_attention_mask (line 389) | def prepare_attention_mask(self, attention_mask, target_length, batch_... method norm_encoder_hidden_states (line 430) | def norm_encoder_hidden_states(self, encoder_hidden_states): class AttnProcessor (line 450) | class AttnProcessor: method __call__ (line 455) | def __call__( class LoRALinearLayer (line 516) | class LoRALinearLayer(nn.Module): method __init__ (line 517) | def __init__(self, in_features, out_features, rank=4, network_alpha=No... method forward (line 533) | def forward(self, hidden_states): class LoRAAttnProcessor (line 546) | class LoRAAttnProcessor(nn.Module): method __init__ (line 561) | def __init__(self, hidden_size, cross_attention_dim=None, rank=4, netw... method __call__ (line 573) | def __call__( class CustomDiffusionAttnProcessor (line 629) | class CustomDiffusionAttnProcessor(nn.Module): method __init__ (line 648) | def __init__( method __call__ (line 674) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class AttnAddedKVProcessor (line 725) | class AttnAddedKVProcessor: method __call__ (line 731) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class AttnAddedKVProcessor2_0 (line 779) | class AttnAddedKVProcessor2_0: method __init__ (line 785) | def __init__(self): method __call__ (line 791) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class LoRAAttnAddedKVProcessor (line 842) | class LoRAAttnAddedKVProcessor(nn.Module): method __init__ (line 857) | def __init__(self, hidden_size, cross_attention_dim=None, rank=4, netw... method __call__ (line 871) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class XFormersAttnAddedKVProcessor (line 923) | class XFormersAttnAddedKVProcessor: method __init__ (line 935) | def __init__(self, attention_op: Optional[Callable] = None): method __call__ (line 938) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class XFormersAttnProcessor (line 988) | class XFormersAttnProcessor: method __init__ (line 1000) | def __init__(self, attention_op: Optional[Callable] = None): method __call__ (line 1003) | def __call__( class AttnProcessor2_0 (line 1105) | class AttnProcessor2_0: method __init__ (line 1110) | def __init__(self): method __call__ (line 1114) | def __call__( class LoRAXFormersAttnProcessor (line 1189) | class LoRAXFormersAttnProcessor(nn.Module): method __init__ (line 1210) | def __init__( method __call__ (line 1225) | def __call__( class LoRAAttnProcessor2_0 (line 1282) | class LoRAAttnProcessor2_0(nn.Module): method __init__ (line 1298) | def __init__(self, hidden_size, cross_attention_dim=None, rank=4, netw... method __call__ (line 1312) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class CustomDiffusionXFormersAttnProcessor (line 1373) | class CustomDiffusionXFormersAttnProcessor(nn.Module): method __init__ (line 1396) | def __init__( method __call__ (line 1424) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class SlicedAttnProcessor (line 1480) | class SlicedAttnProcessor: method __init__ (line 1490) | def __init__(self, slice_size): method __call__ (line 1493) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class SlicedAttnAddedKVProcessor (line 1561) | class SlicedAttnAddedKVProcessor: method __init__ (line 1571) | def __init__(self, slice_size): method __call__ (line 1574) | def __call__(self, attn: "Attention", hidden_states, encoder_hidden_st... class SpatialNorm (line 1664) | class SpatialNorm(nn.Module): method __init__ (line 1669) | def __init__( method forward (line 1679) | def forward(self, f, zq): FILE: 6DoF/diffusers/models/autoencoder_kl.py class AutoencoderKLOutput (line 28) | class AutoencoderKLOutput(BaseOutput): class AutoencoderKL (line 41) | class AutoencoderKL(ModelMixin, ConfigMixin): method __init__ (line 72) | def __init__( method _set_gradient_checkpointing (line 127) | def _set_gradient_checkpointing(self, module, value=False): method enable_tiling (line 131) | def enable_tiling(self, use_tiling: bool = True): method disable_tiling (line 139) | def disable_tiling(self): method enable_slicing (line 146) | def enable_slicing(self): method disable_slicing (line 153) | def disable_slicing(self): method attn_processors (line 162) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 186) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 221) | def set_default_attn_processor(self): method encode (line 228) | def encode(self, x: torch.FloatTensor, return_dict: bool = True) -> Au... method _decode (line 246) | def _decode(self, z: torch.FloatTensor, return_dict: bool = True) -> U... method decode (line 259) | def decode(self, z: torch.FloatTensor, return_dict: bool = True) -> Un... method blend_v (line 271) | def blend_v(self, a, b, blend_extent): method blend_h (line 277) | def blend_h(self, a, b, blend_extent): method tiled_encode (line 283) | def tiled_encode(self, x: torch.FloatTensor, return_dict: bool = True)... method tiled_decode (line 337) | def tiled_decode(self, z: torch.FloatTensor, return_dict: bool = True)... method forward (line 385) | def forward( FILE: 6DoF/diffusers/models/controlnet.py class ControlNetOutput (line 39) | class ControlNetOutput(BaseOutput): class ControlNetConditioningEmbedding (line 58) | class ControlNetConditioningEmbedding(nn.Module): method __init__ (line 68) | def __init__( method forward (line 90) | def forward(self, conditioning): class ControlNetModel (line 103) | class ControlNetModel(ModelMixin, ConfigMixin): method __init__ (line 159) | def __init__( method from_unet (line 344) | def from_unet( method attn_processors (line 400) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 424) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 459) | def set_default_attn_processor(self): method set_attention_slice (line 466) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 531) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 535) | def forward( function zero_module (line 702) | def zero_module(module): FILE: 6DoF/diffusers/models/controlnet_flax.py class FlaxControlNetOutput (line 34) | class FlaxControlNetOutput(BaseOutput): class FlaxControlNetConditioningEmbedding (line 47) | class FlaxControlNetConditioningEmbedding(nn.Module): method setup (line 52) | def setup(self): method __call__ (line 90) | def __call__(self, conditioning): class FlaxControlNetModel (line 104) | class FlaxControlNetModel(nn.Module, FlaxModelMixin, ConfigMixin): method init_weights (line 171) | def init_weights(self, rng: jax.random.KeyArray) -> FrozenDict: method setup (line 185) | def setup(self): method __call__ (line 313) | def __call__( FILE: 6DoF/diffusers/models/cross_attention.py class CrossAttention (line 41) | class CrossAttention(Attention): method __init__ (line 42) | def __init__(self, *args, **kwargs): class CrossAttnProcessor (line 48) | class CrossAttnProcessor(AttnProcessorRename): method __init__ (line 49) | def __init__(self, *args, **kwargs): class LoRACrossAttnProcessor (line 55) | class LoRACrossAttnProcessor(LoRAAttnProcessor): method __init__ (line 56) | def __init__(self, *args, **kwargs): class CrossAttnAddedKVProcessor (line 62) | class CrossAttnAddedKVProcessor(AttnAddedKVProcessor): method __init__ (line 63) | def __init__(self, *args, **kwargs): class XFormersCrossAttnProcessor (line 69) | class XFormersCrossAttnProcessor(XFormersAttnProcessor): method __init__ (line 70) | def __init__(self, *args, **kwargs): class LoRAXFormersCrossAttnProcessor (line 76) | class LoRAXFormersCrossAttnProcessor(LoRAXFormersAttnProcessor): method __init__ (line 77) | def __init__(self, *args, **kwargs): class SlicedCrossAttnProcessor (line 83) | class SlicedCrossAttnProcessor(SlicedAttnProcessor): method __init__ (line 84) | def __init__(self, *args, **kwargs): class SlicedCrossAttnAddedKVProcessor (line 90) | class SlicedCrossAttnAddedKVProcessor(SlicedAttnAddedKVProcessor): method __init__ (line 91) | def __init__(self, *args, **kwargs): FILE: 6DoF/diffusers/models/dual_transformer_2d.py class DualTransformer2DModel (line 21) | class DualTransformer2DModel(nn.Module): method __init__ (line 48) | def __init__( method forward (line 97) | def forward( FILE: 6DoF/diffusers/models/embeddings.py function get_timestep_embedding (line 24) | def get_timestep_embedding( function get_2d_sincos_pos_embed (line 67) | def get_2d_sincos_pos_embed(embed_dim, grid_size, cls_token=False, extra... function get_2d_sincos_pos_embed_from_grid (line 84) | def get_2d_sincos_pos_embed_from_grid(embed_dim, grid): function get_1d_sincos_pos_embed_from_grid (line 96) | def get_1d_sincos_pos_embed_from_grid(embed_dim, pos): class PatchEmbed (line 117) | class PatchEmbed(nn.Module): method __init__ (line 120) | def __init__( method forward (line 148) | def forward(self, latent): class TimestepEmbedding (line 157) | class TimestepEmbedding(nn.Module): method __init__ (line 158) | def __init__( method forward (line 189) | def forward(self, sample, condition=None): class Timesteps (line 204) | class Timesteps(nn.Module): method __init__ (line 205) | def __init__(self, num_channels: int, flip_sin_to_cos: bool, downscale... method forward (line 211) | def forward(self, timesteps): class GaussianFourierProjection (line 221) | class GaussianFourierProjection(nn.Module): method __init__ (line 224) | def __init__( method forward (line 238) | def forward(self, x): class ImagePositionalEmbeddings (line 251) | class ImagePositionalEmbeddings(nn.Module): method __init__ (line 275) | def __init__( method forward (line 293) | def forward(self, index): class LabelEmbedding (line 316) | class LabelEmbedding(nn.Module): method __init__ (line 326) | def __init__(self, num_classes, hidden_size, dropout_prob): method token_drop (line 333) | def token_drop(self, labels, force_drop_ids=None): method forward (line 344) | def forward(self, labels: torch.LongTensor, force_drop_ids=None): class TextImageProjection (line 352) | class TextImageProjection(nn.Module): method __init__ (line 353) | def __init__( method forward (line 366) | def forward(self, text_embeds: torch.FloatTensor, image_embeds: torch.... class ImageProjection (line 379) | class ImageProjection(nn.Module): method __init__ (line 380) | def __init__( method forward (line 392) | def forward(self, image_embeds: torch.FloatTensor): class CombinedTimestepLabelEmbeddings (line 402) | class CombinedTimestepLabelEmbeddings(nn.Module): method __init__ (line 403) | def __init__(self, num_classes, embedding_dim, class_dropout_prob=0.1): method forward (line 410) | def forward(self, timestep, class_labels, hidden_dtype=None): class TextTimeEmbedding (line 421) | class TextTimeEmbedding(nn.Module): method __init__ (line 422) | def __init__(self, encoder_dim: int, time_embed_dim: int, num_heads: i... method forward (line 429) | def forward(self, hidden_states): class TextImageTimeEmbedding (line 437) | class TextImageTimeEmbedding(nn.Module): method __init__ (line 438) | def __init__(self, text_embed_dim: int = 768, image_embed_dim: int = 7... method forward (line 444) | def forward(self, text_embeds: torch.FloatTensor, image_embeds: torch.... class ImageTimeEmbedding (line 455) | class ImageTimeEmbedding(nn.Module): method __init__ (line 456) | def __init__(self, image_embed_dim: int = 768, time_embed_dim: int = 1... method forward (line 461) | def forward(self, image_embeds: torch.FloatTensor): class ImageHintTimeEmbedding (line 468) | class ImageHintTimeEmbedding(nn.Module): method __init__ (line 469) | def __init__(self, image_embed_dim: int = 768, time_embed_dim: int = 1... method forward (line 491) | def forward(self, image_embeds: torch.FloatTensor, hint: torch.FloatTe... class AttentionPooling (line 499) | class AttentionPooling(nn.Module): method __init__ (line 502) | def __init__(self, num_heads, embed_dim, dtype=None): method forward (line 512) | def forward(self, x): FILE: 6DoF/diffusers/models/embeddings_flax.py function get_sinusoidal_embeddings (line 20) | def get_sinusoidal_embeddings( class FlaxTimestepEmbedding (line 58) | class FlaxTimestepEmbedding(nn.Module): method __call__ (line 72) | def __call__(self, temb): class FlaxTimesteps (line 79) | class FlaxTimesteps(nn.Module): method __call__ (line 92) | def __call__(self, timesteps): FILE: 6DoF/diffusers/models/modeling_flax_pytorch_utils.py function rename_key (line 28) | def rename_key(key): function rename_key_and_reshape_tensor (line 43) | def rename_key_and_reshape_tensor(pt_tuple_key, pt_tensor, random_flax_s... function convert_pytorch_state_dict_to_flax (line 90) | def convert_pytorch_state_dict_to_flax(pt_state_dict, flax_model, init_k... FILE: 6DoF/diffusers/models/modeling_flax_utils.py class FlaxModelMixin (line 45) | class FlaxModelMixin: method _from_config (line 59) | def _from_config(cls, config, **kwargs): method _cast_floating_to (line 65) | def _cast_floating_to(self, params: Union[Dict, FrozenDict], dtype: jn... method to_bf16 (line 89) | def to_bf16(self, params: Union[Dict, FrozenDict], mask: Any = None): method to_fp32 (line 128) | def to_fp32(self, params: Union[Dict, FrozenDict], mask: Any = None): method to_fp16 (line 155) | def to_fp16(self, params: Union[Dict, FrozenDict], mask: Any = None): method init_weights (line 194) | def init_weights(self, rng: jax.random.KeyArray) -> Dict: method from_pretrained (line 198) | def from_pretrained( method save_pretrained (line 495) | def save_pretrained( FILE: 6DoF/diffusers/models/modeling_pytorch_flax_utils.py function load_flax_checkpoint_in_pytorch_model (line 37) | def load_flax_checkpoint_in_pytorch_model(pt_model, model_file): function load_flax_weights_in_pytorch_model (line 58) | def load_flax_weights_in_pytorch_model(pt_model, flax_state): FILE: 6DoF/diffusers/models/modeling_utils.py function get_parameter_device (line 63) | def get_parameter_device(parameter: torch.nn.Module): function get_parameter_dtype (line 79) | def get_parameter_dtype(parameter: torch.nn.Module): function load_state_dict (line 101) | def load_state_dict(checkpoint_file: Union[str, os.PathLike], variant: O... function _load_state_dict_into_model (line 132) | def _load_state_dict_into_model(model_to_load, state_dict): class ModelMixin (line 153) | class ModelMixin(torch.nn.Module): method __init__ (line 167) | def __init__(self): method __getattr__ (line 170) | def __getattr__(self, name: str) -> Any: method is_gradient_checkpointing (line 189) | def is_gradient_checkpointing(self) -> bool: method enable_gradient_checkpointing (line 195) | def enable_gradient_checkpointing(self): method disable_gradient_checkpointing (line 204) | def disable_gradient_checkpointing(self): method set_use_memory_efficient_attention_xformers (line 212) | def set_use_memory_efficient_attention_xformers( method enable_xformers_memory_efficient_attention (line 229) | def enable_xformers_memory_efficient_attention(self, attention_op: Opt... method disable_xformers_memory_efficient_attention (line 265) | def disable_xformers_memory_efficient_attention(self): method save_pretrained (line 271) | def save_pretrained( method from_pretrained (line 332) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method _load_pretrained_model (line 732) | def _load_pretrained_model( method device (line 836) | def device(self) -> device: method dtype (line 844) | def dtype(self) -> torch.dtype: method num_parameters (line 850) | def num_parameters(self, only_trainable: bool = False, exclude_embeddi... method _convert_deprecated_attention_blocks (line 888) | def _convert_deprecated_attention_blocks(self, state_dict): method _temp_convert_self_to_deprecated_attention_blocks (line 932) | def _temp_convert_self_to_deprecated_attention_blocks(self): method _undo_temp_convert_self_to_deprecated_attention_blocks (line 959) | def _undo_temp_convert_self_to_deprecated_attention_blocks(self): FILE: 6DoF/diffusers/models/prior_transformer.py class PriorTransformerOutput (line 17) | class PriorTransformerOutput(BaseOutput): class PriorTransformer (line 29) | class PriorTransformer(ModelMixin, ConfigMixin): method __init__ (line 67) | def __init__( method attn_processors (line 164) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 188) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 223) | def set_default_attn_processor(self): method forward (line 229) | def forward( method post_process_latents (line 362) | def post_process_latents(self, prior_latents): FILE: 6DoF/diffusers/models/resnet.py class Upsample1D (line 28) | class Upsample1D(nn.Module): method __init__ (line 42) | def __init__(self, channels, use_conv=False, use_conv_transpose=False,... method forward (line 56) | def forward(self, inputs): class Downsample1D (line 69) | class Downsample1D(nn.Module): method __init__ (line 83) | def __init__(self, channels, use_conv=False, out_channels=None, paddin... method forward (line 98) | def forward(self, inputs): class Upsample2D (line 103) | class Upsample2D(nn.Module): method __init__ (line 117) | def __init__(self, channels, use_conv=False, use_conv_transpose=False,... method forward (line 137) | def forward(self, hidden_states, output_size=None): class Downsample2D (line 175) | class Downsample2D(nn.Module): method __init__ (line 189) | def __init__(self, channels, use_conv=False, out_channels=None, paddin... method forward (line 213) | def forward(self, hidden_states): class FirUpsample2D (line 225) | class FirUpsample2D(nn.Module): method __init__ (line 239) | def __init__(self, channels=None, out_channels=None, use_conv=False, f... method _upsample_2d (line 248) | def _upsample_2d(self, hidden_states, weight=None, kernel=None, factor... method forward (line 328) | def forward(self, hidden_states): class FirDownsample2D (line 338) | class FirDownsample2D(nn.Module): method __init__ (line 352) | def __init__(self, channels=None, out_channels=None, use_conv=False, f... method _downsample_2d (line 361) | def _downsample_2d(self, hidden_states, weight=None, kernel=None, fact... method forward (line 415) | def forward(self, hidden_states): class KDownsample2D (line 426) | class KDownsample2D(nn.Module): method __init__ (line 427) | def __init__(self, pad_mode="reflect"): method forward (line 434) | def forward(self, inputs): class KUpsample2D (line 443) | class KUpsample2D(nn.Module): method __init__ (line 444) | def __init__(self, pad_mode="reflect"): method forward (line 451) | def forward(self, inputs): class ResnetBlock2D (line 460) | class ResnetBlock2D(nn.Module): method __init__ (line 491) | def __init__( method forward (line 590) | def forward(self, input_tensor, temb): function rearrange_dims (line 644) | def rearrange_dims(tensor): class Conv1dBlock (line 655) | class Conv1dBlock(nn.Module): method __init__ (line 660) | def __init__(self, inp_channels, out_channels, kernel_size, n_groups=8): method forward (line 667) | def forward(self, inputs): class ResidualTemporalBlock1D (line 677) | class ResidualTemporalBlock1D(nn.Module): method __init__ (line 678) | def __init__(self, inp_channels, out_channels, embed_dim, kernel_size=5): method forward (line 690) | def forward(self, inputs, t): function upsample_2d (line 706) | def upsample_2d(hidden_states, kernel=None, factor=2, gain=1): function downsample_2d (line 743) | def downsample_2d(hidden_states, kernel=None, factor=2, gain=1): function upfirdn2d_native (line 778) | def upfirdn2d_native(tensor, kernel, up=1, down=1, pad=(0, 0)): class TemporalConvLayer (line 822) | class TemporalConvLayer(nn.Module): method __init__ (line 828) | def __init__(self, in_dim, out_dim=None, dropout=0.0): method forward (line 861) | def forward(self, hidden_states, num_frames=1): FILE: 6DoF/diffusers/models/resnet_flax.py class FlaxUpsample2D (line 19) | class FlaxUpsample2D(nn.Module): method setup (line 23) | def setup(self): method __call__ (line 32) | def __call__(self, hidden_states): class FlaxDownsample2D (line 43) | class FlaxDownsample2D(nn.Module): method setup (line 47) | def setup(self): method __call__ (line 56) | def __call__(self, hidden_states): class FlaxResnetBlock2D (line 63) | class FlaxResnetBlock2D(nn.Module): method setup (line 70) | def setup(self): method __call__ (line 106) | def __call__(self, hidden_states, temb, deterministic=True): FILE: 6DoF/diffusers/models/t5_film_transformer.py class T5FilmDecoder (line 25) | class T5FilmDecoder(ModelMixin, ConfigMixin): method __init__ (line 27) | def __init__( method encoder_decoder_mask (line 66) | def encoder_decoder_mask(self, query_input, key_input): method forward (line 70) | def forward(self, encodings_and_masks, decoder_input_tokens, decoder_n... class DecoderLayer (line 127) | class DecoderLayer(nn.Module): method __init__ (line 128) | def __init__(self, d_model, d_kv, num_heads, d_ff, dropout_rate, layer... method forward (line 153) | def forward( class T5LayerSelfAttentionCond (line 185) | class T5LayerSelfAttentionCond(nn.Module): method __init__ (line 186) | def __init__(self, d_model, d_kv, num_heads, dropout_rate): method forward (line 193) | def forward( class T5LayerCrossAttention (line 213) | class T5LayerCrossAttention(nn.Module): method __init__ (line 214) | def __init__(self, d_model, d_kv, num_heads, dropout_rate, layer_norm_... method forward (line 220) | def forward( class T5LayerFFCond (line 236) | class T5LayerFFCond(nn.Module): method __init__ (line 237) | def __init__(self, d_model, d_ff, dropout_rate, layer_norm_epsilon): method forward (line 244) | def forward(self, hidden_states, conditioning_emb=None): class T5DenseGatedActDense (line 254) | class T5DenseGatedActDense(nn.Module): method __init__ (line 255) | def __init__(self, d_model, d_ff, dropout_rate): method forward (line 263) | def forward(self, hidden_states): class T5LayerNorm (line 273) | class T5LayerNorm(nn.Module): method __init__ (line 274) | def __init__(self, hidden_size, eps=1e-6): method forward (line 282) | def forward(self, hidden_states): class NewGELUActivation (line 298) | class NewGELUActivation(nn.Module): method forward (line 304) | def forward(self, input: torch.Tensor) -> torch.Tensor: class T5FiLMLayer (line 308) | class T5FiLMLayer(nn.Module): method __init__ (line 313) | def __init__(self, in_features, out_features): method forward (line 317) | def forward(self, x, conditioning_emb): FILE: 6DoF/diffusers/models/transformer_2d.py class Transformer2DModelOutput (line 30) | class Transformer2DModelOutput(BaseOutput): class Transformer2DModel (line 43) | class Transformer2DModel(ModelMixin, ConfigMixin): method __init__ (line 72) | def __init__( method forward (line 206) | def forward( FILE: 6DoF/diffusers/models/transformer_temporal.py class TransformerTemporalModelOutput (line 27) | class TransformerTemporalModelOutput(BaseOutput): class TransformerTemporalModel (line 39) | class TransformerTemporalModel(ModelMixin, ConfigMixin): method __init__ (line 61) | def __init__( method forward (line 107) | def forward( FILE: 6DoF/diffusers/models/unet_1d.py class UNet1DOutput (line 29) | class UNet1DOutput(BaseOutput): class UNet1DModel (line 41) | class UNet1DModel(ModelMixin, ConfigMixin): method __init__ (line 75) | def __init__( method forward (line 195) | def forward( FILE: 6DoF/diffusers/models/unet_1d_blocks.py class DownResnetBlock1D (line 24) | class DownResnetBlock1D(nn.Module): method __init__ (line 25) | def __init__( method forward (line 68) | def forward(self, hidden_states, temb=None): class UpResnetBlock1D (line 86) | class UpResnetBlock1D(nn.Module): method __init__ (line 87) | def __init__( method forward (line 128) | def forward(self, hidden_states, res_hidden_states_tuple=None, temb=No... class ValueFunctionMidBlock1D (line 146) | class ValueFunctionMidBlock1D(nn.Module): method __init__ (line 147) | def __init__(self, in_channels, out_channels, embed_dim): method forward (line 158) | def forward(self, x, temb=None): class MidResTemporalBlock1D (line 166) | class MidResTemporalBlock1D(nn.Module): method __init__ (line 167) | def __init__( method forward (line 206) | def forward(self, hidden_states, temb): class OutConv1DBlock (line 219) | class OutConv1DBlock(nn.Module): method __init__ (line 220) | def __init__(self, num_groups_out, out_channels, embed_dim, act_fn): method forward (line 227) | def forward(self, hidden_states, temb=None): class OutValueFunctionBlock (line 237) | class OutValueFunctionBlock(nn.Module): method __init__ (line 238) | def __init__(self, fc_dim, embed_dim): method forward (line 248) | def forward(self, hidden_states, temb): class Downsample1d (line 277) | class Downsample1d(nn.Module): method __init__ (line 278) | def __init__(self, kernel="linear", pad_mode="reflect"): method forward (line 285) | def forward(self, hidden_states): class Upsample1d (line 294) | class Upsample1d(nn.Module): method __init__ (line 295) | def __init__(self, kernel="linear", pad_mode="reflect"): method forward (line 302) | def forward(self, hidden_states, temb=None): class SelfAttention1d (line 311) | class SelfAttention1d(nn.Module): method __init__ (line 312) | def __init__(self, in_channels, n_head=1, dropout_rate=0.0): method transpose_for_scores (line 326) | def transpose_for_scores(self, projection: torch.Tensor) -> torch.Tensor: method forward (line 332) | def forward(self, hidden_states): class ResConvBlock (line 369) | class ResConvBlock(nn.Module): method __init__ (line 370) | def __init__(self, in_channels, mid_channels, out_channels, is_last=Fa... method forward (line 387) | def forward(self, hidden_states): class UNetMidBlock1D (line 403) | class UNetMidBlock1D(nn.Module): method __init__ (line 404) | def __init__(self, mid_channels, in_channels, out_channels=None): method forward (line 432) | def forward(self, hidden_states, temb=None): class AttnDownBlock1D (line 443) | class AttnDownBlock1D(nn.Module): method __init__ (line 444) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 463) | def forward(self, hidden_states, temb=None): class DownBlock1D (line 473) | class DownBlock1D(nn.Module): method __init__ (line 474) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 487) | def forward(self, hidden_states, temb=None): class DownBlock1DNoSkip (line 496) | class DownBlock1DNoSkip(nn.Module): method __init__ (line 497) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 509) | def forward(self, hidden_states, temb=None): class AttnUpBlock1D (line 517) | class AttnUpBlock1D(nn.Module): method __init__ (line 518) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 537) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): class UpBlock1D (line 550) | class UpBlock1D(nn.Module): method __init__ (line 551) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 564) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): class UpBlock1DNoSkip (line 576) | class UpBlock1DNoSkip(nn.Module): method __init__ (line 577) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 589) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): function get_down_block (line 599) | def get_down_block(down_block_type, num_layers, in_channels, out_channel... function get_up_block (line 617) | def get_up_block(up_block_type, num_layers, in_channels, out_channels, t... function get_mid_block (line 635) | def get_mid_block(mid_block_type, num_layers, in_channels, mid_channels,... function get_out_block (line 651) | def get_out_block(*, out_block_type, num_groups_out, embed_dim, out_chan... FILE: 6DoF/diffusers/models/unet_2d.py class UNet2DOutput (line 28) | class UNet2DOutput(BaseOutput): class UNet2DModel (line 40) | class UNet2DModel(ModelMixin, ConfigMixin): method __init__ (line 88) | def __init__( method forward (line 228) | def forward( FILE: 6DoF/diffusers/models/unet_2d_blocks.py function get_down_block (line 32) | def get_down_block( function get_up_block (line 227) | def get_up_block( class UNetMidBlock2D (line 426) | class UNetMidBlock2D(nn.Module): method __init__ (line 427) | def __init__( method forward (line 507) | def forward(self, hidden_states, temb=None): class UNetMidBlock2DCrossAttn (line 517) | class UNetMidBlock2DCrossAttn(nn.Module): method __init__ (line 518) | def __init__( method forward (line 603) | def forward( class UNetMidBlock2DSimpleCrossAttn (line 629) | class UNetMidBlock2DSimpleCrossAttn(nn.Module): method __init__ (line 630) | def __init__( method forward (line 714) | def forward( class AttnDownBlock2D (line 752) | class AttnDownBlock2D(nn.Module): method __init__ (line 753) | def __init__( method forward (line 844) | def forward(self, hidden_states, temb=None, upsample_size=None): class CrossAttnDownBlock2D (line 864) | class CrossAttnDownBlock2D(nn.Module): method __init__ (line 865) | def __init__( method forward (line 952) | def forward( class DownBlock2D (line 1018) | class DownBlock2D(nn.Module): method __init__ (line 1019) | def __init__( method forward (line 1070) | def forward(self, hidden_states, temb=None): class DownEncoderBlock2D (line 1104) | class DownEncoderBlock2D(nn.Module): method __init__ (line 1105) | def __init__( method forward (line 1153) | def forward(self, hidden_states): class AttnDownEncoderBlock2D (line 1164) | class AttnDownEncoderBlock2D(nn.Module): method __init__ (line 1165) | def __init__( method forward (line 1236) | def forward(self, hidden_states): class AttnSkipDownBlock2D (line 1248) | class AttnSkipDownBlock2D(nn.Module): method __init__ (line 1249) | def __init__( method forward (line 1329) | def forward(self, hidden_states, temb=None, skip_sample=None): class SkipDownBlock2D (line 1349) | class SkipDownBlock2D(nn.Module): method __init__ (line 1350) | def __init__( method forward (line 1409) | def forward(self, hidden_states, temb=None, skip_sample=None): class ResnetDownsampleBlock2D (line 1428) | class ResnetDownsampleBlock2D(nn.Module): method __init__ (line 1429) | def __init__( method forward (line 1492) | def forward(self, hidden_states, temb=None): class SimpleCrossAttnDownBlock2D (line 1526) | class SimpleCrossAttnDownBlock2D(nn.Module): method __init__ (line 1527) | def __init__( method forward (line 1621) | def forward( class KDownBlock2D (line 1685) | class KDownBlock2D(nn.Module): method __init__ (line 1686) | def __init__( method forward (line 1731) | def forward(self, hidden_states, temb=None): class KCrossAttnDownBlock2D (line 1763) | class KCrossAttnDownBlock2D(nn.Module): method __init__ (line 1764) | def __init__( method forward (line 1828) | def forward( class AttnUpBlock2D (line 1891) | class AttnUpBlock2D(nn.Module): method __init__ (line 1892) | def __init__( method forward (line 1980) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class CrossAttnUpBlock2D (line 2000) | class CrossAttnUpBlock2D(nn.Module): method __init__ (line 2001) | def __init__( method forward (line 2084) | def forward( class UpBlock2D (line 2151) | class UpBlock2D(nn.Module): method __init__ (line 2152) | def __init__( method forward (line 2199) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class UpDecoderBlock2D (line 2232) | class UpDecoderBlock2D(nn.Module): method __init__ (line 2233) | def __init__( method forward (line 2276) | def forward(self, hidden_states, temb=None): class AttnUpDecoderBlock2D (line 2287) | class AttnUpDecoderBlock2D(nn.Module): method __init__ (line 2288) | def __init__( method forward (line 2355) | def forward(self, hidden_states, temb=None): class AttnSkipUpBlock2D (line 2367) | class AttnSkipUpBlock2D(nn.Module): method __init__ (line 2368) | def __init__( method forward (line 2458) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, s... class SkipUpBlock2D (line 2486) | class SkipUpBlock2D(nn.Module): method __init__ (line 2487) | def __init__( method forward (line 2555) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, s... class ResnetUpsampleBlock2D (line 2581) | class ResnetUpsampleBlock2D(nn.Module): method __init__ (line 2582) | def __init__( method forward (line 2648) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class SimpleCrossAttnUpBlock2D (line 2681) | class SimpleCrossAttnUpBlock2D(nn.Module): method __init__ (line 2682) | def __init__( method forward (line 2778) | def forward( class KUpBlock2D (line 2845) | class KUpBlock2D(nn.Module): method __init__ (line 2846) | def __init__( method forward (line 2893) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class KCrossAttnUpBlock2D (line 2925) | class KCrossAttnUpBlock2D(nn.Module): method __init__ (line 2926) | def __init__( method forward (line 3009) | def forward( class KAttentionBlock (line 3072) | class KAttentionBlock(nn.Module): method __init__ (line 3089) | def __init__( method _to_3d (line 3132) | def _to_3d(self, hidden_states, height, weight): method _to_4d (line 3135) | def _to_4d(self, hidden_states, height, weight): method forward (line 3138) | def forward( FILE: 6DoF/diffusers/models/unet_2d_blocks_flax.py class FlaxCrossAttnDownBlock2D (line 22) | class FlaxCrossAttnDownBlock2D(nn.Module): method setup (line 56) | def setup(self): method __call__ (line 89) | def __call__(self, hidden_states, temb, encoder_hidden_states, determi... class FlaxDownBlock2D (line 104) | class FlaxDownBlock2D(nn.Module): method setup (line 129) | def setup(self): method __call__ (line 147) | def __call__(self, hidden_states, temb, deterministic=True): class FlaxCrossAttnUpBlock2D (line 161) | class FlaxCrossAttnUpBlock2D(nn.Module): method setup (line 196) | def setup(self): method __call__ (line 230) | def __call__(self, hidden_states, res_hidden_states_tuple, temb, encod... class FlaxUpBlock2D (line 246) | class FlaxUpBlock2D(nn.Module): method setup (line 274) | def setup(self): method __call__ (line 294) | def __call__(self, hidden_states, res_hidden_states_tuple, temb, deter... class FlaxUNetMidBlock2DCrossAttn (line 309) | class FlaxUNetMidBlock2DCrossAttn(nn.Module): method setup (line 335) | def setup(self): method __call__ (line 371) | def __call__(self, hidden_states, temb, encoder_hidden_states, determi... FILE: 6DoF/diffusers/models/unet_2d_condition.py class UNet2DConditionOutput (line 54) | class UNet2DConditionOutput(BaseOutput): class UNet2DConditionModel (line 66) | class UNet2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoade... method __init__ (line 157) | def __init__( method attn_processors (line 568) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 591) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 625) | def set_default_attn_processor(self): method set_attention_slice (line 631) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 696) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 700) | def forward( FILE: 6DoF/diffusers/models/unet_2d_condition_flax.py class FlaxUNet2DConditionOutput (line 36) | class FlaxUNet2DConditionOutput(BaseOutput): class FlaxUNet2DConditionModel (line 49) | class FlaxUNet2DConditionModel(nn.Module, FlaxModelMixin, ConfigMixin): method init_weights (line 120) | def init_weights(self, rng: jax.random.KeyArray) -> FrozenDict: method setup (line 132) | def setup(self): method __call__ (line 267) | def __call__( FILE: 6DoF/diffusers/models/unet_3d_blocks.py function get_down_block (line 23) | def get_down_block( function get_up_block (line 79) | def get_up_block( class UNetMidBlock3DCrossAttn (line 135) | class UNetMidBlock3DCrossAttn(nn.Module): method __init__ (line 136) | def __init__( method forward (line 235) | def forward( class CrossAttnDownBlock3D (line 264) | class CrossAttnDownBlock3D(nn.Module): method __init__ (line 265) | def __init__( method forward (line 360) | def forward( class DownBlock3D (line 398) | class DownBlock3D(nn.Module): method __init__ (line 399) | def __init__( method forward (line 459) | def forward(self, hidden_states, temb=None, num_frames=1): class CrossAttnUpBlock3D (line 477) | class CrossAttnUpBlock3D(nn.Module): method __init__ (line 478) | def __init__( method forward (line 569) | def forward( class UpBlock3D (line 608) | class UpBlock3D(nn.Module): method __init__ (line 609) | def __init__( method forward (line 665) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... FILE: 6DoF/diffusers/models/unet_3d_condition.py class UNet3DConditionOutput (line 44) | class UNet3DConditionOutput(BaseOutput): class UNet3DConditionModel (line 56) | class UNet3DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoade... method __init__ (line 90) | def __init__( method attn_processors (line 273) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attention_slice (line 297) | def set_attention_slice(self, slice_size): method set_attn_processor (line 363) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method enable_forward_chunking (line 397) | def enable_forward_chunking(self, chunk_size=None, dim=0): method disable_forward_chunking (line 426) | def disable_forward_chunking(self): method set_default_attn_processor (line 438) | def set_default_attn_processor(self): method _set_gradient_checkpointing (line 444) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 448) | def forward( FILE: 6DoF/diffusers/models/vae.py class DecoderOutput (line 27) | class DecoderOutput(BaseOutput): class Encoder (line 39) | class Encoder(nn.Module): method __init__ (line 40) | def __init__( method forward (line 108) | def forward(self, x): class Decoder (line 152) | class Decoder(nn.Module): method __init__ (line 153) | def __init__( method forward (line 228) | def forward(self, z, latent_embeds=None): class VectorQuantizer (line 283) | class VectorQuantizer(nn.Module): method __init__ (line 292) | def __init__( method remap_to_used (line 321) | def remap_to_used(self, inds): method unmap_to_all (line 335) | def unmap_to_all(self, inds): method forward (line 345) | def forward(self, z): method get_codebook_entry (line 379) | def get_codebook_entry(self, indices, shape): class DiagonalGaussianDistribution (line 397) | class DiagonalGaussianDistribution(object): method __init__ (line 398) | def __init__(self, parameters, deterministic=False): method sample (line 410) | def sample(self, generator: Optional[torch.Generator] = None) -> torch... method kl (line 418) | def kl(self, other=None): method nll (line 434) | def nll(self, sample, dims=[1, 2, 3]): method mode (line 440) | def mode(self): FILE: 6DoF/diffusers/models/vae_flax.py class FlaxDecoderOutput (line 33) | class FlaxDecoderOutput(BaseOutput): class FlaxAutoencoderKLOutput (line 48) | class FlaxAutoencoderKLOutput(BaseOutput): class FlaxUpsample2D (line 61) | class FlaxUpsample2D(nn.Module): method setup (line 75) | def setup(self): method __call__ (line 84) | def __call__(self, hidden_states): class FlaxDownsample2D (line 95) | class FlaxDownsample2D(nn.Module): method setup (line 109) | def setup(self): method __call__ (line 118) | def __call__(self, hidden_states): class FlaxResnetBlock2D (line 125) | class FlaxResnetBlock2D(nn.Module): method setup (line 151) | def setup(self): method __call__ (line 185) | def __call__(self, hidden_states, deterministic=True): class FlaxAttentionBlock (line 202) | class FlaxAttentionBlock(nn.Module): method setup (line 222) | def setup(self): method transpose_for_scores (line 231) | def transpose_for_scores(self, projection): method __call__ (line 239) | def __call__(self, hidden_states): class FlaxDownEncoderBlock2D (line 274) | class FlaxDownEncoderBlock2D(nn.Module): method setup (line 302) | def setup(self): method __call__ (line 320) | def __call__(self, hidden_states, deterministic=True): class FlaxUpDecoderBlock2D (line 330) | class FlaxUpDecoderBlock2D(nn.Module): method setup (line 358) | def setup(self): method __call__ (line 376) | def __call__(self, hidden_states, deterministic=True): class FlaxUNetMidBlock2D (line 386) | class FlaxUNetMidBlock2D(nn.Module): method setup (line 411) | def setup(self): method __call__ (line 448) | def __call__(self, hidden_states, deterministic=True): class FlaxEncoder (line 457) | class FlaxEncoder(nn.Module): method setup (line 501) | def setup(self): method __call__ (line 550) | def __call__(self, sample, deterministic: bool = True): class FlaxDecoder (line 569) | class FlaxDecoder(nn.Module): method setup (line 612) | def setup(self): method __call__ (line 665) | def __call__(self, sample, deterministic: bool = True): class FlaxDiagonalGaussianDistribution (line 683) | class FlaxDiagonalGaussianDistribution(object): method __init__ (line 684) | def __init__(self, parameters, deterministic=False): method sample (line 694) | def sample(self, key): method kl (line 697) | def kl(self, other=None): method nll (line 709) | def nll(self, sample, axis=[1, 2, 3]): method mode (line 716) | def mode(self): class FlaxAutoencoderKL (line 721) | class FlaxAutoencoderKL(nn.Module, FlaxModelMixin, ConfigMixin): method setup (line 783) | def setup(self): method init_weights (line 820) | def init_weights(self, rng: jax.random.KeyArray) -> FrozenDict: method encode (line 830) | def encode(self, sample, deterministic: bool = True, return_dict: bool... method decode (line 842) | def decode(self, latents, deterministic: bool = True, return_dict: boo... method __call__ (line 856) | def __call__(self, sample, sample_posterior=False, deterministic: bool... FILE: 6DoF/diffusers/models/vq_model.py class VQEncoderOutput (line 27) | class VQEncoderOutput(BaseOutput): class VQModel (line 39) | class VQModel(ModelMixin, ConfigMixin): method __init__ (line 70) | def __init__( method encode (line 120) | def encode(self, x: torch.FloatTensor, return_dict: bool = True) -> VQ... method decode (line 130) | def decode( method forward (line 146) | def forward(self, sample: torch.FloatTensor, return_dict: bool = True)... FILE: 6DoF/diffusers/optimization.py class SchedulerType (line 30) | class SchedulerType(Enum): function get_constant_schedule (line 40) | def get_constant_schedule(optimizer: Optimizer, last_epoch: int = -1): function get_constant_schedule_with_warmup (line 56) | def get_constant_schedule_with_warmup(optimizer: Optimizer, num_warmup_s... function get_piecewise_constant_schedule (line 81) | def get_piecewise_constant_schedule(optimizer: Optimizer, step_rules: st... function get_linear_schedule_with_warmup (line 123) | def get_linear_schedule_with_warmup(optimizer, num_warmup_steps, num_tra... function get_cosine_schedule_with_warmup (line 152) | def get_cosine_schedule_with_warmup( function get_cosine_with_hard_restarts_schedule_with_warmup (line 186) | def get_cosine_with_hard_restarts_schedule_with_warmup( function get_polynomial_decay_schedule_with_warmup (line 221) | def get_polynomial_decay_schedule_with_warmup( function get_scheduler (line 282) | def get_scheduler( FILE: 6DoF/diffusers/pipelines/alt_diffusion/__init__.py class AltDiffusionPipelineOutput (line 13) | class AltDiffusionPipelineOutput(BaseOutput): FILE: 6DoF/diffusers/pipelines/alt_diffusion/modeling_roberta_series.py class TransformationModelOutput (line 11) | class TransformationModelOutput(ModelOutput): class RobertaSeriesConfig (line 39) | class RobertaSeriesConfig(XLMRobertaConfig): method __init__ (line 40) | def __init__( class RobertaSeriesModelWithTransformation (line 58) | class RobertaSeriesModelWithTransformation(RobertaPreTrainedModel): method __init__ (line 64) | def __init__(self, config): method forward (line 74) | def forward( FILE: 6DoF/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py function rescale_noise_cfg (line 55) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0): class AltDiffusionPipeline (line 70) | class AltDiffusionPipeline(DiffusionPipeline, TextualInversionLoaderMixi... method __init__ (line 107) | def __init__( method enable_vae_slicing (line 197) | def enable_vae_slicing(self): method disable_vae_slicing (line 206) | def disable_vae_slicing(self): method enable_vae_tiling (line 213) | def enable_vae_tiling(self): method disable_vae_tiling (line 222) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 229) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 254) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 283) | def _execution_device(self): method _encode_prompt (line 300) | def _encode_prompt( method run_safety_checker (line 454) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 468) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 483) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 500) | def check_inputs( method prepare_latents (line 547) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 566) | def __call__( FILE: 6DoF/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py function preprocess (line 71) | def preprocess(image): class AltDiffusionImg2ImgPipeline (line 98) | class AltDiffusionImg2ImgPipeline( method __init__ (line 137) | def __init__( method enable_sequential_cpu_offload (line 227) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 252) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 281) | def _execution_device(self): method _encode_prompt (line 298) | def _encode_prompt( method run_safety_checker (line 452) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 466) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 481) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 498) | def check_inputs( method get_timesteps (line 538) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 547) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 606) | def __call__( FILE: 6DoF/diffusers/pipelines/audio_diffusion/mel.py class Mel (line 37) | class Mel(ConfigMixin, SchedulerMixin): method __init__ (line 52) | def __init__( method set_resolution (line 73) | def set_resolution(self, x_res: int, y_res: int): method load_audio (line 85) | def load_audio(self, audio_file: str = None, raw_audio: np.ndarray = N... method get_number_of_slices (line 101) | def get_number_of_slices(self) -> int: method get_audio_slice (line 109) | def get_audio_slice(self, slice: int = 0) -> np.ndarray: method get_sample_rate (line 120) | def get_sample_rate(self) -> int: method audio_slice_to_image (line 128) | def audio_slice_to_image(self, slice: int) -> Image.Image: method image_to_audio (line 145) | def image_to_audio(self, image: Image.Image) -> np.ndarray: FILE: 6DoF/diffusers/pipelines/audio_diffusion/pipeline_audio_diffusion.py class AudioDiffusionPipeline (line 30) | class AudioDiffusionPipeline(DiffusionPipeline): method __init__ (line 44) | def __init__( method get_default_steps (line 54) | def get_default_steps(self) -> int: method __call__ (line 63) | def __call__( method encode (line 199) | def encode(self, images: List[Image.Image], steps: int = 50) -> np.nda... method slerp (line 236) | def slerp(x0: torch.Tensor, x1: torch.Tensor, alpha: float) -> torch.T... FILE: 6DoF/diffusers/pipelines/audioldm/pipeline_audioldm.py class AudioLDMPipeline (line 46) | class AudioLDMPipeline(DiffusionPipeline): method __init__ (line 72) | def __init__( method enable_vae_slicing (line 94) | def enable_vae_slicing(self): method disable_vae_slicing (line 104) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 111) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 129) | def _execution_device(self): method _encode_prompt (line 146) | def _encode_prompt( method decode_latents (line 285) | def decode_latents(self, latents): method mel_spectrogram_to_waveform (line 290) | def mel_spectrogram_to_waveform(self, mel_spectrogram): method prepare_extra_step_kwargs (line 300) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 317) | def check_inputs( method prepare_latents (line 376) | def prepare_latents(self, batch_size, num_channels_latents, height, dt... method __call__ (line 400) | def __call__( FILE: 6DoF/diffusers/pipelines/consistency_models/pipeline_consistency_models.py class ConsistencyModelPipeline (line 51) | class ConsistencyModelPipeline(DiffusionPipeline): method __init__ (line 69) | def __init__(self, unet: UNet2DModel, scheduler: CMStochasticIterative... method enable_sequential_cpu_offload (line 79) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 104) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 134) | def _execution_device(self): method prepare_latents (line 151) | def prepare_latents(self, batch_size, num_channels, height, width, dty... method postprocess_image (line 169) | def postprocess_image(self, sample: torch.FloatTensor, output_type: st... method prepare_class_labels (line 189) | def prepare_class_labels(self, batch_size, device, class_labels=None): method check_inputs (line 205) | def check_inputs(self, num_inference_steps, timesteps, latents, batch_... method __call__ (line 230) | def __call__( FILE: 6DoF/diffusers/pipelines/controlnet/multicontrolnet.py class MultiControlNetModel (line 15) | class MultiControlNetModel(ModelMixin): method __init__ (line 28) | def __init__(self, controlnets: Union[List[ControlNetModel], Tuple[Con... method forward (line 32) | def forward( method save_pretrained (line 73) | def save_pretrained( method from_pretrained (line 116) | def from_pretrained(cls, pretrained_model_path: Optional[Union[str, os... FILE: 6DoF/diffusers/pipelines/controlnet/pipeline_controlnet.py class StableDiffusionControlNetPipeline (line 93) | class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInvers... method __init__ (line 129) | def __init__( method enable_vae_slicing (line 180) | def enable_vae_slicing(self): method disable_vae_slicing (line 190) | def disable_vae_slicing(self): method enable_vae_tiling (line 198) | def enable_vae_tiling(self): method disable_vae_tiling (line 208) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 215) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 236) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 266) | def _execution_device(self): method _encode_prompt (line 284) | def _encode_prompt( method run_safety_checker (line 439) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 454) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 468) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 485) | def check_inputs( method check_image (line 619) | def check_image(self, image, prompt, prompt_embeds): method prepare_image (line 656) | def prepare_image( method prepare_latents (line 687) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 706) | def __call__( FILE: 6DoF/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py function prepare_image (line 95) | def prepare_image(image): class StableDiffusionControlNetImg2ImgPipeline (line 119) | class StableDiffusionControlNetImg2ImgPipeline(DiffusionPipeline, Textua... method __init__ (line 155) | def __init__( method enable_vae_slicing (line 206) | def enable_vae_slicing(self): method disable_vae_slicing (line 216) | def disable_vae_slicing(self): method enable_vae_tiling (line 224) | def enable_vae_tiling(self): method disable_vae_tiling (line 234) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 241) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 262) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 292) | def _execution_device(self): method _encode_prompt (line 310) | def _encode_prompt( method run_safety_checker (line 465) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 480) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 494) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 511) | def check_inputs( method check_image (line 646) | def check_image(self, image, prompt, prompt_embeds): method prepare_control_image (line 684) | def prepare_control_image( method get_timesteps (line 715) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 725) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 784) | def __call__( FILE: 6DoF/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py function prepare_mask_and_masked_image (line 108) | def prepare_mask_and_masked_image(image, mask, height, width, return_ima... class StableDiffusionControlNetInpaintPipeline (line 225) | class StableDiffusionControlNetInpaintPipeline(DiffusionPipeline, Textua... method __init__ (line 272) | def __init__( method enable_vae_slicing (line 323) | def enable_vae_slicing(self): method disable_vae_slicing (line 333) | def disable_vae_slicing(self): method enable_vae_tiling (line 341) | def enable_vae_tiling(self): method disable_vae_tiling (line 351) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 358) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 379) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 409) | def _execution_device(self): method _encode_prompt (line 427) | def _encode_prompt( method run_safety_checker (line 582) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 597) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 611) | def prepare_extra_step_kwargs(self, generator, eta): method get_timesteps (line 629) | def get_timesteps(self, num_inference_steps, strength, device): method check_inputs (line 638) | def check_inputs( method check_image (line 778) | def check_image(self, image, prompt, prompt_embeds): method prepare_control_image (line 816) | def prepare_control_image( method prepare_latents (line 847) | def prepare_latents( method _default_height_width (line 900) | def _default_height_width(self, height, width, image): method prepare_mask_latents (line 926) | def prepare_mask_latents( method _encode_vae_image (line 968) | def _encode_vae_image(self, image: torch.Tensor, generator: torch.Gene... method __call__ (line 984) | def __call__( FILE: 6DoF/diffusers/pipelines/controlnet/pipeline_flax_controlnet.py class FlaxStableDiffusionControlNetPipeline (line 119) | class FlaxStableDiffusionControlNetPipeline(FlaxDiffusionPipeline): method __init__ (line 150) | def __init__( method prepare_text_inputs (line 189) | def prepare_text_inputs(self, prompt: Union[str, List[str]]): method prepare_image_inputs (line 203) | def prepare_image_inputs(self, image: Union[Image.Image, List[Image.Im... method _get_has_nsfw_concepts (line 214) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 218) | def _run_safety_checker(self, images, safety_model_params, jit=False): method _generate (line 248) | def _generate( method __call__ (line 358) | def __call__( function _p_generate (line 493) | def _p_generate( function _p_get_has_nsfw_concepts (line 519) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 523) | def unshard(x: jnp.ndarray): function preprocess (line 530) | def preprocess(image, dtype): FILE: 6DoF/diffusers/pipelines/dance_diffusion/pipeline_dance_diffusion.py class DanceDiffusionPipeline (line 27) | class DanceDiffusionPipeline(DiffusionPipeline): method __init__ (line 39) | def __init__(self, unet, scheduler): method __call__ (line 44) | def __call__( FILE: 6DoF/diffusers/pipelines/ddim/pipeline_ddim.py class DDIMPipeline (line 24) | class DDIMPipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet, scheduler): method __call__ (line 45) | def __call__( FILE: 6DoF/diffusers/pipelines/ddpm/pipeline_ddpm.py class DDPMPipeline (line 24) | class DDPMPipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet, scheduler): method __call__ (line 41) | def __call__( FILE: 6DoF/diffusers/pipelines/deepfloyd_if/__init__.py class IFPipelineOutput (line 21) | class IFPipelineOutput(BaseOutput): FILE: 6DoF/diffusers/pipelines/deepfloyd_if/pipeline_if.py class IFPipeline (line 89) | class IFPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 107) | def __init__( method enable_sequential_cpu_offload (line 147) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 171) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 212) | def remove_all_hooks(self): method _execution_device (line 228) | def _execution_device(self): method encode_prompt (line 246) | def encode_prompt( method run_safety_checker (line 397) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 414) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 431) | def check_inputs( method prepare_intermediate_images (line 473) | def prepare_intermediate_images(self, batch_size, num_channels, height... method _text_preprocessing (line 487) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 511) | def _clean_caption(self, caption): method __call__ (line 627) | def __call__( FILE: 6DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py function resize (line 41) | def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image: class IFImg2ImgPipeline (line 113) | class IFImg2ImgPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 131) | def __init__( method enable_sequential_cpu_offload (line 172) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 197) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 239) | def remove_all_hooks(self): method _execution_device (line 255) | def _execution_device(self): method encode_prompt (line 274) | def encode_prompt( method run_safety_checker (line 426) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 443) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 460) | def check_inputs( method _text_preprocessing (line 534) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 559) | def _clean_caption(self, caption): method preprocess_image (line 673) | def preprocess_image(self, image: PIL.Image.Image) -> torch.Tensor: method get_timesteps (line 709) | def get_timesteps(self, num_inference_steps, strength): method prepare_intermediate_images (line 718) | def prepare_intermediate_images( method __call__ (line 742) | def __call__( FILE: 6DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py function resize (line 44) | def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image: class IFImg2ImgSuperResolutionPipeline (line 116) | class IFImg2ImgSuperResolutionPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 135) | def __init__( method enable_sequential_cpu_offload (line 183) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 208) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 250) | def remove_all_hooks(self): method _text_preprocessing (line 265) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 290) | def _clean_caption(self, caption): method _execution_device (line 406) | def _execution_device(self): method encode_prompt (line 425) | def encode_prompt( method run_safety_checker (line 577) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 594) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 611) | def check_inputs( method preprocess_original_image (line 721) | def preprocess_original_image(self, image: PIL.Image.Image) -> torch.T... method preprocess_image (line 758) | def preprocess_image(self, image: PIL.Image.Image, num_images_per_prom... method get_timesteps (line 790) | def get_timesteps(self, num_inference_steps, strength): method prepare_intermediate_images (line 800) | def prepare_intermediate_images( method __call__ (line 824) | def __call__( FILE: 6DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py function resize (line 42) | def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image: class IFInpaintingPipeline (line 116) | class IFInpaintingPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 134) | def __init__( method enable_sequential_cpu_offload (line 175) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 200) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 242) | def remove_all_hooks(self): method _execution_device (line 258) | def _execution_device(self): method encode_prompt (line 277) | def encode_prompt( method run_safety_checker (line 429) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 446) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 463) | def check_inputs( method _text_preprocessing (line 573) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 598) | def _clean_caption(self, caption): method preprocess_image (line 713) | def preprocess_image(self, image: PIL.Image.Image) -> torch.Tensor: method preprocess_mask_image (line 749) | def preprocess_mask_image(self, mask_image) -> torch.Tensor: method get_timesteps (line 799) | def get_timesteps(self, num_inference_steps, strength): method prepare_intermediate_images (line 808) | def prepare_intermediate_images( method __call__ (line 834) | def __call__( FILE: 6DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py function resize (line 44) | def resize(images: PIL.Image.Image, img_size: int) -> PIL.Image.Image: class IFInpaintingSuperResolutionPipeline (line 118) | class IFInpaintingSuperResolutionPipeline(DiffusionPipeline, LoraLoaderM... method __init__ (line 137) | def __init__( method enable_sequential_cpu_offload (line 185) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 210) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 252) | def remove_all_hooks(self): method _text_preprocessing (line 267) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 292) | def _clean_caption(self, caption): method _execution_device (line 408) | def _execution_device(self): method encode_prompt (line 427) | def encode_prompt( method run_safety_checker (line 579) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 596) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 613) | def check_inputs( method preprocess_original_image (line 757) | def preprocess_original_image(self, image: PIL.Image.Image) -> torch.T... method preprocess_image (line 794) | def preprocess_image(self, image: PIL.Image.Image, num_images_per_prom... method preprocess_mask_image (line 826) | def preprocess_mask_image(self, mask_image) -> torch.Tensor: method get_timesteps (line 876) | def get_timesteps(self, num_inference_steps, strength): method prepare_intermediate_images (line 886) | def prepare_intermediate_images( method __call__ (line 912) | def __call__( FILE: 6DoF/diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py class IFSuperResolutionPipeline (line 74) | class IFSuperResolutionPipeline(DiffusionPipeline, LoraLoaderMixin): method __init__ (line 93) | def __init__( method enable_sequential_cpu_offload (line 141) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 166) | def enable_model_cpu_offload(self, gpu_id=0): method remove_all_hooks (line 208) | def remove_all_hooks(self): method _text_preprocessing (line 223) | def _text_preprocessing(self, text, clean_caption=False): method _clean_caption (line 248) | def _clean_caption(self, caption): method _execution_device (line 364) | def _execution_device(self): method encode_prompt (line 383) | def encode_prompt( method run_safety_checker (line 535) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 552) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 569) | def check_inputs( method prepare_intermediate_images (line 649) | def prepare_intermediate_images(self, batch_size, num_channels, height... method preprocess_image (line 663) | def preprocess_image(self, image, num_images_per_prompt, device): method __call__ (line 696) | def __call__( FILE: 6DoF/diffusers/pipelines/deepfloyd_if/safety_checker.py class IFSafetyChecker (line 12) | class IFSafetyChecker(PreTrainedModel): method __init__ (line 17) | def __init__(self, config: CLIPConfig): method forward (line 26) | def forward(self, clip_input, images, p_threshold=0.5, w_threshold=0.5): FILE: 6DoF/diffusers/pipelines/deepfloyd_if/watermark.py class IFWatermarker (line 12) | class IFWatermarker(ModelMixin, ConfigMixin): method __init__ (line 13) | def __init__(self): method apply_watermark (line 19) | def apply_watermark(self, images: List[PIL.Image.Image], sample_size=N... FILE: 6DoF/diffusers/pipelines/dit/pipeline_dit.py class DiTPipeline (line 31) | class DiTPipeline(DiffusionPipeline): method __init__ (line 45) | def __init__( method get_label_ids (line 63) | def get_label_ids(self, label: Union[str, List[str]]) -> List[int]: method __call__ (line 87) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky/pipeline_kandinsky.py function get_new_h_w (line 69) | def get_new_h_w(h, w, scale_factor=8): class KandinskyPipeline (line 79) | class KandinskyPipeline(DiffusionPipeline): method __init__ (line 99) | def __init__( method prepare_latents (line 119) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 130) | def _encode_prompt( method enable_sequential_cpu_offload (line 232) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 254) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 284) | def _execution_device(self): method __call__ (line 303) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky/pipeline_kandinsky_img2img.py function get_new_h_w (line 82) | def get_new_h_w(h, w, scale_factor=8): function prepare_image (line 92) | def prepare_image(pil_image, w=512, h=512): class KandinskyImg2ImgPipeline (line 101) | class KandinskyImg2ImgPipeline(DiffusionPipeline): method __init__ (line 121) | def __init__( method get_timesteps (line 140) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 149) | def prepare_latents(self, latents, latent_timestep, shape, dtype, devi... method _encode_prompt (line 165) | def _encode_prompt( method enable_sequential_cpu_offload (line 267) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 289) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 319) | def _execution_device(self): method add_noise (line 337) | def add_noise( method __call__ (line 365) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky/pipeline_kandinsky_inpaint.py function get_new_h_w (line 89) | def get_new_h_w(h, w, scale_factor=8): function prepare_mask (line 99) | def prepare_mask(masks): function prepare_mask_and_masked_image (line 123) | def prepare_mask_and_masked_image(image, mask, height, width): class KandinskyInpaintPipeline (line 239) | class KandinskyInpaintPipeline(DiffusionPipeline): method __init__ (line 259) | def __init__( method prepare_latents (line 279) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 290) | def _encode_prompt( method enable_sequential_cpu_offload (line 392) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 414) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 444) | def _execution_device(self): method __call__ (line 463) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky/pipeline_kandinsky_prior.py class KandinskyPriorPipelineOutput (line 114) | class KandinskyPriorPipelineOutput(BaseOutput): class KandinskyPriorPipeline (line 129) | class KandinskyPriorPipeline(DiffusionPipeline): method __init__ (line 150) | def __init__( method interpolate (line 172) | def interpolate( method prepare_latents (line 278) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method get_zero_embed (line 289) | def get_zero_embed(self, batch_size=1, device=None): method enable_sequential_cpu_offload (line 298) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 320) | def _execution_device(self): method _encode_prompt (line 337) | def _encode_prompt( method __call__ (line 436) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky/text_encoder.py class MCLIPConfig (line 5) | class MCLIPConfig(XLMRobertaConfig): method __init__ (line 8) | def __init__(self, transformerDimSize=1024, imageDimSize=768, **kwargs): class MultilingualCLIP (line 14) | class MultilingualCLIP(PreTrainedModel): method __init__ (line 17) | def __init__(self, config, *args, **kwargs): method forward (line 24) | def forward(self, input_ids, attention_mask): FILE: 6DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2.py function downscale_height_and_width (line 60) | def downscale_height_and_width(height, width, scale_factor=8): class KandinskyV22Pipeline (line 70) | class KandinskyV22Pipeline(DiffusionPipeline): method __init__ (line 86) | def __init__( method prepare_latents (line 102) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 113) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 134) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 161) | def _execution_device(self): method __call__ (line 180) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet.py function downscale_height_and_width (line 100) | def downscale_height_and_width(height, width, scale_factor=8): class KandinskyV22ControlnetPipeline (line 110) | class KandinskyV22ControlnetPipeline(DiffusionPipeline): method __init__ (line 126) | def __init__( method prepare_latents (line 142) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 154) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 176) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 203) | def _execution_device(self): method __call__ (line 222) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_controlnet_img2img.py function downscale_height_and_width (line 104) | def downscale_height_and_width(height, width, scale_factor=8): function prepare_image (line 115) | def prepare_image(pil_image, w=512, h=512): class KandinskyV22ControlnetImg2ImgPipeline (line 124) | class KandinskyV22ControlnetImg2ImgPipeline(DiffusionPipeline): method __init__ (line 140) | def __init__( method get_timesteps (line 156) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 166) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method enable_sequential_cpu_offload (line 209) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 231) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 258) | def _execution_device(self): method __call__ (line 277) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_img2img.py function downscale_height_and_width (line 78) | def downscale_height_and_width(height, width, scale_factor=8): function prepare_image (line 89) | def prepare_image(pil_image, w=512, h=512): class KandinskyV22Img2ImgPipeline (line 98) | class KandinskyV22Img2ImgPipeline(DiffusionPipeline): method __init__ (line 114) | def __init__( method get_timesteps (line 130) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 139) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method enable_sequential_cpu_offload (line 182) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 204) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 231) | def _execution_device(self): method __call__ (line 250) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_inpainting.py function downscale_height_and_width (line 85) | def downscale_height_and_width(height, width, scale_factor=8): function prepare_mask (line 96) | def prepare_mask(masks): function prepare_mask_and_masked_image (line 121) | def prepare_mask_and_masked_image(image, mask, height, width): class KandinskyV22InpaintPipeline (line 237) | class KandinskyV22InpaintPipeline(DiffusionPipeline): method __init__ (line 253) | def __init__( method prepare_latents (line 269) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 281) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 303) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 330) | def _execution_device(self): method __call__ (line 349) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior.py class KandinskyV22PriorPipeline (line 85) | class KandinskyV22PriorPipeline(DiffusionPipeline): method __init__ (line 108) | def __init__( method interpolate (line 130) | def interpolate( method prepare_latents (line 236) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method get_zero_embed (line 248) | def get_zero_embed(self, batch_size=1, device=None): method enable_sequential_cpu_offload (line 258) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 281) | def _execution_device(self): method _encode_prompt (line 299) | def _encode_prompt( method __call__ (line 398) | def __call__( FILE: 6DoF/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_prior_emb2emb.py class KandinskyV22PriorEmb2EmbPipeline (line 103) | class KandinskyV22PriorEmb2EmbPipeline(DiffusionPipeline): method __init__ (line 124) | def __init__( method get_timesteps (line 144) | def get_timesteps(self, num_inference_steps, strength, device): method interpolate (line 155) | def interpolate( method _encode_image (line 243) | def _encode_image( method prepare_latents (line 260) | def prepare_latents(self, emb, timestep, batch_size, num_images_per_pr... method get_zero_embed (line 287) | def get_zero_embed(self, batch_size=1, device=None): method enable_sequential_cpu_offload (line 297) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 320) | def _execution_device(self): method _encode_prompt (line 338) | def _encode_prompt( method __call__ (line 437) | def __call__( FILE: 6DoF/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py class LDMTextToImagePipeline (line 32) | class LDMTextToImagePipeline(DiffusionPipeline): method __init__ (line 51) | def __init__( method __call__ (line 64) | def __call__( class LDMBertConfig (line 220) | class LDMBertConfig(PretrainedConfig): method __init__ (line 225) | def __init__( function _expand_mask (line 267) | def _expand_mask(mask: torch.Tensor, dtype: torch.dtype, tgt_len: Option... class LDMBertAttention (line 282) | class LDMBertAttention(nn.Module): method __init__ (line 285) | def __init__( method _shape (line 309) | def _shape(self, tensor: torch.Tensor, seq_len: int, bsz: int): method forward (line 312) | def forward( class LDMBertEncoderLayer (line 426) | class LDMBertEncoderLayer(nn.Module): method __init__ (line 427) | def __init__(self, config: LDMBertConfig): method forward (line 444) | def forward( class LDMBertPreTrainedModel (line 496) | class LDMBertPreTrainedModel(PreTrainedModel): method _init_weights (line 502) | def _init_weights(self, module): method _set_gradient_checkpointing (line 513) | def _set_gradient_checkpointing(self, module, value=False): method dummy_inputs (line 518) | def dummy_inputs(self): class LDMBertEncoder (line 528) | class LDMBertEncoder(LDMBertPreTrainedModel): method __init__ (line 538) | def __init__(self, config: LDMBertConfig): method get_input_embeddings (line 556) | def get_input_embeddings(self): method set_input_embeddings (line 559) | def set_input_embeddings(self, value): method forward (line 562) | def forward( class LDMBertModel (line 695) | class LDMBertModel(LDMBertPreTrainedModel): method __init__ (line 698) | def __init__(self, config: LDMBertConfig): method forward (line 703) | def forward( FILE: 6DoF/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py function preprocess (line 22) | def preprocess(image): class LDMSuperResolutionPipeline (line 32) | class LDMSuperResolutionPipeline(DiffusionPipeline): method __init__ (line 49) | def __init__( method __call__ (line 66) | def __call__( FILE: 6DoF/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py class LDMPipeline (line 26) | class LDMPipeline(DiffusionPipeline): method __init__ (line 39) | def __init__(self, vqvae: VQModel, unet: UNet2DModel, scheduler: DDIMS... method __call__ (line 44) | def __call__( FILE: 6DoF/diffusers/pipelines/onnx_utils.py class OnnxRuntimeModel (line 51) | class OnnxRuntimeModel: method __init__ (line 52) | def __init__(self, model=None, **kwargs): method __call__ (line 58) | def __call__(self, **kwargs): method load_model (line 63) | def load_model(path: Union[str, Path], provider=None, sess_options=None): method _save_pretrained (line 79) | def _save_pretrained(self, save_directory: Union[str, Path], file_name... method save_pretrained (line 110) | def save_pretrained( method _from_pretrained (line 133) | def _from_pretrained( method from_pretrained (line 193) | def from_pretrained( FILE: 6DoF/diffusers/pipelines/paint_by_example/image_encoder.py class PaintByExampleImageEncoder (line 25) | class PaintByExampleImageEncoder(CLIPPreTrainedModel): method __init__ (line 26) | def __init__(self, config, proj_size=768): method forward (line 38) | def forward(self, pixel_values, return_uncond_vector=False): class PaintByExampleMapper (line 50) | class PaintByExampleMapper(nn.Module): method __init__ (line 51) | def __init__(self, config): method forward (line 63) | def forward(self, hidden_states): FILE: 6DoF/diffusers/pipelines/paint_by_example/pipeline_paint_by_example.py function prepare_mask_and_masked_image (line 39) | def prepare_mask_and_masked_image(image, mask): class PaintByExamplePipeline (line 139) | class PaintByExamplePipeline(DiffusionPipeline): method __init__ (line 168) | def __init__( method enable_sequential_cpu_offload (line 192) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 213) | def _execution_device(self): method run_safety_checker (line 231) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 246) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 264) | def decode_latents(self, latents): method check_inputs (line 278) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 301) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_mask_latents (line 319) | def prepare_mask_latents( method _encode_vae_image (line 361) | def _encode_vae_image(self, image: torch.Tensor, generator: torch.Gene... method _encode_image (line 375) | def _encode_image(self, image, device, num_images_per_prompt, do_class... method __call__ (line 401) | def __call__( FILE: 6DoF/diffusers/pipelines/pipeline_flax_utils.py function import_flax_or_no_model (line 66) | def import_flax_or_no_model(module, class_name): class FlaxImagePipelineOutput (line 80) | class FlaxImagePipelineOutput(BaseOutput): class FlaxDiffusionPipeline (line 93) | class FlaxDiffusionPipeline(ConfigMixin): method register_modules (line 110) | def register_modules(self, **kwargs): method save_pretrained (line 143) | def save_pretrained(self, save_directory: Union[str, os.PathLike], par... method from_pretrained (line 194) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method _get_signature_keys (line 496) | def _get_signature_keys(obj): method components (line 504) | def components(self) -> Dict[str, Any]: method numpy_to_pil (line 541) | def numpy_to_pil(images): method progress_bar (line 557) | def progress_bar(self, iterable): method set_progress_bar_config (line 567) | def set_progress_bar_config(self, **kwargs): FILE: 6DoF/diffusers/pipelines/pipeline_utils.py class ImagePipelineOutput (line 112) | class ImagePipelineOutput(BaseOutput): class AudioPipelineOutput (line 126) | class AudioPipelineOutput(BaseOutput): function is_safetensors_compatible (line 138) | def is_safetensors_compatible(filenames, variant=None, passed_components... function variant_compatible_siblings (line 187) | def variant_compatible_siblings(filenames, variant=None) -> Union[List[o... function warn_deprecated_model_variant (line 254) | def warn_deprecated_model_variant(pretrained_model_name_or_path, use_aut... function maybe_raise_or_warn (line 276) | def maybe_raise_or_warn( function get_class_obj_and_candidates (line 308) | def get_class_obj_and_candidates(library_name, class_name, importable_cl... function _get_pipeline_class (line 325) | def _get_pipeline_class(class_obj, config, custom_pipeline=None, cache_d... function load_sub_model (line 346) | def load_sub_model( class DiffusionPipeline (line 458) | class DiffusionPipeline(ConfigMixin): method register_modules (line 478) | def register_modules(self, **kwargs): method __setattr__ (line 521) | def __setattr__(self, name: str, value: Any): method save_pretrained (line 536) | def save_pretrained( method to (line 623) | def to( method device (line 699) | def device(self) -> torch.device: method from_pretrained (line 714) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method download (line 1079) | def download(cls, pretrained_model_name, **kwargs) -> Union[str, os.Pa... method _get_signature_keys (line 1340) | def _get_signature_keys(obj): method components (line 1348) | def components(self) -> Dict[str, Any]: method numpy_to_pil (line 1384) | def numpy_to_pil(images): method progress_bar (line 1390) | def progress_bar(self, iterable=None, total=None): method set_progress_bar_config (line 1405) | def set_progress_bar_config(self, **kwargs): method enable_xformers_memory_efficient_attention (line 1408) | def enable_xformers_memory_efficient_attention(self, attention_op: Opt... method disable_xformers_memory_efficient_attention (line 1444) | def disable_xformers_memory_efficient_attention(self): method set_use_memory_efficient_attention_xformers (line 1450) | def set_use_memory_efficient_attention_xformers( method enable_attention_slicing (line 1470) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 1486) | def disable_attention_slicing(self): method set_attention_slice (line 1494) | def set_attention_slice(self, slice_size: Optional[int]): FILE: 6DoF/diffusers/pipelines/pndm/pipeline_pndm.py class PNDMPipeline (line 26) | class PNDMPipeline(DiffusionPipeline): method __init__ (line 40) | def __init__(self, unet: UNet2DModel, scheduler: PNDMScheduler): method __call__ (line 48) | def __call__( FILE: 6DoF/diffusers/pipelines/repaint/pipeline_repaint.py function _preprocess_image (line 33) | def _preprocess_image(image: Union[List, PIL.Image.Image, torch.Tensor]): function _preprocess_mask (line 59) | def _preprocess_mask(mask: Union[List, PIL.Image.Image, torch.Tensor]): class RePaintPipeline (line 79) | class RePaintPipeline(DiffusionPipeline): method __init__ (line 83) | def __init__(self, unet, scheduler): method __call__ (line 88) | def __call__( FILE: 6DoF/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py class ScoreSdeVePipeline (line 25) | class ScoreSdeVePipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet: UNet2DModel, scheduler: ScoreSdeVeScheduler): method __call__ (line 41) | def __call__( FILE: 6DoF/diffusers/pipelines/semantic_stable_diffusion/__init__.py class SemanticStableDiffusionPipelineOutput (line 13) | class SemanticStableDiffusionPipelineOutput(BaseOutput): FILE: 6DoF/diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py class SemanticStableDiffusionPipeline (line 63) | class SemanticStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 95) | def __init__( method run_safety_checker (line 138) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 153) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 167) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 185) | def check_inputs( method prepare_latents (line 233) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 251) | def __call__( FILE: 6DoF/diffusers/pipelines/shap_e/camera.py class DifferentiableProjectiveCamera (line 23) | class DifferentiableProjectiveCamera: method __post_init__ (line 38) | def __post_init__(self): method resolution (line 43) | def resolution(self): method fov (line 46) | def fov(self): method get_image_coords (line 49) | def get_image_coords(self) -> torch.Tensor: method camera_rays (line 64) | def camera_rays(self): method get_camera_rays (line 76) | def get_camera_rays(self, coords: torch.Tensor) -> torch.Tensor: method resize_image (line 105) | def resize_image(self, width: int, height: int) -> "DifferentiableProj... function create_pan_cameras (line 122) | def create_pan_cameras(size: int) -> DifferentiableProjectiveCamera: FILE: 6DoF/diffusers/pipelines/shap_e/pipeline_shap_e.py class ShapEPipelineOutput (line 69) | class ShapEPipelineOutput(BaseOutput): class ShapEPipeline (line 81) | class ShapEPipeline(DiffusionPipeline): method __init__ (line 103) | def __init__( method prepare_latents (line 122) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 133) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 151) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 180) | def _execution_device(self): method _encode_prompt (line 197) | def _encode_prompt( method __call__ (line 248) | def __call__( FILE: 6DoF/diffusers/pipelines/shap_e/pipeline_shap_e_img2img.py class ShapEPipelineOutput (line 69) | class ShapEPipelineOutput(BaseOutput): class ShapEImg2ImgPipeline (line 81) | class ShapEImg2ImgPipeline(DiffusionPipeline): method __init__ (line 103) | def __init__( method prepare_latents (line 122) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method enable_sequential_cpu_offload (line 133) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 152) | def _execution_device(self): method _encode_image (line 169) | def _encode_image( method __call__ (line 201) | def __call__( FILE: 6DoF/diffusers/pipelines/shap_e/renderer.py function sample_pmf (line 30) | def sample_pmf(pmf: torch.Tensor, n_samples: int) -> torch.Tensor: function posenc_nerf (line 53) | def posenc_nerf(x: torch.Tensor, min_deg: int = 0, max_deg: int = 15) ->... function encode_position (line 70) | def encode_position(position): function encode_direction (line 74) | def encode_direction(position, direction=None): function _sanitize_name (line 81) | def _sanitize_name(x: str) -> str: function integrate_samples (line 85) | def integrate_samples(volume_range, ts, density, channels): class VoidNeRFModel (line 119) | class VoidNeRFModel(nn.Module): method __init__ (line 124) | def __init__(self, background, channel_scale=255.0): method forward (line 130) | def forward(self, position): class VolumeRange (line 142) | class VolumeRange: method __post_init__ (line 147) | def __post_init__(self): method partition (line 150) | def partition(self, ts): class BoundingBoxVolume (line 174) | class BoundingBoxVolume(nn.Module): method __init__ (line 179) | def __init__( method intersect (line 205) | def intersect( class StratifiedRaySampler (line 258) | class StratifiedRaySampler(nn.Module): method __init__ (line 263) | def __init__(self, depth_mode: str = "linear"): method sample (line 271) | def sample( class ImportanceRaySampler (line 310) | class ImportanceRaySampler(nn.Module): method __init__ (line 315) | def __init__( method sample (line 338) | def sample(self, t0: torch.Tensor, t1: torch.Tensor, n_samples: int) -... class MLPNeRFModelOutput (line 372) | class MLPNeRFModelOutput(BaseOutput): class MLPNeRSTFModel (line 379) | class MLPNeRSTFModel(ModelMixin, ConfigMixin): method __init__ (line 381) | def __init__( method map_indices_to_keys (line 418) | def map_indices_to_keys(self, output): method forward (line 432) | def forward(self, *, position, direction, ts, nerf_level="coarse"): class ChannelsProj (line 471) | class ChannelsProj(nn.Module): method __init__ (line 472) | def __init__( method forward (line 486) | def forward(self, x: torch.Tensor) -> torch.Tensor: class ShapEParamsProjModel (line 497) | class ShapEParamsProjModel(ModelMixin, ConfigMixin): method __init__ (line 505) | def __init__( method forward (line 535) | def forward(self, x: torch.Tensor): class ShapERenderer (line 547) | class ShapERenderer(ModelMixin, ConfigMixin): method __init__ (line 549) | def __init__( method render_rays (line 588) | def render_rays(self, rays, sampler, n_samples, prev_model_out=None, r... method decode (line 667) | def decode( FILE: 6DoF/diffusers/pipelines/spectrogram_diffusion/continous_encoder.py class SpectrogramContEncoder (line 29) | class SpectrogramContEncoder(ModelMixin, ConfigMixin, ModuleUtilsMixin): method __init__ (line 31) | def __init__( method forward (line 71) | def forward(self, encoder_inputs, encoder_inputs_mask): FILE: 6DoF/diffusers/pipelines/spectrogram_diffusion/midi_utils.py class NoteRepresentationConfig (line 84) | class NoteRepresentationConfig: class NoteEventData (line 92) | class NoteEventData: class NoteEncodingState (line 101) | class NoteEncodingState: class EventRange (line 109) | class EventRange: class Event (line 116) | class Event: class Tokenizer (line 121) | class Tokenizer: method __init__ (line 122) | def __init__(self, regular_ids: int): method encode (line 127) | def encode(self, token_ids): class Codec (line 145) | class Codec: method __init__ (line 156) | def __init__(self, max_shift_steps: int, steps_per_second: float, even... method num_classes (line 172) | def num_classes(self) -> int: method is_shift_event_index (line 178) | def is_shift_event_index(self, index: int) -> bool: method max_shift_steps (line 182) | def max_shift_steps(self) -> int: method encode_event (line 185) | def encode_event(self, event: Event) -> int: method event_type_range (line 200) | def event_type_range(self, event_type: str) -> Tuple[int, int]: method decode_event_index (line 210) | def decode_event_index(self, index: int) -> Event: class ProgramGranularity (line 222) | class ProgramGranularity: function drop_programs (line 228) | def drop_programs(tokens, codec: Codec): function programs_to_midi_classes (line 234) | def programs_to_midi_classes(tokens, codec): function frame (line 254) | def frame(signal, frame_length, frame_step, pad_end=False, pad_value=0, ... function program_to_slakh_program (line 272) | def program_to_slakh_program(program): function audio_to_frames (line 279) | def audio_to_frames( function note_sequence_to_onsets_and_offsets_and_programs (line 302) | def note_sequence_to_onsets_and_offsets_and_programs( function num_velocity_bins_from_codec (line 332) | def num_velocity_bins_from_codec(codec: Codec): function segment (line 339) | def segment(a, n): function velocity_to_bin (line 343) | def velocity_to_bin(velocity, num_velocity_bins): function note_event_data_to_events (line 350) | def note_event_data_to_events( function note_encoding_state_to_events (line 382) | def note_encoding_state_to_events(state: NoteEncodingState) -> Sequence[... function encode_and_index_events (line 392) | def encode_and_index_events( function extract_sequence_with_indices (line 498) | def extract_sequence_with_indices(features, state_events_end_token=None,... function map_midi_programs (line 524) | def map_midi_programs( function run_length_encode_shifts_fn (line 534) | def run_length_encode_shifts_fn( function note_representation_processor_chain (line 604) | def note_representation_processor_chain(features, codec: Codec, note_rep... class MidiProcessor (line 619) | class MidiProcessor: method __init__ (line 620) | def __init__(self): method __call__ (line 635) | def __call__(self, midi: Union[bytes, os.PathLike, str]): FILE: 6DoF/diffusers/pipelines/spectrogram_diffusion/notes_encoder.py class SpectrogramNotesEncoder (line 25) | class SpectrogramNotesEncoder(ModelMixin, ConfigMixin, ModuleUtilsMixin): method __init__ (line 27) | def __init__( method forward (line 69) | def forward(self, encoder_input_tokens, encoder_inputs_mask): FILE: 6DoF/diffusers/pipelines/spectrogram_diffusion/pipeline_spectrogram_diffusion.py class SpectrogramDiffusionPipeline (line 40) | class SpectrogramDiffusionPipeline(DiffusionPipeline): method __init__ (line 43) | def __init__( method scale_features (line 66) | def scale_features(self, features, output_range=(-1.0, 1.0), clip=False): method scale_to_features (line 76) | def scale_to_features(self, outputs, input_range=(-1.0, 1.0), clip=Fal... method encode (line 85) | def encode(self, input_tokens, continuous_inputs, continuous_mask): method decode (line 97) | def decode(self, encodings_and_masks, input_tokens, noise_time): method __call__ (line 113) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/__init__.py class StableDiffusionPipelineOutput (line 22) | class StableDiffusionPipelineOutput(BaseOutput): class FlaxStableDiffusionPipelineOutput (line 117) | class FlaxStableDiffusionPipelineOutput(BaseOutput): FILE: 6DoF/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py function shave_segments (line 68) | def shave_segments(path, n_shave_prefix_segments=1): function renew_resnet_paths (line 78) | def renew_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_vae_resnet_paths (line 100) | def renew_vae_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_attention_paths (line 116) | def renew_attention_paths(old_list, n_shave_prefix_segments=0): function renew_vae_attention_paths (line 137) | def renew_vae_attention_paths(old_list, n_shave_prefix_segments=0): function assign_to_checkpoint (line 167) | def assign_to_checkpoint( function conv_attn_to_linear (line 222) | def conv_attn_to_linear(checkpoint): function create_unet_diffusers_config (line 234) | def create_unet_diffusers_config(original_config, image_size: int, contr... function create_vae_diffusers_config (line 334) | def create_vae_diffusers_config(original_config, image_size: int): function create_diffusers_schedular (line 358) | def create_diffusers_schedular(original_config): function create_ldm_bert_config (line 368) | def create_ldm_bert_config(original_config): function convert_ldm_unet_checkpoint (line 378) | def convert_ldm_unet_checkpoint( function convert_ldm_vae_checkpoint (line 620) | def convert_ldm_vae_checkpoint(checkpoint, config): function convert_ldm_bert_checkpoint (line 727) | def convert_ldm_bert_checkpoint(checkpoint, config): function convert_ldm_clip_checkpoint (line 777) | def convert_ldm_clip_checkpoint(checkpoint, local_files_only=False, text... function convert_paint_by_example_checkpoint (line 827) | def convert_paint_by_example_checkpoint(checkpoint): function convert_open_clip_checkpoint (line 894) | def convert_open_clip_checkpoint( function stable_unclip_image_encoder (line 959) | def stable_unclip_image_encoder(original_config): function stable_unclip_image_noising_components (line 992) | def stable_unclip_image_noising_components( function convert_controlnet_checkpoint (line 1037) | def convert_controlnet_checkpoint( function download_from_original_stable_diffusion_ckpt (line 1081) | def download_from_original_stable_diffusion_ckpt( function download_controlnet_from_original_ckpt (line 1563) | def download_controlnet_from_original_ckpt( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py function preprocess (line 42) | def preprocess(image): function posterior_sample (line 68) | def posterior_sample(scheduler, latents, timestep, clean_latents, genera... function compute_noise (line 95) | def compute_noise(scheduler, prev_latents, latents, timestep, noise_pred... class CycleDiffusionPipeline (line 129) | class CycleDiffusionPipeline(DiffusionPipeline, TextualInversionLoaderMi... method __init__ (line 158) | def __init__( method enable_sequential_cpu_offload (line 235) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 261) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 291) | def _execution_device(self): method _encode_prompt (line 309) | def _encode_prompt( method check_inputs (line 464) | def check_inputs( method prepare_extra_step_kwargs (line 505) | def prepare_extra_step_kwargs(self, generator, eta): method run_safety_checker (line 523) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 538) | def decode_latents(self, latents): method get_timesteps (line 552) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 561) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 616) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py class FlaxStableDiffusionPipeline (line 81) | class FlaxStableDiffusionPipeline(FlaxDiffusionPipeline): method __init__ (line 110) | def __init__( method prepare_inputs (line 168) | def prepare_inputs(self, prompt: Union[str, List[str]]): method _get_has_nsfw_concepts (line 181) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 185) | def _run_safety_checker(self, images, safety_model_params, jit=False): method _generate (line 215) | def _generate( method __call__ (line 312) | def __call__( function _p_generate (line 436) | def _p_generate( function _p_get_has_nsfw_concepts (line 462) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 466) | def unshard(x: jnp.ndarray): FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py class FlaxStableDiffusionImg2ImgPipeline (line 105) | class FlaxStableDiffusionImg2ImgPipeline(FlaxDiffusionPipeline): method __init__ (line 134) | def __init__( method prepare_inputs (line 171) | def prepare_inputs(self, prompt: Union[str, List[str]], image: Union[I... method _get_has_nsfw_concepts (line 192) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 196) | def _run_safety_checker(self, images, safety_model_params, jit=False): method get_timestep_start (line 226) | def get_timestep_start(self, num_inference_steps, strength): method _generate (line 234) | def _generate( method __call__ (line 339) | def __call__( function _p_generate (line 480) | def _p_generate( function _p_get_has_nsfw_concepts (line 510) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 514) | def unshard(x: jnp.ndarray): function preprocess (line 521) | def preprocess(image, dtype): FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_inpaint.py class FlaxStableDiffusionInpaintPipeline (line 102) | class FlaxStableDiffusionInpaintPipeline(FlaxDiffusionPipeline): method __init__ (line 131) | def __init__( method prepare_inputs (line 189) | def prepare_inputs( method _get_has_nsfw_concepts (line 228) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 232) | def _run_safety_checker(self, images, safety_model_params, jit=False): method _generate (line 262) | def _generate( method __call__ (line 390) | def __call__( function _p_generate (line 523) | def _p_generate( function _p_get_has_nsfw_concepts (line 553) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 557) | def unshard(x: jnp.ndarray): function preprocess_image (line 564) | def preprocess_image(image, dtype): function preprocess_mask (line 573) | def preprocess_mask(mask, dtype): FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py class OnnxStableDiffusionPipeline (line 33) | class OnnxStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 45) | def __init__( method _encode_prompt (line 114) | def _encode_prompt( method check_inputs (line 217) | def check_inputs( method __call__ (line 264) | def __call__( class StableDiffusionOnnxPipeline (line 462) | class StableDiffusionOnnxPipeline(OnnxStableDiffusionPipeline): method __init__ (line 463) | def __init__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py function preprocess (line 36) | def preprocess(image): class OnnxStableDiffusionImg2ImgPipeline (line 64) | class OnnxStableDiffusionImg2ImgPipeline(DiffusionPipeline): method __init__ (line 102) | def __init__( method _encode_prompt (line 172) | def _encode_prompt( method check_inputs (line 275) | def check_inputs( method __call__ (line 317) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py function prepare_mask_and_masked_image (line 38) | def prepare_mask_and_masked_image(image, mask, latents_shape): class OnnxStableDiffusionInpaintPipeline (line 56) | class OnnxStableDiffusionInpaintPipeline(DiffusionPipeline): method __init__ (line 94) | def __init__( method _encode_prompt (line 165) | def _encode_prompt( method check_inputs (line 269) | def check_inputs( method __call__ (line 317) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint_legacy.py function preprocess (line 20) | def preprocess(image): function preprocess_mask (line 29) | def preprocess_mask(mask, scale_factor=8): class OnnxStableDiffusionInpaintPipelineLegacy (line 41) | class OnnxStableDiffusionInpaintPipelineLegacy(DiffusionPipeline): method __init__ (line 80) | def __init__( method _encode_prompt (line 150) | def _encode_prompt( method check_inputs (line 253) | def check_inputs( method __call__ (line 295) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_upscale.py function preprocess (line 26) | def preprocess(image): class OnnxStableDiffusionUpscalePipeline (line 48) | class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline): method __init__ (line 49) | def __init__( method __call__ (line 72) | def __call__( method decode_latents (line 281) | def decode_latents(self, latents): method _encode_prompt (line 288) | def _encode_prompt( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py function rescale_noise_cfg (line 58) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0): class StableDiffusionPipeline (line 72) | class StableDiffusionPipeline(DiffusionPipeline, TextualInversionLoaderM... method __init__ (line 109) | def __init__( method enable_vae_slicing (line 199) | def enable_vae_slicing(self): method disable_vae_slicing (line 208) | def disable_vae_slicing(self): method enable_vae_tiling (line 215) | def enable_vae_tiling(self): method disable_vae_tiling (line 224) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 231) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 256) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 285) | def _execution_device(self): method _encode_prompt (line 302) | def _encode_prompt( method run_safety_checker (line 456) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 470) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 483) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 500) | def check_inputs( method prepare_latents (line 547) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 566) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py class AttentionStore (line 74) | class AttentionStore: method get_empty_store (line 76) | def get_empty_store(): method __call__ (line 79) | def __call__(self, attn, is_cross: bool, place_in_unet: str): method between_steps (line 89) | def between_steps(self): method get_average_attention (line 93) | def get_average_attention(self): method aggregate_attention (line 97) | def aggregate_attention(self, from_where: List[str]) -> torch.Tensor: method reset (line 109) | def reset(self): method __init__ (line 114) | def __init__(self, attn_res): class AttendExciteAttnProcessor (line 127) | class AttendExciteAttnProcessor: method __init__ (line 128) | def __init__(self, attnstore, place_in_unet): method __call__ (line 133) | def __call__(self, attn: Attention, hidden_states, encoder_hidden_stat... class StableDiffusionAttendAndExcitePipeline (line 165) | class StableDiffusionAttendAndExcitePipeline(DiffusionPipeline, TextualI... method __init__ (line 194) | def __init__( method enable_vae_slicing (line 237) | def enable_vae_slicing(self): method disable_vae_slicing (line 247) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 255) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 282) | def _execution_device(self): method _encode_prompt (line 300) | def _encode_prompt( method run_safety_checker (line 455) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 470) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 484) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 501) | def check_inputs( method prepare_latents (line 575) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method _compute_max_attention_per_index (line 593) | def _compute_max_attention_per_index( method _aggregate_and_get_max_attention_per_token (line 615) | def _aggregate_and_get_max_attention_per_token( method _compute_loss (line 630) | def _compute_loss(max_attention_per_index: List[torch.Tensor]) -> torc... method _update_latent (line 637) | def _update_latent(latents: torch.Tensor, loss: torch.Tensor, step_siz... method _perform_iterative_refinement_step (line 643) | def _perform_iterative_refinement_step( method register_attention_control (line 697) | def register_attention_control(self): method get_indices (line 716) | def get_indices(self, prompt: str) -> Dict[str, int]: method __call__ (line 724) | def __call__( class GaussianSmoothing (line 1015) | class GaussianSmoothing(torch.nn.Module): method __init__ (line 1028) | def __init__( method forward (line 1069) | def forward(self, input): FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py function preprocess (line 39) | def preprocess(image): class StableDiffusionDepth2ImgPipeline (line 65) | class StableDiffusionDepth2ImgPipeline(DiffusionPipeline, TextualInversi... method __init__ (line 95) | def __init__( method enable_sequential_cpu_offload (line 140) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 159) | def _execution_device(self): method _encode_prompt (line 177) | def _encode_prompt( method run_safety_checker (line 332) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 347) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 361) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 379) | def check_inputs( method get_timesteps (line 420) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 430) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method prepare_depth_map (line 487) | def prepare_depth_map(self, image, depth_map, batch_size, do_classifie... method __call__ (line 532) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_diffedit.py class DiffEditInversionPipelineOutput (line 50) | class DiffEditInversionPipelineOutput(BaseOutput): function auto_corr_loss (line 140) | def auto_corr_loss(hidden_states, generator=None): function kl_divergence (line 156) | def kl_divergence(hidden_states): function preprocess (line 161) | def preprocess(image): function preprocess_mask (line 187) | def preprocess_mask(mask, batch_size: int = 1): class StableDiffusionDiffEditPipeline (line 240) | class StableDiffusionDiffEditPipeline(DiffusionPipeline, TextualInversio... method __init__ (line 277) | def __init__( method enable_vae_slicing (line 371) | def enable_vae_slicing(self): method disable_vae_slicing (line 381) | def disable_vae_slicing(self): method enable_vae_tiling (line 389) | def enable_vae_tiling(self): method disable_vae_tiling (line 399) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 407) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 433) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 463) | def _execution_device(self): method _encode_prompt (line 481) | def _encode_prompt( method run_safety_checker (line 636) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 651) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 669) | def decode_latents(self, latents): method check_inputs (line 682) | def check_inputs( method check_source_inputs (line 730) | def check_source_inputs( method get_timesteps (line 766) | def get_timesteps(self, num_inference_steps, strength, device): method get_inverse_timesteps (line 775) | def get_inverse_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 789) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_image_latents (line 807) | def prepare_image_latents(self, image, batch_size, dtype, device, gene... method get_epsilon (line 856) | def get_epsilon(self, model_output: torch.Tensor, sample: torch.Tensor... method generate_mask (line 874) | def generate_mask( method invert (line 1097) | def invert( method __call__ (line 1335) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py class StableDiffusionImageVariationPipeline (line 37) | class StableDiffusionImageVariationPipeline(DiffusionPipeline): method __init__ (line 65) | def __init__( method enable_sequential_cpu_offload (line 126) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 145) | def _execution_device(self): method _encode_image (line 162) | def _encode_image(self, image, device, num_images_per_prompt, do_class... method run_safety_checker (line 188) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 203) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 217) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 234) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 257) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 275) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py function preprocess (line 75) | def preprocess(image): class StableDiffusionImg2ImgPipeline (line 101) | class StableDiffusionImg2ImgPipeline( method __init__ (line 140) | def __init__( method enable_sequential_cpu_offload (line 231) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 257) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 287) | def _execution_device(self): method _encode_prompt (line 305) | def _encode_prompt( method run_safety_checker (line 460) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 475) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 489) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 506) | def check_inputs( method get_timesteps (line 546) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 555) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 614) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py function prepare_mask_and_masked_image (line 39) | def prepare_mask_and_masked_image(image, mask, height, width, return_ima... class StableDiffusionInpaintPipeline (line 156) | class StableDiffusionInpaintPipeline( method __init__ (line 204) | def __init__( method enable_sequential_cpu_offload (line 300) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 326) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 356) | def _execution_device(self): method _encode_prompt (line 374) | def _encode_prompt( method run_safety_checker (line 529) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 544) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 562) | def decode_latents(self, latents): method check_inputs (line 575) | def check_inputs( method prepare_latents (line 626) | def prepare_latents( method _encode_vae_image (line 679) | def _encode_vae_image(self, image: torch.Tensor, generator: torch.Gene... method prepare_mask_latents (line 693) | def prepare_mask_latents( method get_timesteps (line 735) | def get_timesteps(self, num_inference_steps, strength, device): method __call__ (line 745) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py function preprocess_image (line 46) | def preprocess_image(image, batch_size): function preprocess_mask (line 56) | def preprocess_mask(mask, batch_size, scale_factor=8): class StableDiffusionInpaintPipelineLegacy (line 87) | class StableDiffusionInpaintPipelineLegacy( method __init__ (line 126) | def __init__( method enable_sequential_cpu_offload (line 224) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 250) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 280) | def _execution_device(self): method _encode_prompt (line 298) | def _encode_prompt( method run_safety_checker (line 453) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 468) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 482) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 500) | def check_inputs( method get_timesteps (line 541) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 550) | def prepare_latents(self, image, timestep, num_images_per_prompt, dtyp... method __call__ (line 567) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py function preprocess (line 45) | def preprocess(image): class StableDiffusionInstructPix2PixPipeline (line 71) | class StableDiffusionInstructPix2PixPipeline(DiffusionPipeline, TextualI... method __init__ (line 107) | def __init__( method __call__ (line 150) | def __call__( method enable_sequential_cpu_offload (line 432) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 458) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 488) | def _execution_device(self): method _encode_prompt (line 505) | def _encode_prompt( method run_safety_checker (line 653) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 668) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 686) | def decode_latents(self, latents): method check_inputs (line 699) | def check_inputs( method prepare_latents (line 737) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_image_latents (line 754) | def prepare_image_latents( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py class ModelWrapper (line 34) | class ModelWrapper: method __init__ (line 35) | def __init__(self, model, alphas_cumprod): method apply_model (line 39) | def apply_model(self, *args, **kwargs): class StableDiffusionKDiffusionPipeline (line 48) | class StableDiffusionKDiffusionPipeline(DiffusionPipeline, TextualInvers... method __init__ (line 83) | def __init__( method set_scheduler (line 124) | def set_scheduler(self, scheduler_type: str): method enable_sequential_cpu_offload (line 130) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 156) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 186) | def _execution_device(self): method _encode_prompt (line 204) | def _encode_prompt( method run_safety_checker (line 359) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 374) | def decode_latents(self, latents): method check_inputs (line 388) | def check_inputs( method prepare_latents (line 435) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 448) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py function preprocess (line 35) | def preprocess(image): class StableDiffusionLatentUpscalePipeline (line 61) | class StableDiffusionLatentUpscalePipeline(DiffusionPipeline): method __init__ (line 84) | def __init__( method enable_sequential_cpu_offload (line 104) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 123) | def _execution_device(self): method _encode_prompt (line 140) | def _encode_prompt(self, prompt, device, do_classifier_free_guidance, ... method decode_latents (line 231) | def decode_latents(self, latents): method check_inputs (line 244) | def check_inputs(self, prompt, image, callback_steps): method prepare_latents (line 282) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 296) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py class LDM3DPipelineOutput (line 59) | class LDM3DPipelineOutput(BaseOutput): class StableDiffusionLDM3DPipeline (line 77) | class StableDiffusionLDM3DPipeline( method __init__ (line 118) | def __init__( method enable_vae_slicing (line 161) | def enable_vae_slicing(self): method disable_vae_slicing (line 171) | def disable_vae_slicing(self): method enable_vae_tiling (line 179) | def enable_vae_tiling(self): method disable_vae_tiling (line 189) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 196) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 222) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 252) | def _execution_device(self): method _encode_prompt (line 270) | def _encode_prompt( method run_safety_checker (line 424) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 440) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 458) | def check_inputs( method prepare_latents (line 505) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 524) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_model_editing.py class StableDiffusionModelEditingPipeline (line 58) | class StableDiffusionModelEditingPipeline(DiffusionPipeline, TextualInve... method __init__ (line 90) | def __init__( method enable_vae_slicing (line 168) | def enable_vae_slicing(self): method disable_vae_slicing (line 178) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 186) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 213) | def _execution_device(self): method _encode_prompt (line 231) | def _encode_prompt( method run_safety_checker (line 386) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 401) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 415) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 433) | def check_inputs( method prepare_latents (line 481) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method edit_model (line 499) | def edit_model( method __call__ (line 617) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_panorama.py class StableDiffusionPanoramaPipeline (line 54) | class StableDiffusionPanoramaPipeline(DiffusionPipeline, TextualInversio... method __init__ (line 87) | def __init__( method enable_vae_slicing (line 130) | def enable_vae_slicing(self): method disable_vae_slicing (line 140) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 148) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 175) | def _execution_device(self): method _encode_prompt (line 193) | def _encode_prompt( method run_safety_checker (line 348) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 363) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 377) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 395) | def check_inputs( method prepare_latents (line 443) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method get_views (line 460) | def get_views(self, panorama_height, panorama_width, window_size=64, s... method __call__ (line 479) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_paradigms.py class StableDiffusionParadigmsPipeline (line 62) | class StableDiffusionParadigmsPipeline( method __init__ (line 104) | def __init__( method enable_vae_slicing (line 150) | def enable_vae_slicing(self): method disable_vae_slicing (line 160) | def disable_vae_slicing(self): method enable_vae_tiling (line 168) | def enable_vae_tiling(self): method disable_vae_tiling (line 178) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 186) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 212) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 242) | def _execution_device(self): method _encode_prompt (line 260) | def _encode_prompt( method run_safety_checker (line 415) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 430) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 448) | def check_inputs( method prepare_latents (line 496) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method _cumsum (line 513) | def _cumsum(self, input, dim, debug=False): method __call__ (line 523) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py class Pix2PixInversionPipelineOutput (line 57) | class Pix2PixInversionPipelineOutput(BaseOutput, TextualInversionLoaderM... function preprocess (line 179) | def preprocess(image): function prepare_unet (line 205) | def prepare_unet(unet: UNet2DConditionModel): class Pix2PixZeroL2Loss (line 222) | class Pix2PixZeroL2Loss: method __init__ (line 223) | def __init__(self): method compute_loss (line 226) | def compute_loss(self, predictions, targets): class Pix2PixZeroAttnProcessor (line 230) | class Pix2PixZeroAttnProcessor: method __init__ (line 234) | def __init__(self, is_pix2pix_zero=False): method __call__ (line 239) | def __call__( class StableDiffusionPix2PixZeroPipeline (line 285) | class StableDiffusionPix2PixZeroPipeline(DiffusionPipeline): method __init__ (line 323) | def __init__( method enable_sequential_cpu_offload (line 372) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 397) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 423) | def _execution_device(self): method _encode_prompt (line 441) | def _encode_prompt( method run_safety_checker (line 596) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 611) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 625) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 642) | def check_inputs( method prepare_latents (line 673) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method generate_caption (line 691) | def generate_caption(self, images): method construct_direction (line 710) | def construct_direction(self, embs_source: torch.Tensor, embs_target: ... method get_embeds (line 715) | def get_embeds(self, prompt: List[str], batch_size: int = 16) -> torch... method prepare_image_latents (line 734) | def prepare_image_latents(self, image, batch_size, dtype, device, gene... method get_epsilon (line 783) | def get_epsilon(self, model_output: torch.Tensor, sample: torch.Tensor... method auto_corr_loss (line 800) | def auto_corr_loss(self, hidden_states, generator=None): method kl_divergence (line 815) | def kl_divergence(self, hidden_states): method __call__ (line 822) | def __call__( method invert (line 1101) | def invert( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_sag.py class CrossAttnStoreProcessor (line 53) | class CrossAttnStoreProcessor: method __init__ (line 54) | def __init__(self): method __call__ (line 57) | def __call__( class StableDiffusionSAGPipeline (line 93) | class StableDiffusionSAGPipeline(DiffusionPipeline, TextualInversionLoad... method __init__ (line 122) | def __init__( method enable_vae_slicing (line 149) | def enable_vae_slicing(self): method disable_vae_slicing (line 159) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 167) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 194) | def _execution_device(self): method _encode_prompt (line 212) | def _encode_prompt( method run_safety_checker (line 367) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 382) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 396) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 414) | def check_inputs( method prepare_latents (line 462) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 481) | def __call__( method sag_masking (line 724) | def sag_masking(self, original_latents, attn_map, map_size, t, eps): method pred_x0 (line 754) | def pred_x0(self, sample, model_output, timestep): method pred_epsilon (line 774) | def pred_epsilon(self, sample, model_output, timestep): function gaussian_blur_2d (line 794) | def gaussian_blur_2d(img, kernel_size, sigma): FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py function preprocess (line 42) | def preprocess(image): class StableDiffusionUpscalePipeline (line 68) | class StableDiffusionUpscalePipeline(DiffusionPipeline, TextualInversion... method __init__ (line 95) | def __init__( method enable_sequential_cpu_offload (line 143) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 160) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 188) | def _execution_device(self): method run_safety_checker (line 206) | def run_safety_checker(self, image, device, dtype): method _encode_prompt (line 223) | def _encode_prompt( method prepare_extra_step_kwargs (line 378) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 396) | def decode_latents(self, latents): method check_inputs (line 409) | def check_inputs( method prepare_latents (line 491) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 505) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py class StableUnCLIPPipeline (line 53) | class StableUnCLIPPipeline(DiffusionPipeline, TextualInversionLoaderMixi... method __init__ (line 106) | def __init__( method enable_vae_slicing (line 144) | def enable_vae_slicing(self): method disable_vae_slicing (line 154) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 161) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 185) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 212) | def _execution_device(self): method _encode_prior_prompt (line 230) | def _encode_prior_prompt( method _encode_prompt (line 332) | def _encode_prompt( method decode_latents (line 487) | def decode_latents(self, latents): method prepare_prior_extra_step_kwargs (line 501) | def prepare_prior_extra_step_kwargs(self, generator, eta): method prepare_extra_step_kwargs (line 519) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 536) | def check_inputs( method prepare_latents (line 597) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method noise_image_embeddings (line 608) | def noise_image_embeddings( method __call__ (line 656) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py class StableUnCLIPImg2ImgPipeline (line 66) | class StableUnCLIPImg2ImgPipeline(DiffusionPipeline, TextualInversionLoa... method __init__ (line 112) | def __init__( method enable_vae_slicing (line 146) | def enable_vae_slicing(self): method disable_vae_slicing (line 156) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 163) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 187) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 214) | def _execution_device(self): method _encode_prompt (line 232) | def _encode_prompt( method _encode_image (line 386) | def _encode_image( method decode_latents (line 442) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 456) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 473) | def check_inputs( method prepare_latents (line 557) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method noise_image_embeddings (line 575) | def noise_image_embeddings( method __call__ (line 623) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/safety_checker.py function cosine_distance (line 26) | def cosine_distance(image_embeds, text_embeds): class StableDiffusionSafetyChecker (line 32) | class StableDiffusionSafetyChecker(PreTrainedModel): method __init__ (line 37) | def __init__(self, config: CLIPConfig): method forward (line 50) | def forward(self, clip_input, images): method forward_onnx (line 102) | def forward_onnx(self, clip_input: torch.FloatTensor, images: torch.Fl... FILE: 6DoF/diffusers/pipelines/stable_diffusion/safety_checker_flax.py function jax_cosine_distance (line 25) | def jax_cosine_distance(emb_1, emb_2, eps=1e-12): class FlaxStableDiffusionSafetyCheckerModule (line 31) | class FlaxStableDiffusionSafetyCheckerModule(nn.Module): method setup (line 35) | def setup(self): method __call__ (line 47) | def __call__(self, clip_input): class FlaxStableDiffusionSafetyChecker (line 71) | class FlaxStableDiffusionSafetyChecker(FlaxPreTrainedModel): method __init__ (line 76) | def __init__( method init_weights (line 90) | def init_weights(self, rng: jax.random.KeyArray, input_shape: Tuple, p... method __call__ (line 101) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion/stable_unclip_image_normalizer.py class StableUnCLIPImageNormalizer (line 24) | class StableUnCLIPImageNormalizer(ModelMixin, ConfigMixin): method __init__ (line 33) | def __init__( method to (line 42) | def to( method scale (line 51) | def scale(self, embeds): method unscale (line 55) | def unscale(self, embeds): FILE: 6DoF/diffusers/pipelines/stable_diffusion_safe/__init__.py class SafetyConfig (line 13) | class SafetyConfig(object): class StableDiffusionSafePipelineOutput (line 45) | class StableDiffusionSafePipelineOutput(BaseOutput): FILE: 6DoF/diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py class StableDiffusionPipelineSafe (line 22) | class StableDiffusionPipelineSafe(DiffusionPipeline): method __init__ (line 54) | def __init__( method safety_concept (line 150) | def safety_concept(self): method safety_concept (line 160) | def safety_concept(self, concept): method enable_sequential_cpu_offload (line 170) | def enable_sequential_cpu_offload(self): method _execution_device (line 189) | def _execution_device(self): method _encode_prompt (line 206) | def _encode_prompt( method run_safety_checker (line 341) | def run_safety_checker(self, image, device, dtype, enable_safety_guida... method decode_latents (line 365) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 379) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 397) | def check_inputs( method prepare_latents (line 445) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method perform_safety_guidance (line 462) | def perform_safety_guidance( method __call__ (line 505) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion_safe/safety_checker.py function cosine_distance (line 25) | def cosine_distance(image_embeds, text_embeds): class SafeStableDiffusionSafetyChecker (line 31) | class SafeStableDiffusionSafetyChecker(PreTrainedModel): method __init__ (line 36) | def __init__(self, config: CLIPConfig): method forward (line 49) | def forward(self, clip_input, images): method forward_onnx (line 88) | def forward_onnx(self, clip_input: torch.FloatTensor, images: torch.Fl... FILE: 6DoF/diffusers/pipelines/stable_diffusion_xl/__init__.py class StableDiffusionXLPipelineOutput (line 11) | class StableDiffusionXLPipelineOutput(BaseOutput): FILE: 6DoF/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py function rescale_noise_cfg (line 62) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0): class StableDiffusionXLPipeline (line 76) | class StableDiffusionXLPipeline(DiffusionPipeline, FromSingleFileMixin): method __init__ (line 107) | def __init__( method enable_vae_slicing (line 137) | def enable_vae_slicing(self): method disable_vae_slicing (line 147) | def disable_vae_slicing(self): method enable_vae_tiling (line 155) | def enable_vae_tiling(self): method disable_vae_tiling (line 165) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 172) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 194) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 226) | def _execution_device(self): method encode_prompt (line 243) | def encode_prompt( method prepare_extra_step_kwargs (line 430) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 447) | def check_inputs( method prepare_latents (line 507) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method _get_add_time_ids (line 524) | def _get_add_time_ids(self, original_size, crops_coords_top_left, targ... method __call__ (line 542) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py function rescale_noise_cfg (line 67) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0): class StableDiffusionXLImg2ImgPipeline (line 81) | class StableDiffusionXLImg2ImgPipeline(DiffusionPipeline, FromSingleFile... method __init__ (line 113) | def __init__( method enable_vae_slicing (line 144) | def enable_vae_slicing(self): method disable_vae_slicing (line 154) | def disable_vae_slicing(self): method enable_vae_tiling (line 162) | def enable_vae_tiling(self): method disable_vae_tiling (line 172) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 180) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 203) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 235) | def _execution_device(self): method encode_prompt (line 253) | def encode_prompt( method prepare_extra_step_kwargs (line 440) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 457) | def check_inputs( method get_timesteps (line 497) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 506) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method _get_add_time_ids (line 568) | def _get_add_time_ids( method __call__ (line 609) | def __call__( FILE: 6DoF/diffusers/pipelines/stable_diffusion_xl/watermark.py class StableDiffusionXLWatermarker (line 12) | class StableDiffusionXLWatermarker: method __init__ (line 13) | def __init__(self): method apply_watermark (line 19) | def apply_watermark(self, images: torch.FloatTensor): FILE: 6DoF/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py class KarrasVePipeline (line 25) | class KarrasVePipeline(DiffusionPipeline): method __init__ (line 44) | def __init__(self, unet: UNet2DModel, scheduler: KarrasVeScheduler): method __call__ (line 49) | def __call__( FILE: 6DoF/diffusers/pipelines/text_to_video_synthesis/__init__.py class TextToVideoSDPipelineOutput (line 11) | class TextToVideoSDPipelineOutput(BaseOutput): FILE: 6DoF/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py function tensor2vid (line 58) | def tensor2vid(video: torch.Tensor, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5,... class TextToVideoSDPipeline (line 76) | class TextToVideoSDPipeline(DiffusionPipeline, TextualInversionLoaderMix... method __init__ (line 97) | def __init__( method enable_vae_slicing (line 117) | def enable_vae_slicing(self): method disable_vae_slicing (line 127) | def disable_vae_slicing(self): method enable_vae_tiling (line 135) | def enable_vae_tiling(self): method disable_vae_tiling (line 145) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 152) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 173) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 200) | def _execution_device(self): method _encode_prompt (line 218) | def _encode_prompt( method decode_latents (line 372) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 396) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 414) | def check_inputs( method prepare_latents (line 461) | def prepare_latents( method __call__ (line 488) | def __call__( FILE: 6DoF/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth_img2img.py function tensor2vid (line 79) | def tensor2vid(video: torch.Tensor, mean=[0.5, 0.5, 0.5], std=[0.5, 0.5,... function preprocess_video (line 97) | def preprocess_video(video): class VideoToVideoSDPipeline (line 138) | class VideoToVideoSDPipeline(DiffusionPipeline, TextualInversionLoaderMi... method __init__ (line 159) | def __init__( method enable_vae_slicing (line 179) | def enable_vae_slicing(self): method disable_vae_slicing (line 189) | def disable_vae_slicing(self): method enable_vae_tiling (line 197) | def enable_vae_tiling(self): method disable_vae_tiling (line 207) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 214) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 235) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 262) | def _execution_device(self): method _encode_prompt (line 280) | def _encode_prompt( method decode_latents (line 435) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 459) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 477) | def check_inputs( method get_timesteps (line 518) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 527) | def prepare_latents(self, video, timestep, batch_size, dtype, device, ... method __call__ (line 573) | def __call__( FILE: 6DoF/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py function rearrange_0 (line 18) | def rearrange_0(tensor, f): function rearrange_1 (line 24) | def rearrange_1(tensor): function rearrange_3 (line 29) | def rearrange_3(tensor, f): function rearrange_4 (line 34) | def rearrange_4(tensor): class CrossFrameAttnProcessor (line 39) | class CrossFrameAttnProcessor: method __init__ (line 49) | def __init__(self, batch_size=2): method __call__ (line 52) | def __call__(self, attn, hidden_states, encoder_hidden_states=None, at... class CrossFrameAttnProcessor2_0 (line 98) | class CrossFrameAttnProcessor2_0: method __init__ (line 108) | def __init__(self, batch_size=2): method __call__ (line 113) | def __call__(self, attn, hidden_states, encoder_hidden_states=None, at... class TextToVideoPipelineOutput (line 174) | class TextToVideoPipelineOutput(BaseOutput): function coords_grid (line 179) | def coords_grid(batch, ht, wd, device): function warp_single_latent (line 186) | def warp_single_latent(latent, reference_flow): function create_motion_field (line 213) | def create_motion_field(motion_field_strength_x, motion_field_strength_y... function create_motion_field_and_warp_latents (line 236) | def create_motion_field_and_warp_latents(motion_field_strength_x, motion... class TextToVideoZeroPipeline (line 263) | class TextToVideoZeroPipeline(StableDiffusionPipeline): method __init__ (line 291) | def __init__( method forward_loop (line 312) | def forward_loop(self, x_t0, t0, t1, generator): method backward_loop (line 330) | def backward_loop( method __call__ (line 400) | def __call__( FILE: 6DoF/diffusers/pipelines/unclip/pipeline_unclip.py class UnCLIPPipeline (line 34) | class UnCLIPPipeline(DiffusionPipeline): method __init__ (line 78) | def __init__( method prepare_latents (line 106) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 117) | def _encode_prompt( method enable_sequential_cpu_offload (line 208) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 234) | def _execution_device(self): method __call__ (line 252) | def __call__( FILE: 6DoF/diffusers/pipelines/unclip/pipeline_unclip_image_variation.py class UnCLIPImageVariationPipeline (line 38) | class UnCLIPImageVariationPipeline(DiffusionPipeline): method __init__ (line 84) | def __init__( method prepare_latents (line 113) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 124) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method _encode_image (line 187) | def _encode_image(self, image, device, num_images_per_prompt, image_em... method enable_sequential_cpu_offload (line 201) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 227) | def _execution_device(self): method __call__ (line 245) | def __call__( FILE: 6DoF/diffusers/pipelines/unclip/text_proj.py class UnCLIPTextProjModel (line 22) | class UnCLIPTextProjModel(ModelMixin, ConfigMixin): method __init__ (line 31) | def __init__( method forward (line 55) | def forward(self, *, image_embeddings, prompt_embeds, text_encoder_hid... FILE: 6DoF/diffusers/pipelines/unidiffuser/modeling_text_decoder.py class UniDiffuserTextDecoder (line 14) | class UniDiffuserTextDecoder(ModelMixin, ConfigMixin, ModuleUtilsMixin): method __init__ (line 67) | def __init__( method forward (line 131) | def forward( method get_dummy_token (line 163) | def get_dummy_token(self, batch_size: int, device: torch.device) -> to... method encode (line 166) | def encode(self, prefix): method generate_captions (line 170) | def generate_captions(self, features, eos_token_id, device): method generate_beam (line 202) | def generate_beam( FILE: 6DoF/diffusers/pipelines/unidiffuser/modeling_uvit.py function _no_grad_trunc_normal_ (line 19) | def _no_grad_trunc_normal_(tensor, mean, std, a, b): function trunc_normal_ (line 56) | def trunc_normal_(tensor, mean=0.0, std=1.0, a=-2.0, b=2.0): class PatchEmbed (line 75) | class PatchEmbed(nn.Module): method __init__ (line 78) | def __init__( method forward (line 109) | def forward(self, latent): class SkipBlock (line 121) | class SkipBlock(nn.Module): method __init__ (line 122) | def __init__(self, dim: int): method forward (line 130) | def forward(self, x, skip): class UTransformerBlock (line 140) | class UTransformerBlock(nn.Module): method __init__ (line 174) | def __init__( method forward (line 251) | def forward( class UniDiffuserBlock (line 333) | class UniDiffuserBlock(nn.Module): method __init__ (line 369) | def __init__( method forward (line 446) | def forward( class UTransformer2DModel (line 530) | class UTransformer2DModel(ModelMixin, ConfigMixin): method __init__ (line 589) | def __init__( method forward (line 727) | def forward( class UniDiffuserModel (line 832) | class UniDiffuserModel(ModelMixin, ConfigMixin): method __init__ (line 897) | def __init__( method no_weight_decay (line 1034) | def no_weight_decay(self): method forward (line 1037) | def forward( FILE: 6DoF/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py function preprocess (line 37) | def preprocess(image): class ImageTextPipelineOutput (line 65) | class ImageTextPipelineOutput(BaseOutput): class UniDiffuserPipeline (line 82) | class UniDiffuserPipeline(DiffusionPipeline): method __init__ (line 126) | def __init__( method enable_sequential_cpu_offload (line 177) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 204) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 234) | def _execution_device(self): method prepare_extra_step_kwargs (line 252) | def prepare_extra_step_kwargs(self, generator, eta): method _infer_mode (line 269) | def _infer_mode(self, prompt, prompt_embeds, image, latents, prompt_la... method set_text_mode (line 328) | def set_text_mode(self): method set_image_mode (line 332) | def set_image_mode(self): method set_text_to_image_mode (line 336) | def set_text_to_image_mode(self): method set_image_to_text_mode (line 340) | def set_image_to_text_mode(self): method set_joint_mode (line 344) | def set_joint_mode(self): method reset_mode (line 348) | def reset_mode(self): method _infer_batch_size (line 352) | def _infer_batch_size( method _encode_prompt (line 430) | def _encode_prompt( method encode_image_vae_latents (line 570) | def encode_image_vae_latents( method encode_image_clip_latents (line 630) | def encode_image_clip_latents( method decode_image_latents (line 689) | def decode_image_latents(self, latents): method prepare_text_latents (line 697) | def prepare_text_latents( method prepare_image_vae_latents (line 721) | def prepare_image_vae_latents( method prepare_image_clip_latents (line 756) | def prepare_image_clip_latents( method _split (line 778) | def _split(self, x, height, width): method _combine (line 794) | def _combine(self, img_vae, img_clip): method _split_joint (line 803) | def _split_joint(self, x, height, width): method _combine_joint (line 822) | def _combine_joint(self, img_vae, img_clip, text): method _get_noise_pred (line 833) | def _get_noise_pred( method check_latents_shape (line 955) | def check_latents_shape(self, latents_name, latents, expected_shape): method check_inputs (line 971) | def check_inputs( method __call__ (line 1085) | def __call__( FILE: 6DoF/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py function get_down_block (line 38) | def get_down_block( function get_up_block (line 97) | def get_up_block( class UNetFlatConditionModel (line 157) | class UNetFlatConditionModel(ModelMixin, ConfigMixin): method __init__ (line 248) | def __init__( method attn_processors (line 674) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 697) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 731) | def set_default_attn_processor(self): method set_attention_slice (line 737) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 802) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 806) | def forward( class LinearMultiDim (line 1096) | class LinearMultiDim(nn.Linear): method __init__ (line 1097) | def __init__(self, in_features, out_features=None, second_dim=4, *args... method forward (line 1106) | def forward(self, input_tensor, *args, **kwargs): class ResnetBlockFlat (line 1115) | class ResnetBlockFlat(nn.Module): method __init__ (line 1116) | def __init__( method forward (line 1176) | def forward(self, input_tensor, temb): class DownBlockFlat (line 1210) | class DownBlockFlat(nn.Module): method __init__ (line 1211) | def __init__( method forward (line 1262) | def forward(self, hidden_states, temb=None): class CrossAttnDownBlockFlat (line 1297) | class CrossAttnDownBlockFlat(nn.Module): method __init__ (line 1298) | def __init__( method forward (line 1385) | def forward( class UpBlockFlat (line 1449) | class UpBlockFlat(nn.Module): method __init__ (line 1450) | def __init__( method forward (line 1497) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class CrossAttnUpBlockFlat (line 1531) | class CrossAttnUpBlockFlat(nn.Module): method __init__ (line 1532) | def __init__( method forward (line 1615) | def forward( class UNetMidBlockFlatCrossAttn (line 1680) | class UNetMidBlockFlatCrossAttn(nn.Module): method __init__ (line 1681) | def __init__( method forward (line 1766) | def forward( class UNetMidBlockFlatSimpleCrossAttn (line 1791) | class UNetMidBlockFlatSimpleCrossAttn(nn.Module): method __init__ (line 1792) | def __init__( method forward (line 1876) | def forward( FILE: 6DoF/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion.py class VersatileDiffusionPipeline (line 20) | class VersatileDiffusionPipeline(DiffusionPipeline): method __init__ (line 57) | def __init__( method image_variation (line 83) | def image_variation( method text_to_image (line 199) | def text_to_image( method dual_guided (line 311) | def dual_guided( FILE: 6DoF/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py class VersatileDiffusionDualGuidedPipeline (line 41) | class VersatileDiffusionDualGuidedPipeline(DiffusionPipeline): method __init__ (line 70) | def __init__( method remove_unused_weights (line 101) | def remove_unused_weights(self): method _convert_to_dual_attention (line 104) | def _convert_to_dual_attention(self): method _revert_dual_attention (line 138) | def _revert_dual_attention(self): method enable_sequential_cpu_offload (line 151) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 170) | def _execution_device(self): method _encode_text_prompt (line 187) | def _encode_text_prompt(self, prompt, device, num_images_per_prompt, d... method _encode_image_prompt (line 278) | def _encode_image_prompt(self, prompt, device, num_images_per_prompt, ... method decode_latents (line 334) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 348) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 365) | def check_inputs(self, prompt, image, height, width, callback_steps): method prepare_latents (line 383) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method set_transformer_params (line 400) | def set_transformer_params(self, mix_ratio: float = 0.5, condition_typ... method __call__ (line 414) | def __call__( FILE: 6DoF/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py class VersatileDiffusionImageVariationPipeline (line 35) | class VersatileDiffusionImageVariationPipeline(DiffusionPipeline): method __init__ (line 59) | def __init__( method enable_sequential_cpu_offload (line 78) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 97) | def _execution_device(self): method _encode_prompt (line 114) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method decode_latents (line 194) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 208) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 226) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 249) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 267) | def __call__( FILE: 6DoF/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py class VersatileDiffusionTextToImagePipeline (line 34) | class VersatileDiffusionTextToImagePipeline(DiffusionPipeline): method __init__ (line 62) | def __init__( method _swap_unet_attention_blocks (line 86) | def _swap_unet_attention_blocks(self): method remove_unused_weights (line 99) | def remove_unused_weights(self): method enable_sequential_cpu_offload (line 102) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 121) | def _execution_device(self): method _encode_prompt (line 138) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method decode_latents (line 251) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 265) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 283) | def check_inputs( method prepare_latents (line 331) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 349) | def __call__( FILE: 6DoF/diffusers/pipelines/vq_diffusion/pipeline_vq_diffusion.py class LearnedClassifierFreeSamplingEmbeddings (line 30) | class LearnedClassifierFreeSamplingEmbeddings(ModelMixin, ConfigMixin): method __init__ (line 36) | def __init__(self, learnable: bool, hidden_size: Optional[int] = None,... class VQDiffusionPipeline (line 52) | class VQDiffusionPipeline(DiffusionPipeline): method __init__ (line 83) | def __init__( method _encode_prompt (line 103) | def _encode_prompt(self, prompt, num_images_per_prompt, do_classifier_... method __call__ (line 167) | def __call__( method truncate (line 310) | def truncate(self, log_p_x_0: torch.FloatTensor, truncation_rate: floa... FILE: 6DoF/diffusers/schedulers/scheduling_consistency_models.py class CMStochasticIterativeSchedulerOutput (line 30) | class CMStochasticIterativeSchedulerOutput(BaseOutput): class CMStochasticIterativeScheduler (line 43) | class CMStochasticIterativeScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 82) | def __init__( method index_for_timestep (line 106) | def index_for_timestep(self, timestep, schedule_timesteps=None): method scale_model_input (line 113) | def scale_model_input( method sigma_to_t (line 136) | def sigma_to_t(self, sigmas: Union[float, np.ndarray]): method set_timesteps (line 152) | def set_timesteps( method _convert_to_karras (line 223) | def _convert_to_karras(self, ramp): method get_scalings (line 235) | def get_scalings(self, sigma): method get_scalings_for_boundary_condition (line 242) | def get_scalings_for_boundary_condition(self, sigma): method step (line 264) | def step( method add_noise (line 354) | def add_noise( method __len__ (line 379) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_ddim.py class DDIMSchedulerOutput (line 32) | class DDIMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar( function rescale_zero_terminal_snr (line 94) | def rescale_zero_terminal_snr(betas): class DDIMScheduler (line 130) | class DDIMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 190) | def __init__( method scale_model_input (line 243) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method _get_variance (line 257) | def _get_variance(self, timestep, prev_timestep): method _threshold_sample (line 268) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method set_timesteps (line 302) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 347) | def step( method add_noise (line 470) | def add_noise( method get_velocity (line 494) | def get_velocity( method __len__ (line 514) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_ddim_flax.py class DDIMSchedulerState (line 36) | class DDIMSchedulerState: method create (line 46) | def create( class FlaxDDIMSchedulerOutput (line 62) | class FlaxDDIMSchedulerOutput(FlaxSchedulerOutput): class FlaxDDIMScheduler (line 66) | class FlaxDDIMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 109) | def has_state(self): method __init__ (line 113) | def __init__( method create_state (line 127) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 151) | def scale_model_input( method set_timesteps (line 165) | def set_timesteps( method _get_variance (line 187) | def _get_variance(self, state: DDIMSchedulerState, timestep, prev_time... method step (line 199) | def step( method add_noise (line 286) | def add_noise( method get_velocity (line 295) | def get_velocity( method __len__ (line 304) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_ddim_inverse.py class DDIMSchedulerOutput (line 31) | class DDIMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 49) | def betas_for_alpha_bar( class DDIMInverseScheduler (line 93) | class DDIMInverseScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 134) | def __init__( method scale_model_input (line 187) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 201) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 225) | def step( method __len__ (line 282) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_ddim_parallel.py class DDIMParallelSchedulerOutput (line 32) | class DDIMParallelSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar( function rescale_zero_terminal_snr (line 95) | def rescale_zero_terminal_snr(betas): class DDIMParallelScheduler (line 131) | class DDIMParallelScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 193) | def __init__( method scale_model_input (line 247) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method _get_variance (line 261) | def _get_variance(self, timestep, prev_timestep=None): method _batch_get_variance (line 274) | def _batch_get_variance(self, t, prev_t): method _threshold_sample (line 286) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method set_timesteps (line 321) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 366) | def step( method batch_step_no_noise (line 488) | def batch_step_no_noise( method add_noise (line 597) | def add_noise( method get_velocity (line 621) | def get_velocity( method __len__ (line 641) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_ddpm.py class DDPMSchedulerOutput (line 30) | class DDPMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 47) | def betas_for_alpha_bar( class DDPMScheduler (line 91) | class DDPMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 145) | def __init__( method scale_model_input (line 195) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 209) | def set_timesteps( method _get_variance (line 284) | def _get_variance(self, t, predicted_variance=None, variance_type=None): method _threshold_sample (line 324) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method step (line 358) | def step( method add_noise (line 454) | def add_noise( method get_velocity (line 477) | def get_velocity( method __len__ (line 497) | def __len__(self): method previous_timestep (line 500) | def previous_timestep(self, timestep): FILE: 6DoF/diffusers/schedulers/scheduling_ddpm_flax.py class DDPMSchedulerState (line 36) | class DDPMSchedulerState: method create (line 45) | def create(cls, common: CommonSchedulerState, init_noise_sigma: jnp.nd... class FlaxDDPMSchedulerOutput (line 50) | class FlaxDDPMSchedulerOutput(FlaxSchedulerOutput): class FlaxDDPMScheduler (line 54) | class FlaxDDPMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 92) | def has_state(self): method __init__ (line 96) | def __init__( method create_state (line 110) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 125) | def scale_model_input( method set_timesteps (line 139) | def set_timesteps( method _get_variance (line 162) | def _get_variance(self, state: DDPMSchedulerState, t, predicted_varian... method step (line 195) | def step( method add_noise (line 280) | def add_noise( method get_velocity (line 289) | def get_velocity( method __len__ (line 298) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_ddpm_parallel.py class DDPMParallelSchedulerOutput (line 31) | class DDPMParallelSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 49) | def betas_for_alpha_bar( class DDPMParallelScheduler (line 93) | class DDPMParallelScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 149) | def __init__( method scale_model_input (line 200) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 215) | def set_timesteps( method _get_variance (line 291) | def _get_variance(self, t, predicted_variance=None, variance_type=None): method _threshold_sample (line 332) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method step (line 366) | def step( method batch_step_no_noise (line 462) | def batch_step_no_noise( method add_noise (line 543) | def add_noise( method get_velocity (line 567) | def get_velocity( method __len__ (line 587) | def __len__(self): method previous_timestep (line 591) | def previous_timestep(self, timestep): FILE: 6DoF/diffusers/schedulers/scheduling_deis_multistep.py function betas_for_alpha_bar (line 29) | def betas_for_alpha_bar( class DEISMultistepScheduler (line 73) | class DEISMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 138) | def __init__( method set_timesteps (line 202) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _threshold_sample (line 261) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method convert_model_output (line 295) | def convert_model_output( method deis_first_order_update (line 333) | def deis_first_order_update( method multistep_deis_second_order_update (line 363) | def multistep_deis_second_order_update( method multistep_deis_third_order_update (line 405) | def multistep_deis_third_order_update( method step (line 462) | def step( method scale_model_input (line 530) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 544) | def add_noise( method __len__ (line 567) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_dpmsolver_multistep.py function betas_for_alpha_bar (line 29) | def betas_for_alpha_bar( class DPMSolverMultistepScheduler (line 73) | class DPMSolverMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 165) | def __init__( method set_timesteps (line 231) | def set_timesteps(self, num_inference_steps: int = None, device: Union... method _threshold_sample (line 292) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method _sigma_to_t (line 327) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 351) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method convert_model_output (line 364) | def convert_model_output( method dpm_solver_first_order_update (line 439) | def dpm_solver_first_order_update( method multistep_dpm_solver_second_order_update (line 486) | def multistep_dpm_solver_second_order_update( method multistep_dpm_solver_third_order_update (line 578) | def multistep_dpm_solver_third_order_update( method step (line 633) | def step( method scale_model_input (line 711) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 725) | def add_noise( method __len__ (line 748) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py class DPMSolverMultistepSchedulerState (line 35) | class DPMSolverMultistepSchedulerState: method create (line 53) | def create( class FlaxDPMSolverMultistepSchedulerOutput (line 73) | class FlaxDPMSolverMultistepSchedulerOutput(FlaxSchedulerOutput): class FlaxDPMSolverMultistepScheduler (line 77) | class FlaxDPMSolverMultistepScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 147) | def has_state(self): method __init__ (line 151) | def __init__( method create_state (line 170) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method set_timesteps (line 199) | def set_timesteps( method convert_model_output (line 236) | def convert_model_output( method dpm_solver_first_order_update (line 306) | def dpm_solver_first_order_update( method multistep_dpm_solver_second_order_update (line 341) | def multistep_dpm_solver_second_order_update( method multistep_dpm_solver_third_order_update (line 401) | def multistep_dpm_solver_third_order_update( method step (line 457) | def step( method scale_model_input (line 594) | def scale_model_input( method add_noise (line 612) | def add_noise( method __len__ (line 621) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py function betas_for_alpha_bar (line 29) | def betas_for_alpha_bar( class DPMSolverMultistepInverseScheduler (line 73) | class DPMSolverMultistepInverseScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 147) | def __init__( method set_timesteps (line 212) | def set_timesteps(self, num_inference_steps: int = None, device: Union... method _threshold_sample (line 252) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method _sigma_to_t (line 287) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 311) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method convert_model_output (line 325) | def convert_model_output( method dpm_solver_first_order_update (line 401) | def dpm_solver_first_order_update( method multistep_dpm_solver_second_order_update (line 449) | def multistep_dpm_solver_second_order_update( method multistep_dpm_solver_third_order_update (line 542) | def multistep_dpm_solver_third_order_update( method step (line 597) | def step( method scale_model_input (line 678) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 692) | def add_noise( method __len__ (line 715) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_dpmsolver_sde.py class BatchedBrownianTree (line 27) | class BatchedBrownianTree: method __init__ (line 30) | def __init__(self, x, t0, t1, seed=None, **kwargs): method sort (line 45) | def sort(a, b): method __call__ (line 48) | def __call__(self, t0, t1): class BrownianTreeNoiseSampler (line 54) | class BrownianTreeNoiseSampler: method __init__ (line 69) | def __init__(self, x, sigma_min, sigma_max, seed=None, transform=lambd... method __call__ (line 74) | def __call__(self, sigma, sigma_next): function betas_for_alpha_bar (line 80) | def betas_for_alpha_bar( class DPMSolverSDEScheduler (line 124) | class DPMSolverSDEScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 165) | def __init__( method index_for_timestep (line 203) | def index_for_timestep(self, timestep, schedule_timesteps=None): method init_noise_sigma (line 222) | def init_noise_sigma(self): method scale_model_input (line 229) | def scale_model_input( method set_timesteps (line 249) | def set_timesteps( method _second_order_timesteps (line 321) | def _second_order_timesteps(self, sigmas, log_sigmas): method _sigma_to_t (line 337) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 361) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor) -> torch.Fl... method state_in_first_order (line 375) | def state_in_first_order(self): method step (line 378) | def step( method add_noise (line 483) | def add_noise( method __len__ (line 508) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_dpmsolver_singlestep.py function betas_for_alpha_bar (line 32) | def betas_for_alpha_bar( class DPMSolverSinglestepScheduler (line 76) | class DPMSolverSinglestepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 157) | def __init__( method get_order_list (line 221) | def get_order_list(self, num_inference_steps: int) -> List[int]: method set_timesteps (line 255) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _threshold_sample (line 298) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method _sigma_to_t (line 333) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 357) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method convert_model_output (line 370) | def convert_model_output( method dpm_solver_first_order_update (line 436) | def dpm_solver_first_order_update( method singlestep_dpm_solver_second_order_update (line 468) | def singlestep_dpm_solver_second_order_update( method singlestep_dpm_solver_third_order_update (line 529) | def singlestep_dpm_solver_third_order_update( method singlestep_dpm_solver_update (line 600) | def singlestep_dpm_solver_update( method step (line 637) | def step( method scale_model_input (line 699) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 713) | def add_noise( method __len__ (line 736) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_euler_ancestral_discrete.py class EulerAncestralDiscreteSchedulerOutput (line 32) | class EulerAncestralDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar( class EulerAncestralDiscreteScheduler (line 94) | class EulerAncestralDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 130) | def __init__( method init_noise_sigma (line 170) | def init_noise_sigma(self): method scale_model_input (line 177) | def scale_model_input( method set_timesteps (line 198) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 242) | def step( method add_noise (line 332) | def add_noise( method __len__ (line 357) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_euler_discrete.py class EulerDiscreteSchedulerOutput (line 32) | class EulerDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar( class EulerDiscreteScheduler (line 94) | class EulerDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 138) | def __init__( method init_noise_sigma (line 181) | def init_noise_sigma(self): method scale_model_input (line 188) | def scale_model_input( method set_timesteps (line 211) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _sigma_to_t (line 270) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 294) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method step (line 307) | def step( method add_noise (line 406) | def add_noise( method __len__ (line 431) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_heun_discrete.py function betas_for_alpha_bar (line 27) | def betas_for_alpha_bar( class HeunDiscreteScheduler (line 71) | class HeunDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 114) | def __init__( method index_for_timestep (line 152) | def index_for_timestep(self, timestep, schedule_timesteps=None): method init_noise_sigma (line 171) | def init_noise_sigma(self): method scale_model_input (line 178) | def scale_model_input( method set_timesteps (line 197) | def set_timesteps( method _sigma_to_t (line 266) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 290) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor, num_inferen... method state_in_first_order (line 304) | def state_in_first_order(self): method step (line 307) | def step( method add_noise (line 400) | def add_noise( method __len__ (line 425) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_ipndm.py class IPNDMScheduler (line 25) | class IPNDMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 44) | def __init__( method set_timesteps (line 61) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 85) | def step( method scale_model_input (line 135) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method _get_prev_sample (line 148) | def _get_prev_sample(self, sample, timestep_index, prev_timestep_index... method __len__ (line 160) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar( class KDPM2AncestralDiscreteScheduler (line 72) | class KDPM2AncestralDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 110) | def __init__( method index_for_timestep (line 143) | def index_for_timestep(self, timestep, schedule_timesteps=None): method init_noise_sigma (line 162) | def init_noise_sigma(self): method scale_model_input (line 169) | def scale_model_input( method set_timesteps (line 192) | def set_timesteps( method sigma_to_t (line 274) | def sigma_to_t(self, sigma): method state_in_first_order (line 298) | def state_in_first_order(self): method step (line 301) | def step( method add_noise (line 394) | def add_noise( method __len__ (line 419) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_k_dpm_2_discrete.py function betas_for_alpha_bar (line 27) | def betas_for_alpha_bar( class KDPM2DiscreteScheduler (line 71) | class KDPM2DiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 109) | def __init__( method index_for_timestep (line 142) | def index_for_timestep(self, timestep, schedule_timesteps=None): method init_noise_sigma (line 161) | def init_noise_sigma(self): method scale_model_input (line 168) | def scale_model_input( method set_timesteps (line 191) | def set_timesteps( method sigma_to_t (line 263) | def sigma_to_t(self, sigma): method state_in_first_order (line 287) | def state_in_first_order(self): method step (line 290) | def step( method add_noise (line 375) | def add_noise( method __len__ (line 400) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_karras_ve.py class KarrasVeOutput (line 28) | class KarrasVeOutput(BaseOutput): class KarrasVeScheduler (line 48) | class KarrasVeScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 83) | def __init__( method scale_model_input (line 100) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 114) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method add_noise_to_input (line 135) | def add_noise_to_input( method step (line 156) | def step( method step_correct (line 194) | def step_correct( method add_noise (line 231) | def add_noise(self, original_samples, noise, timesteps): FILE: 6DoF/diffusers/schedulers/scheduling_karras_ve_flax.py class KarrasVeSchedulerState (line 29) | class KarrasVeSchedulerState: method create (line 36) | def create(cls): class FlaxKarrasVeOutput (line 41) | class FlaxKarrasVeOutput(BaseOutput): class FlaxKarrasVeScheduler (line 59) | class FlaxKarrasVeScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 91) | def has_state(self): method __init__ (line 95) | def __init__( method create_state (line 106) | def create_state(self): method set_timesteps (line 109) | def set_timesteps( method add_noise_to_input (line 137) | def add_noise_to_input( method step (line 163) | def step( method step_correct (line 199) | def step_correct( method add_noise (line 236) | def add_noise(self, state: KarrasVeSchedulerState, original_samples, n... FILE: 6DoF/diffusers/schedulers/scheduling_lms_discrete.py class LMSDiscreteSchedulerOutput (line 30) | class LMSDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 48) | def betas_for_alpha_bar( class LMSDiscreteScheduler (line 92) | class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 133) | def __init__( method init_noise_sigma (line 175) | def init_noise_sigma(self): method scale_model_input (line 182) | def scale_model_input( method get_lms_coefficient (line 203) | def get_lms_coefficient(self, order, t, current_order): method set_timesteps (line 225) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _sigma_to_t (line 279) | def _sigma_to_t(self, sigma, log_sigmas): method _convert_to_karras (line 303) | def _convert_to_karras(self, in_sigmas: torch.FloatTensor) -> torch.Fl... method step (line 316) | def step( method add_noise (line 387) | def add_noise( method __len__ (line 412) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_lms_discrete_flax.py class LMSDiscreteSchedulerState (line 33) | class LMSDiscreteSchedulerState: method create (line 46) | def create( class FlaxLMSSchedulerOutput (line 53) | class FlaxLMSSchedulerOutput(FlaxSchedulerOutput): class FlaxLMSDiscreteScheduler (line 57) | class FlaxLMSDiscreteScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 90) | def has_state(self): method __init__ (line 94) | def __init__( method create_state (line 106) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 123) | def scale_model_input(self, state: LMSDiscreteSchedulerState, sample: ... method get_lms_coefficient (line 145) | def get_lms_coefficient(self, state: LMSDiscreteSchedulerState, order,... method set_timesteps (line 167) | def set_timesteps( method step (line 203) | def step( method add_noise (line 268) | def add_noise( method __len__ (line 282) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_pndm.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar( class PNDMScheduler (line 72) | class PNDMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 116) | def __init__( method set_timesteps (line 170) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 228) | def step( method step_prk (line 259) | def step_prk( method step_plms (line 314) | def step_plms( method scale_model_input (line 381) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method _get_prev_sample (line 394) | def _get_prev_sample(self, sample, timestep, prev_timestep, model_outp... method add_noise (line 438) | def add_noise( method __len__ (line 461) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_pndm_flax.py class PNDMSchedulerState (line 35) | class PNDMSchedulerState: method create (line 53) | def create( class FlaxPNDMSchedulerOutput (line 69) | class FlaxPNDMSchedulerOutput(FlaxSchedulerOutput): class FlaxPNDMScheduler (line 73) | class FlaxPNDMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 119) | def has_state(self): method __init__ (line 123) | def __init__( method create_state (line 143) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method set_timesteps (line 167) | def set_timesteps(self, state: PNDMSchedulerState, num_inference_steps... method scale_model_input (line 222) | def scale_model_input( method step (line 239) | def step( method step_prk (line 294) | def step_prk( method step_plms (line 362) | def step_plms( method _get_prev_sample (line 456) | def _get_prev_sample(self, state: PNDMSchedulerState, sample, timestep... method add_noise (line 501) | def add_noise( method __len__ (line 510) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_repaint.py class RePaintSchedulerOutput (line 28) | class RePaintSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 46) | def betas_for_alpha_bar( class RePaintScheduler (line 90) | class RePaintScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 124) | def __init__( method scale_model_input (line 168) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 182) | def set_timesteps( method _get_variance (line 212) | def _get_variance(self, t): method step (line 231) | def step( method undo_step (line 318) | def undo_step(self, sample, timestep, generator=None): method add_noise (line 335) | def add_noise( method __len__ (line 343) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_sde_ve.py class SdeVeOutput (line 29) | class SdeVeOutput(BaseOutput): class ScoreSdeVeScheduler (line 45) | class ScoreSdeVeScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 72) | def __init__( method scale_model_input (line 89) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 103) | def set_timesteps( method set_sigmas (line 120) | def set_sigmas( method get_adjacent_sigma (line 149) | def get_adjacent_sigma(self, timesteps, t): method step_pred (line 156) | def step_pred( method step_correct (line 219) | def step_correct( method add_noise (line 270) | def add_noise( method __len__ (line 287) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_sde_ve_flax.py class ScoreSdeVeSchedulerState (line 29) | class ScoreSdeVeSchedulerState: method create (line 36) | def create(cls): class FlaxSdeVeOutput (line 41) | class FlaxSdeVeOutput(FlaxSchedulerOutput): class FlaxScoreSdeVeScheduler (line 59) | class FlaxScoreSdeVeScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 84) | def has_state(self): method __init__ (line 88) | def __init__( method create_state (line 99) | def create_state(self): method set_timesteps (line 109) | def set_timesteps( method set_sigmas (line 128) | def set_sigmas( method get_adjacent_sigma (line 163) | def get_adjacent_sigma(self, state, timesteps, t): method step_pred (line 166) | def step_pred( method step_correct (line 226) | def step_correct( method __len__ (line 278) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_sde_vp.py class ScoreSdeVpScheduler (line 27) | class ScoreSdeVpScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 45) | def __init__(self, num_train_timesteps=2000, beta_min=0.1, beta_max=20... method set_timesteps (line 50) | def set_timesteps(self, num_inference_steps, device: Union[str, torch.... method step_pred (line 53) | def step_pred(self, score, x, t, generator=None): method __len__ (line 89) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_unclip.py class UnCLIPSchedulerOutput (line 29) | class UnCLIPSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 47) | def betas_for_alpha_bar( class UnCLIPScheduler (line 91) | class UnCLIPScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 121) | def __init__( method scale_model_input (line 148) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 162) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _get_variance (line 179) | def _get_variance(self, t, prev_timestep=None, predicted_variance=None... method step (line 215) | def step( method add_noise (line 327) | def add_noise( FILE: 6DoF/diffusers/schedulers/scheduling_unipc_multistep.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class UniPCMultistepScheduler (line 57) | class UniPCMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 137) | def __init__( method set_timesteps (line 201) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _threshold_sample (line 263) | def _threshold_sample(self, sample: torch.FloatTensor) -> torch.FloatT... method convert_model_output (line 297) | def convert_model_output( method multistep_uni_p_bh_update (line 348) | def multistep_uni_p_bh_update( method multistep_uni_c_bh_update (line 453) | def multistep_uni_c_bh_update( method step (line 559) | def step( method scale_model_input (line 643) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 657) | def add_noise( method __len__ (line 680) | def __len__(self): FILE: 6DoF/diffusers/schedulers/scheduling_utils.py class KarrasDiffusionSchedulers (line 32) | class KarrasDiffusionSchedulers(Enum): class SchedulerOutput (line 50) | class SchedulerOutput(BaseOutput): class SchedulerMixin (line 63) | class SchedulerMixin: method from_pretrained (line 78) | def from_pretrained( method save_pretrained (line 149) | def save_pretrained(self, save_directory: Union[str, os.PathLike], pus... method compatibles (line 161) | def compatibles(self): method _get_compatibles (line 171) | def _get_compatibles(cls): FILE: 6DoF/diffusers/schedulers/scheduling_utils_flax.py class FlaxKarrasDiffusionSchedulers (line 34) | class FlaxKarrasDiffusionSchedulers(Enum): class FlaxSchedulerOutput (line 43) | class FlaxSchedulerOutput(BaseOutput): class FlaxSchedulerMixin (line 56) | class FlaxSchedulerMixin: method from_pretrained (line 72) | def from_pretrained( method save_pretrained (line 151) | def save_pretrained(self, save_directory: Union[str, os.PathLike], pus... method compatibles (line 163) | def compatibles(self): method _get_compatibles (line 173) | def _get_compatibles(cls): function broadcast_to_shape_from_left (line 182) | def broadcast_to_shape_from_left(x: jnp.ndarray, shape: Tuple[int]) -> j... function betas_for_alpha_bar (line 187) | def betas_for_alpha_bar(num_diffusion_timesteps: int, max_beta=0.999, dt... class CommonSchedulerState (line 217) | class CommonSchedulerState: method create (line 223) | def create(cls, scheduler): function get_sqrt_alpha_prod (line 257) | def get_sqrt_alpha_prod( function add_noise_common (line 273) | def add_noise_common( function get_velocity_common (line 281) | def get_velocity_common(state: CommonSchedulerState, sample: jnp.ndarray... FILE: 6DoF/diffusers/schedulers/scheduling_vq_diffusion.py class VQDiffusionSchedulerOutput (line 28) | class VQDiffusionSchedulerOutput(BaseOutput): function index_to_log_onehot (line 41) | def index_to_log_onehot(x: torch.LongTensor, num_classes: int) -> torch.... function gumbel_noised (line 62) | def gumbel_noised(logits: torch.FloatTensor, generator: Optional[torch.G... function alpha_schedules (line 72) | def alpha_schedules(num_diffusion_timesteps: int, alpha_cum_start=0.9999... function gamma_schedules (line 88) | def gamma_schedules(num_diffusion_timesteps: int, gamma_cum_start=0.0000... class VQDiffusionScheduler (line 106) | class VQDiffusionScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 144) | def __init__( method set_timesteps (line 190) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 212) | def step( method q_posterior (line 260) | def q_posterior(self, log_p_x_0, x_t, t): method log_Q_t_transitioning_to_known_class (line 379) | def log_Q_t_transitioning_to_known_class( method apply_cumulative_transitions (line 484) | def apply_cumulative_transitions(self, q, t): FILE: 6DoF/diffusers/training_utils.py function set_seed (line 16) | def set_seed(seed: int): class EMAModel (line 30) | class EMAModel: method __init__ (line 35) | def __init__( method from_pretrained (line 117) | def from_pretrained(cls, path, model_cls) -> "EMAModel": method save_pretrained (line 126) | def save_pretrained(self, path): method get_decay (line 141) | def get_decay(self, optimization_step: int) -> float: method step (line 161) | def step(self, parameters: Iterable[torch.nn.Parameter]): method copy_to (line 198) | def copy_to(self, parameters: Iterable[torch.nn.Parameter]) -> None: method to (line 211) | def to(self, device=None, dtype=None) -> None: method state_dict (line 223) | def state_dict(self) -> dict: method store (line 242) | def store(self, parameters: Iterable[torch.nn.Parameter]) -> None: method restore (line 251) | def restore(self, parameters: Iterable[torch.nn.Parameter]) -> None: method load_state_dict (line 269) | def load_state_dict(self, state_dict: dict) -> None: FILE: 6DoF/diffusers/utils/__init__.py function check_min_version (line 113) | def check_min_version(min_version): FILE: 6DoF/diffusers/utils/accelerate_utils.py function apply_forward_hook (line 27) | def apply_forward_hook(method): FILE: 6DoF/diffusers/utils/deprecation_utils.py function deprecate (line 8) | def deprecate(*args, take_from: Optional[Union[Dict, Any]] = None, stand... FILE: 6DoF/diffusers/utils/doc_utils.py function replace_example_docstring (line 20) | def replace_example_docstring(example_docstring): FILE: 6DoF/diffusers/utils/dummy_flax_and_transformers_objects.py class FlaxStableDiffusionControlNetPipeline (line 5) | class FlaxStableDiffusionControlNetPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class FlaxStableDiffusionImg2ImgPipeline (line 20) | class FlaxStableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class FlaxStableDiffusionInpaintPipeline (line 35) | class FlaxStableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class FlaxStableDiffusionPipeline (line 50) | class FlaxStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_flax_objects.py class FlaxControlNetModel (line 5) | class FlaxControlNetModel(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class FlaxModelMixin (line 20) | class FlaxModelMixin(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class FlaxUNet2DConditionModel (line 35) | class FlaxUNet2DConditionModel(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class FlaxAutoencoderKL (line 50) | class FlaxAutoencoderKL(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class FlaxDiffusionPipeline (line 65) | class FlaxDiffusionPipeline(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class FlaxDDIMScheduler (line 80) | class FlaxDDIMScheduler(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class FlaxDDPMScheduler (line 95) | class FlaxDDPMScheduler(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class FlaxDPMSolverMultistepScheduler (line 110) | class FlaxDPMSolverMultistepScheduler(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class FlaxKarrasVeScheduler (line 125) | class FlaxKarrasVeScheduler(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): class FlaxLMSDiscreteScheduler (line 140) | class FlaxLMSDiscreteScheduler(metaclass=DummyObject): method __init__ (line 143) | def __init__(self, *args, **kwargs): method from_config (line 147) | def from_config(cls, *args, **kwargs): method from_pretrained (line 151) | def from_pretrained(cls, *args, **kwargs): class FlaxPNDMScheduler (line 155) | class FlaxPNDMScheduler(metaclass=DummyObject): method __init__ (line 158) | def __init__(self, *args, **kwargs): method from_config (line 162) | def from_config(cls, *args, **kwargs): method from_pretrained (line 166) | def from_pretrained(cls, *args, **kwargs): class FlaxSchedulerMixin (line 170) | class FlaxSchedulerMixin(metaclass=DummyObject): method __init__ (line 173) | def __init__(self, *args, **kwargs): method from_config (line 177) | def from_config(cls, *args, **kwargs): method from_pretrained (line 181) | def from_pretrained(cls, *args, **kwargs): class FlaxScoreSdeVeScheduler (line 185) | class FlaxScoreSdeVeScheduler(metaclass=DummyObject): method __init__ (line 188) | def __init__(self, *args, **kwargs): method from_config (line 192) | def from_config(cls, *args, **kwargs): method from_pretrained (line 196) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_note_seq_objects.py class MidiProcessor (line 5) | class MidiProcessor(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_onnx_objects.py class OnnxRuntimeModel (line 5) | class OnnxRuntimeModel(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_pt_objects.py class AutoencoderKL (line 5) | class AutoencoderKL(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class ControlNetModel (line 20) | class ControlNetModel(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class ModelMixin (line 35) | class ModelMixin(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class PriorTransformer (line 50) | class PriorTransformer(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class T5FilmDecoder (line 65) | class T5FilmDecoder(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class Transformer2DModel (line 80) | class Transformer2DModel(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class UNet1DModel (line 95) | class UNet1DModel(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class UNet2DConditionModel (line 110) | class UNet2DConditionModel(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class UNet2DModel (line 125) | class UNet2DModel(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): class UNet3DConditionModel (line 140) | class UNet3DConditionModel(metaclass=DummyObject): method __init__ (line 143) | def __init__(self, *args, **kwargs): method from_config (line 147) | def from_config(cls, *args, **kwargs): method from_pretrained (line 151) | def from_pretrained(cls, *args, **kwargs): class VQModel (line 155) | class VQModel(metaclass=DummyObject): method __init__ (line 158) | def __init__(self, *args, **kwargs): method from_config (line 162) | def from_config(cls, *args, **kwargs): method from_pretrained (line 166) | def from_pretrained(cls, *args, **kwargs): function get_constant_schedule (line 170) | def get_constant_schedule(*args, **kwargs): function get_constant_schedule_with_warmup (line 174) | def get_constant_schedule_with_warmup(*args, **kwargs): function get_cosine_schedule_with_warmup (line 178) | def get_cosine_schedule_with_warmup(*args, **kwargs): function get_cosine_with_hard_restarts_schedule_with_warmup (line 182) | def get_cosine_with_hard_restarts_schedule_with_warmup(*args, **kwargs): function get_linear_schedule_with_warmup (line 186) | def get_linear_schedule_with_warmup(*args, **kwargs): function get_polynomial_decay_schedule_with_warmup (line 190) | def get_polynomial_decay_schedule_with_warmup(*args, **kwargs): function get_scheduler (line 194) | def get_scheduler(*args, **kwargs): class AudioPipelineOutput (line 198) | class AudioPipelineOutput(metaclass=DummyObject): method __init__ (line 201) | def __init__(self, *args, **kwargs): method from_config (line 205) | def from_config(cls, *args, **kwargs): method from_pretrained (line 209) | def from_pretrained(cls, *args, **kwargs): class ConsistencyModelPipeline (line 213) | class ConsistencyModelPipeline(metaclass=DummyObject): method __init__ (line 216) | def __init__(self, *args, **kwargs): method from_config (line 220) | def from_config(cls, *args, **kwargs): method from_pretrained (line 224) | def from_pretrained(cls, *args, **kwargs): class DanceDiffusionPipeline (line 228) | class DanceDiffusionPipeline(metaclass=DummyObject): method __init__ (line 231) | def __init__(self, *args, **kwargs): method from_config (line 235) | def from_config(cls, *args, **kwargs): method from_pretrained (line 239) | def from_pretrained(cls, *args, **kwargs): class DDIMPipeline (line 243) | class DDIMPipeline(metaclass=DummyObject): method __init__ (line 246) | def __init__(self, *args, **kwargs): method from_config (line 250) | def from_config(cls, *args, **kwargs): method from_pretrained (line 254) | def from_pretrained(cls, *args, **kwargs): class DDPMPipeline (line 258) | class DDPMPipeline(metaclass=DummyObject): method __init__ (line 261) | def __init__(self, *args, **kwargs): method from_config (line 265) | def from_config(cls, *args, **kwargs): method from_pretrained (line 269) | def from_pretrained(cls, *args, **kwargs): class DiffusionPipeline (line 273) | class DiffusionPipeline(metaclass=DummyObject): method __init__ (line 276) | def __init__(self, *args, **kwargs): method from_config (line 280) | def from_config(cls, *args, **kwargs): method from_pretrained (line 284) | def from_pretrained(cls, *args, **kwargs): class DiTPipeline (line 288) | class DiTPipeline(metaclass=DummyObject): method __init__ (line 291) | def __init__(self, *args, **kwargs): method from_config (line 295) | def from_config(cls, *args, **kwargs): method from_pretrained (line 299) | def from_pretrained(cls, *args, **kwargs): class ImagePipelineOutput (line 303) | class ImagePipelineOutput(metaclass=DummyObject): method __init__ (line 306) | def __init__(self, *args, **kwargs): method from_config (line 310) | def from_config(cls, *args, **kwargs): method from_pretrained (line 314) | def from_pretrained(cls, *args, **kwargs): class KarrasVePipeline (line 318) | class KarrasVePipeline(metaclass=DummyObject): method __init__ (line 321) | def __init__(self, *args, **kwargs): method from_config (line 325) | def from_config(cls, *args, **kwargs): method from_pretrained (line 329) | def from_pretrained(cls, *args, **kwargs): class LDMPipeline (line 333) | class LDMPipeline(metaclass=DummyObject): method __init__ (line 336) | def __init__(self, *args, **kwargs): method from_config (line 340) | def from_config(cls, *args, **kwargs): method from_pretrained (line 344) | def from_pretrained(cls, *args, **kwargs): class LDMSuperResolutionPipeline (line 348) | class LDMSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 351) | def __init__(self, *args, **kwargs): method from_config (line 355) | def from_config(cls, *args, **kwargs): method from_pretrained (line 359) | def from_pretrained(cls, *args, **kwargs): class PNDMPipeline (line 363) | class PNDMPipeline(metaclass=DummyObject): method __init__ (line 366) | def __init__(self, *args, **kwargs): method from_config (line 370) | def from_config(cls, *args, **kwargs): method from_pretrained (line 374) | def from_pretrained(cls, *args, **kwargs): class RePaintPipeline (line 378) | class RePaintPipeline(metaclass=DummyObject): method __init__ (line 381) | def __init__(self, *args, **kwargs): method from_config (line 385) | def from_config(cls, *args, **kwargs): method from_pretrained (line 389) | def from_pretrained(cls, *args, **kwargs): class ScoreSdeVePipeline (line 393) | class ScoreSdeVePipeline(metaclass=DummyObject): method __init__ (line 396) | def __init__(self, *args, **kwargs): method from_config (line 400) | def from_config(cls, *args, **kwargs): method from_pretrained (line 404) | def from_pretrained(cls, *args, **kwargs): class CMStochasticIterativeScheduler (line 408) | class CMStochasticIterativeScheduler(metaclass=DummyObject): method __init__ (line 411) | def __init__(self, *args, **kwargs): method from_config (line 415) | def from_config(cls, *args, **kwargs): method from_pretrained (line 419) | def from_pretrained(cls, *args, **kwargs): class DDIMInverseScheduler (line 423) | class DDIMInverseScheduler(metaclass=DummyObject): method __init__ (line 426) | def __init__(self, *args, **kwargs): method from_config (line 430) | def from_config(cls, *args, **kwargs): method from_pretrained (line 434) | def from_pretrained(cls, *args, **kwargs): class DDIMParallelScheduler (line 438) | class DDIMParallelScheduler(metaclass=DummyObject): method __init__ (line 441) | def __init__(self, *args, **kwargs): method from_config (line 445) | def from_config(cls, *args, **kwargs): method from_pretrained (line 449) | def from_pretrained(cls, *args, **kwargs): class DDIMScheduler (line 453) | class DDIMScheduler(metaclass=DummyObject): method __init__ (line 456) | def __init__(self, *args, **kwargs): method from_config (line 460) | def from_config(cls, *args, **kwargs): method from_pretrained (line 464) | def from_pretrained(cls, *args, **kwargs): class DDPMParallelScheduler (line 468) | class DDPMParallelScheduler(metaclass=DummyObject): method __init__ (line 471) | def __init__(self, *args, **kwargs): method from_config (line 475) | def from_config(cls, *args, **kwargs): method from_pretrained (line 479) | def from_pretrained(cls, *args, **kwargs): class DDPMScheduler (line 483) | class DDPMScheduler(metaclass=DummyObject): method __init__ (line 486) | def __init__(self, *args, **kwargs): method from_config (line 490) | def from_config(cls, *args, **kwargs): method from_pretrained (line 494) | def from_pretrained(cls, *args, **kwargs): class DEISMultistepScheduler (line 498) | class DEISMultistepScheduler(metaclass=DummyObject): method __init__ (line 501) | def __init__(self, *args, **kwargs): method from_config (line 505) | def from_config(cls, *args, **kwargs): method from_pretrained (line 509) | def from_pretrained(cls, *args, **kwargs): class DPMSolverMultistepInverseScheduler (line 513) | class DPMSolverMultistepInverseScheduler(metaclass=DummyObject): method __init__ (line 516) | def __init__(self, *args, **kwargs): method from_config (line 520) | def from_config(cls, *args, **kwargs): method from_pretrained (line 524) | def from_pretrained(cls, *args, **kwargs): class DPMSolverMultistepScheduler (line 528) | class DPMSolverMultistepScheduler(metaclass=DummyObject): method __init__ (line 531) | def __init__(self, *args, **kwargs): method from_config (line 535) | def from_config(cls, *args, **kwargs): method from_pretrained (line 539) | def from_pretrained(cls, *args, **kwargs): class DPMSolverSinglestepScheduler (line 543) | class DPMSolverSinglestepScheduler(metaclass=DummyObject): method __init__ (line 546) | def __init__(self, *args, **kwargs): method from_config (line 550) | def from_config(cls, *args, **kwargs): method from_pretrained (line 554) | def from_pretrained(cls, *args, **kwargs): class EulerAncestralDiscreteScheduler (line 558) | class EulerAncestralDiscreteScheduler(metaclass=DummyObject): method __init__ (line 561) | def __init__(self, *args, **kwargs): method from_config (line 565) | def from_config(cls, *args, **kwargs): method from_pretrained (line 569) | def from_pretrained(cls, *args, **kwargs): class EulerDiscreteScheduler (line 573) | class EulerDiscreteScheduler(metaclass=DummyObject): method __init__ (line 576) | def __init__(self, *args, **kwargs): method from_config (line 580) | def from_config(cls, *args, **kwargs): method from_pretrained (line 584) | def from_pretrained(cls, *args, **kwargs): class HeunDiscreteScheduler (line 588) | class HeunDiscreteScheduler(metaclass=DummyObject): method __init__ (line 591) | def __init__(self, *args, **kwargs): method from_config (line 595) | def from_config(cls, *args, **kwargs): method from_pretrained (line 599) | def from_pretrained(cls, *args, **kwargs): class IPNDMScheduler (line 603) | class IPNDMScheduler(metaclass=DummyObject): method __init__ (line 606) | def __init__(self, *args, **kwargs): method from_config (line 610) | def from_config(cls, *args, **kwargs): method from_pretrained (line 614) | def from_pretrained(cls, *args, **kwargs): class KarrasVeScheduler (line 618) | class KarrasVeScheduler(metaclass=DummyObject): method __init__ (line 621) | def __init__(self, *args, **kwargs): method from_config (line 625) | def from_config(cls, *args, **kwargs): method from_pretrained (line 629) | def from_pretrained(cls, *args, **kwargs): class KDPM2AncestralDiscreteScheduler (line 633) | class KDPM2AncestralDiscreteScheduler(metaclass=DummyObject): method __init__ (line 636) | def __init__(self, *args, **kwargs): method from_config (line 640) | def from_config(cls, *args, **kwargs): method from_pretrained (line 644) | def from_pretrained(cls, *args, **kwargs): class KDPM2DiscreteScheduler (line 648) | class KDPM2DiscreteScheduler(metaclass=DummyObject): method __init__ (line 651) | def __init__(self, *args, **kwargs): method from_config (line 655) | def from_config(cls, *args, **kwargs): method from_pretrained (line 659) | def from_pretrained(cls, *args, **kwargs): class PNDMScheduler (line 663) | class PNDMScheduler(metaclass=DummyObject): method __init__ (line 666) | def __init__(self, *args, **kwargs): method from_config (line 670) | def from_config(cls, *args, **kwargs): method from_pretrained (line 674) | def from_pretrained(cls, *args, **kwargs): class RePaintScheduler (line 678) | class RePaintScheduler(metaclass=DummyObject): method __init__ (line 681) | def __init__(self, *args, **kwargs): method from_config (line 685) | def from_config(cls, *args, **kwargs): method from_pretrained (line 689) | def from_pretrained(cls, *args, **kwargs): class SchedulerMixin (line 693) | class SchedulerMixin(metaclass=DummyObject): method __init__ (line 696) | def __init__(self, *args, **kwargs): method from_config (line 700) | def from_config(cls, *args, **kwargs): method from_pretrained (line 704) | def from_pretrained(cls, *args, **kwargs): class ScoreSdeVeScheduler (line 708) | class ScoreSdeVeScheduler(metaclass=DummyObject): method __init__ (line 711) | def __init__(self, *args, **kwargs): method from_config (line 715) | def from_config(cls, *args, **kwargs): method from_pretrained (line 719) | def from_pretrained(cls, *args, **kwargs): class UnCLIPScheduler (line 723) | class UnCLIPScheduler(metaclass=DummyObject): method __init__ (line 726) | def __init__(self, *args, **kwargs): method from_config (line 730) | def from_config(cls, *args, **kwargs): method from_pretrained (line 734) | def from_pretrained(cls, *args, **kwargs): class UniPCMultistepScheduler (line 738) | class UniPCMultistepScheduler(metaclass=DummyObject): method __init__ (line 741) | def __init__(self, *args, **kwargs): method from_config (line 745) | def from_config(cls, *args, **kwargs): method from_pretrained (line 749) | def from_pretrained(cls, *args, **kwargs): class VQDiffusionScheduler (line 753) | class VQDiffusionScheduler(metaclass=DummyObject): method __init__ (line 756) | def __init__(self, *args, **kwargs): method from_config (line 760) | def from_config(cls, *args, **kwargs): method from_pretrained (line 764) | def from_pretrained(cls, *args, **kwargs): class EMAModel (line 768) | class EMAModel(metaclass=DummyObject): method __init__ (line 771) | def __init__(self, *args, **kwargs): method from_config (line 775) | def from_config(cls, *args, **kwargs): method from_pretrained (line 779) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_torch_and_librosa_objects.py class AudioDiffusionPipeline (line 5) | class AudioDiffusionPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class Mel (line 20) | class Mel(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_torch_and_scipy_objects.py class LMSDiscreteScheduler (line 5) | class LMSDiscreteScheduler(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_torch_and_torchsde_objects.py class DPMSolverSDEScheduler (line 5) | class DPMSolverSDEScheduler(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_torch_and_transformers_and_invisible_watermark_objects.py class StableDiffusionXLImg2ImgPipeline (line 5) | class StableDiffusionXLImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionXLPipeline (line 20) | class StableDiffusionXLPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_torch_and_transformers_and_k_diffusion_objects.py class StableDiffusionKDiffusionPipeline (line 5) | class StableDiffusionKDiffusionPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_torch_and_transformers_and_onnx_objects.py class OnnxStableDiffusionImg2ImgPipeline (line 5) | class OnnxStableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionInpaintPipeline (line 20) | class OnnxStableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionInpaintPipelineLegacy (line 35) | class OnnxStableDiffusionInpaintPipelineLegacy(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionPipeline (line 50) | class OnnxStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionUpscalePipeline (line 65) | class OnnxStableDiffusionUpscalePipeline(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionOnnxPipeline (line 80) | class StableDiffusionOnnxPipeline(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_torch_and_transformers_objects.py class AltDiffusionImg2ImgPipeline (line 5) | class AltDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class AltDiffusionPipeline (line 20) | class AltDiffusionPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class AudioLDMPipeline (line 35) | class AudioLDMPipeline(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class CycleDiffusionPipeline (line 50) | class CycleDiffusionPipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class IFImg2ImgPipeline (line 65) | class IFImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class IFImg2ImgSuperResolutionPipeline (line 80) | class IFImg2ImgSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class IFInpaintingPipeline (line 95) | class IFInpaintingPipeline(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class IFInpaintingSuperResolutionPipeline (line 110) | class IFInpaintingSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class IFPipeline (line 125) | class IFPipeline(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): class IFSuperResolutionPipeline (line 140) | class IFSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 143) | def __init__(self, *args, **kwargs): method from_config (line 147) | def from_config(cls, *args, **kwargs): method from_pretrained (line 151) | def from_pretrained(cls, *args, **kwargs): class ImageTextPipelineOutput (line 155) | class ImageTextPipelineOutput(metaclass=DummyObject): method __init__ (line 158) | def __init__(self, *args, **kwargs): method from_config (line 162) | def from_config(cls, *args, **kwargs): method from_pretrained (line 166) | def from_pretrained(cls, *args, **kwargs): class KandinskyImg2ImgPipeline (line 170) | class KandinskyImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 173) | def __init__(self, *args, **kwargs): method from_config (line 177) | def from_config(cls, *args, **kwargs): method from_pretrained (line 181) | def from_pretrained(cls, *args, **kwargs): class KandinskyInpaintPipeline (line 185) | class KandinskyInpaintPipeline(metaclass=DummyObject): method __init__ (line 188) | def __init__(self, *args, **kwargs): method from_config (line 192) | def from_config(cls, *args, **kwargs): method from_pretrained (line 196) | def from_pretrained(cls, *args, **kwargs): class KandinskyPipeline (line 200) | class KandinskyPipeline(metaclass=DummyObject): method __init__ (line 203) | def __init__(self, *args, **kwargs): method from_config (line 207) | def from_config(cls, *args, **kwargs): method from_pretrained (line 211) | def from_pretrained(cls, *args, **kwargs): class KandinskyPriorPipeline (line 215) | class KandinskyPriorPipeline(metaclass=DummyObject): method __init__ (line 218) | def __init__(self, *args, **kwargs): method from_config (line 222) | def from_config(cls, *args, **kwargs): method from_pretrained (line 226) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22ControlnetImg2ImgPipeline (line 230) | class KandinskyV22ControlnetImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 233) | def __init__(self, *args, **kwargs): method from_config (line 237) | def from_config(cls, *args, **kwargs): method from_pretrained (line 241) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22ControlnetPipeline (line 245) | class KandinskyV22ControlnetPipeline(metaclass=DummyObject): method __init__ (line 248) | def __init__(self, *args, **kwargs): method from_config (line 252) | def from_config(cls, *args, **kwargs): method from_pretrained (line 256) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22Img2ImgPipeline (line 260) | class KandinskyV22Img2ImgPipeline(metaclass=DummyObject): method __init__ (line 263) | def __init__(self, *args, **kwargs): method from_config (line 267) | def from_config(cls, *args, **kwargs): method from_pretrained (line 271) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22InpaintPipeline (line 275) | class KandinskyV22InpaintPipeline(metaclass=DummyObject): method __init__ (line 278) | def __init__(self, *args, **kwargs): method from_config (line 282) | def from_config(cls, *args, **kwargs): method from_pretrained (line 286) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22Pipeline (line 290) | class KandinskyV22Pipeline(metaclass=DummyObject): method __init__ (line 293) | def __init__(self, *args, **kwargs): method from_config (line 297) | def from_config(cls, *args, **kwargs): method from_pretrained (line 301) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22PriorEmb2EmbPipeline (line 305) | class KandinskyV22PriorEmb2EmbPipeline(metaclass=DummyObject): method __init__ (line 308) | def __init__(self, *args, **kwargs): method from_config (line 312) | def from_config(cls, *args, **kwargs): method from_pretrained (line 316) | def from_pretrained(cls, *args, **kwargs): class KandinskyV22PriorPipeline (line 320) | class KandinskyV22PriorPipeline(metaclass=DummyObject): method __init__ (line 323) | def __init__(self, *args, **kwargs): method from_config (line 327) | def from_config(cls, *args, **kwargs): method from_pretrained (line 331) | def from_pretrained(cls, *args, **kwargs): class LDMTextToImagePipeline (line 335) | class LDMTextToImagePipeline(metaclass=DummyObject): method __init__ (line 338) | def __init__(self, *args, **kwargs): method from_config (line 342) | def from_config(cls, *args, **kwargs): method from_pretrained (line 346) | def from_pretrained(cls, *args, **kwargs): class PaintByExamplePipeline (line 350) | class PaintByExamplePipeline(metaclass=DummyObject): method __init__ (line 353) | def __init__(self, *args, **kwargs): method from_config (line 357) | def from_config(cls, *args, **kwargs): method from_pretrained (line 361) | def from_pretrained(cls, *args, **kwargs): class SemanticStableDiffusionPipeline (line 365) | class SemanticStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 368) | def __init__(self, *args, **kwargs): method from_config (line 372) | def from_config(cls, *args, **kwargs): method from_pretrained (line 376) | def from_pretrained(cls, *args, **kwargs): class ShapEImg2ImgPipeline (line 380) | class ShapEImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 383) | def __init__(self, *args, **kwargs): method from_config (line 387) | def from_config(cls, *args, **kwargs): method from_pretrained (line 391) | def from_pretrained(cls, *args, **kwargs): class ShapEPipeline (line 395) | class ShapEPipeline(metaclass=DummyObject): method __init__ (line 398) | def __init__(self, *args, **kwargs): method from_config (line 402) | def from_config(cls, *args, **kwargs): method from_pretrained (line 406) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionAttendAndExcitePipeline (line 410) | class StableDiffusionAttendAndExcitePipeline(metaclass=DummyObject): method __init__ (line 413) | def __init__(self, *args, **kwargs): method from_config (line 417) | def from_config(cls, *args, **kwargs): method from_pretrained (line 421) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionControlNetImg2ImgPipeline (line 425) | class StableDiffusionControlNetImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 428) | def __init__(self, *args, **kwargs): method from_config (line 432) | def from_config(cls, *args, **kwargs): method from_pretrained (line 436) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionControlNetInpaintPipeline (line 440) | class StableDiffusionControlNetInpaintPipeline(metaclass=DummyObject): method __init__ (line 443) | def __init__(self, *args, **kwargs): method from_config (line 447) | def from_config(cls, *args, **kwargs): method from_pretrained (line 451) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionControlNetPipeline (line 455) | class StableDiffusionControlNetPipeline(metaclass=DummyObject): method __init__ (line 458) | def __init__(self, *args, **kwargs): method from_config (line 462) | def from_config(cls, *args, **kwargs): method from_pretrained (line 466) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionDepth2ImgPipeline (line 470) | class StableDiffusionDepth2ImgPipeline(metaclass=DummyObject): method __init__ (line 473) | def __init__(self, *args, **kwargs): method from_config (line 477) | def from_config(cls, *args, **kwargs): method from_pretrained (line 481) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionDiffEditPipeline (line 485) | class StableDiffusionDiffEditPipeline(metaclass=DummyObject): method __init__ (line 488) | def __init__(self, *args, **kwargs): method from_config (line 492) | def from_config(cls, *args, **kwargs): method from_pretrained (line 496) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionImageVariationPipeline (line 500) | class StableDiffusionImageVariationPipeline(metaclass=DummyObject): method __init__ (line 503) | def __init__(self, *args, **kwargs): method from_config (line 507) | def from_config(cls, *args, **kwargs): method from_pretrained (line 511) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionImg2ImgPipeline (line 515) | class StableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 518) | def __init__(self, *args, **kwargs): method from_config (line 522) | def from_config(cls, *args, **kwargs): method from_pretrained (line 526) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInpaintPipeline (line 530) | class StableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 533) | def __init__(self, *args, **kwargs): method from_config (line 537) | def from_config(cls, *args, **kwargs): method from_pretrained (line 541) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInpaintPipelineLegacy (line 545) | class StableDiffusionInpaintPipelineLegacy(metaclass=DummyObject): method __init__ (line 548) | def __init__(self, *args, **kwargs): method from_config (line 552) | def from_config(cls, *args, **kwargs): method from_pretrained (line 556) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInstructPix2PixPipeline (line 560) | class StableDiffusionInstructPix2PixPipeline(metaclass=DummyObject): method __init__ (line 563) | def __init__(self, *args, **kwargs): method from_config (line 567) | def from_config(cls, *args, **kwargs): method from_pretrained (line 571) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionLatentUpscalePipeline (line 575) | class StableDiffusionLatentUpscalePipeline(metaclass=DummyObject): method __init__ (line 578) | def __init__(self, *args, **kwargs): method from_config (line 582) | def from_config(cls, *args, **kwargs): method from_pretrained (line 586) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionLDM3DPipeline (line 590) | class StableDiffusionLDM3DPipeline(metaclass=DummyObject): method __init__ (line 593) | def __init__(self, *args, **kwargs): method from_config (line 597) | def from_config(cls, *args, **kwargs): method from_pretrained (line 601) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionModelEditingPipeline (line 605) | class StableDiffusionModelEditingPipeline(metaclass=DummyObject): method __init__ (line 608) | def __init__(self, *args, **kwargs): method from_config (line 612) | def from_config(cls, *args, **kwargs): method from_pretrained (line 616) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPanoramaPipeline (line 620) | class StableDiffusionPanoramaPipeline(metaclass=DummyObject): method __init__ (line 623) | def __init__(self, *args, **kwargs): method from_config (line 627) | def from_config(cls, *args, **kwargs): method from_pretrained (line 631) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionParadigmsPipeline (line 635) | class StableDiffusionParadigmsPipeline(metaclass=DummyObject): method __init__ (line 638) | def __init__(self, *args, **kwargs): method from_config (line 642) | def from_config(cls, *args, **kwargs): method from_pretrained (line 646) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPipeline (line 650) | class StableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 653) | def __init__(self, *args, **kwargs): method from_config (line 657) | def from_config(cls, *args, **kwargs): method from_pretrained (line 661) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPipelineSafe (line 665) | class StableDiffusionPipelineSafe(metaclass=DummyObject): method __init__ (line 668) | def __init__(self, *args, **kwargs): method from_config (line 672) | def from_config(cls, *args, **kwargs): method from_pretrained (line 676) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPix2PixZeroPipeline (line 680) | class StableDiffusionPix2PixZeroPipeline(metaclass=DummyObject): method __init__ (line 683) | def __init__(self, *args, **kwargs): method from_config (line 687) | def from_config(cls, *args, **kwargs): method from_pretrained (line 691) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionSAGPipeline (line 695) | class StableDiffusionSAGPipeline(metaclass=DummyObject): method __init__ (line 698) | def __init__(self, *args, **kwargs): method from_config (line 702) | def from_config(cls, *args, **kwargs): method from_pretrained (line 706) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionUpscalePipeline (line 710) | class StableDiffusionUpscalePipeline(metaclass=DummyObject): method __init__ (line 713) | def __init__(self, *args, **kwargs): method from_config (line 717) | def from_config(cls, *args, **kwargs): method from_pretrained (line 721) | def from_pretrained(cls, *args, **kwargs): class StableUnCLIPImg2ImgPipeline (line 725) | class StableUnCLIPImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 728) | def __init__(self, *args, **kwargs): method from_config (line 732) | def from_config(cls, *args, **kwargs): method from_pretrained (line 736) | def from_pretrained(cls, *args, **kwargs): class StableUnCLIPPipeline (line 740) | class StableUnCLIPPipeline(metaclass=DummyObject): method __init__ (line 743) | def __init__(self, *args, **kwargs): method from_config (line 747) | def from_config(cls, *args, **kwargs): method from_pretrained (line 751) | def from_pretrained(cls, *args, **kwargs): class TextToVideoSDPipeline (line 755) | class TextToVideoSDPipeline(metaclass=DummyObject): method __init__ (line 758) | def __init__(self, *args, **kwargs): method from_config (line 762) | def from_config(cls, *args, **kwargs): method from_pretrained (line 766) | def from_pretrained(cls, *args, **kwargs): class TextToVideoZeroPipeline (line 770) | class TextToVideoZeroPipeline(metaclass=DummyObject): method __init__ (line 773) | def __init__(self, *args, **kwargs): method from_config (line 777) | def from_config(cls, *args, **kwargs): method from_pretrained (line 781) | def from_pretrained(cls, *args, **kwargs): class UnCLIPImageVariationPipeline (line 785) | class UnCLIPImageVariationPipeline(metaclass=DummyObject): method __init__ (line 788) | def __init__(self, *args, **kwargs): method from_config (line 792) | def from_config(cls, *args, **kwargs): method from_pretrained (line 796) | def from_pretrained(cls, *args, **kwargs): class UnCLIPPipeline (line 800) | class UnCLIPPipeline(metaclass=DummyObject): method __init__ (line 803) | def __init__(self, *args, **kwargs): method from_config (line 807) | def from_config(cls, *args, **kwargs): method from_pretrained (line 811) | def from_pretrained(cls, *args, **kwargs): class UniDiffuserModel (line 815) | class UniDiffuserModel(metaclass=DummyObject): method __init__ (line 818) | def __init__(self, *args, **kwargs): method from_config (line 822) | def from_config(cls, *args, **kwargs): method from_pretrained (line 826) | def from_pretrained(cls, *args, **kwargs): class UniDiffuserPipeline (line 830) | class UniDiffuserPipeline(metaclass=DummyObject): method __init__ (line 833) | def __init__(self, *args, **kwargs): method from_config (line 837) | def from_config(cls, *args, **kwargs): method from_pretrained (line 841) | def from_pretrained(cls, *args, **kwargs): class UniDiffuserTextDecoder (line 845) | class UniDiffuserTextDecoder(metaclass=DummyObject): method __init__ (line 848) | def __init__(self, *args, **kwargs): method from_config (line 852) | def from_config(cls, *args, **kwargs): method from_pretrained (line 856) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionDualGuidedPipeline (line 860) | class VersatileDiffusionDualGuidedPipeline(metaclass=DummyObject): method __init__ (line 863) | def __init__(self, *args, **kwargs): method from_config (line 867) | def from_config(cls, *args, **kwargs): method from_pretrained (line 871) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionImageVariationPipeline (line 875) | class VersatileDiffusionImageVariationPipeline(metaclass=DummyObject): method __init__ (line 878) | def __init__(self, *args, **kwargs): method from_config (line 882) | def from_config(cls, *args, **kwargs): method from_pretrained (line 886) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionPipeline (line 890) | class VersatileDiffusionPipeline(metaclass=DummyObject): method __init__ (line 893) | def __init__(self, *args, **kwargs): method from_config (line 897) | def from_config(cls, *args, **kwargs): method from_pretrained (line 901) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionTextToImagePipeline (line 905) | class VersatileDiffusionTextToImagePipeline(metaclass=DummyObject): method __init__ (line 908) | def __init__(self, *args, **kwargs): method from_config (line 912) | def from_config(cls, *args, **kwargs): method from_pretrained (line 916) | def from_pretrained(cls, *args, **kwargs): class VideoToVideoSDPipeline (line 920) | class VideoToVideoSDPipeline(metaclass=DummyObject): method __init__ (line 923) | def __init__(self, *args, **kwargs): method from_config (line 927) | def from_config(cls, *args, **kwargs): method from_pretrained (line 931) | def from_pretrained(cls, *args, **kwargs): class VQDiffusionPipeline (line 935) | class VQDiffusionPipeline(metaclass=DummyObject): method __init__ (line 938) | def __init__(self, *args, **kwargs): method from_config (line 942) | def from_config(cls, *args, **kwargs): method from_pretrained (line 946) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dummy_transformers_and_torch_and_note_seq_objects.py class SpectrogramDiffusionPipeline (line 5) | class SpectrogramDiffusionPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: 6DoF/diffusers/utils/dynamic_modules_utils.py function get_diffusers_versions (line 43) | def get_diffusers_versions(): function init_hf_modules (line 49) | def init_hf_modules(): function create_dynamic_module (line 64) | def create_dynamic_module(name: Union[str, os.PathLike]): function get_relative_imports (line 79) | def get_relative_imports(module_file): function get_relative_import_files (line 97) | def get_relative_import_files(module_file): function check_imports (line 126) | def check_imports(filename): function get_class_in_module (line 158) | def get_class_in_module(class_name, module_path): function find_pipeline_class (line 170) | def find_pipeline_class(loaded_module): function get_cached_module_file (line 197) | def get_cached_module_file( function get_class_from_dynamic_module (line 369) | def get_class_from_dynamic_module( FILE: 6DoF/diffusers/utils/hub_utils.py function http_user_agent (line 68) | def http_user_agent(user_agent: Union[Dict, str, None] = None) -> str: function get_full_repo_name (line 92) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function create_model_card (line 102) | def create_model_card(args, model_name): function extract_commit_hash (line 151) | def extract_commit_hash(resolved_file: Optional[str], commit_hash: Optio... function move_cache (line 175) | def move_cache(old_cache_dir: Optional[str] = None, new_cache_dir: Optio... function _add_variant (line 237) | def _add_variant(weights_name: str, variant: Optional[str] = None) -> str: function _get_model_file (line 246) | def _get_model_file( FILE: 6DoF/diffusers/utils/import_utils.py function is_torch_available (line 305) | def is_torch_available(): function is_safetensors_available (line 309) | def is_safetensors_available(): function is_tf_available (line 313) | def is_tf_available(): function is_flax_available (line 317) | def is_flax_available(): function is_transformers_available (line 321) | def is_transformers_available(): function is_inflect_available (line 325) | def is_inflect_available(): function is_unidecode_available (line 329) | def is_unidecode_available(): function is_onnx_available (line 333) | def is_onnx_available(): function is_opencv_available (line 337) | def is_opencv_available(): function is_scipy_available (line 341) | def is_scipy_available(): function is_librosa_available (line 345) | def is_librosa_available(): function is_xformers_available (line 349) | def is_xformers_available(): function is_accelerate_available (line 353) | def is_accelerate_available(): function is_k_diffusion_available (line 357) | def is_k_diffusion_available(): function is_note_seq_available (line 361) | def is_note_seq_available(): function is_wandb_available (line 365) | def is_wandb_available(): function is_omegaconf_available (line 369) | def is_omegaconf_available(): function is_tensorboard_available (line 373) | def is_tensorboard_available(): function is_compel_available (line 377) | def is_compel_available(): function is_ftfy_available (line 381) | def is_ftfy_available(): function is_bs4_available (line 385) | def is_bs4_available(): function is_torchsde_available (line 389) | def is_torchsde_available(): function is_invisible_watermark_available (line 393) | def is_invisible_watermark_available(): function requires_backends (line 536) | def requires_backends(obj, backends): class DummyObject (line 567) | class DummyObject(type): method __getattr__ (line 573) | def __getattr__(cls, key): function compare_versions (line 580) | def compare_versions(library_or_version: Union[str, Version], operation:... function is_torch_version (line 600) | def is_torch_version(operation: str, version: str): function is_transformers_version (line 612) | def is_transformers_version(operation: str, version: str): function is_accelerate_version (line 626) | def is_accelerate_version(operation: str, version: str): function is_k_diffusion_version (line 640) | def is_k_diffusion_version(operation: str, version: str): class OptionalDependencyNotAvailable (line 654) | class OptionalDependencyNotAvailable(BaseException): FILE: 6DoF/diffusers/utils/logging.py function _get_default_logging_level (line 52) | def _get_default_logging_level(): function _get_library_name (line 69) | def _get_library_name() -> str: function _get_library_root_logger (line 73) | def _get_library_root_logger() -> logging.Logger: function _configure_library_root_logger (line 77) | def _configure_library_root_logger() -> None: function _reset_library_root_logger (line 94) | def _reset_library_root_logger() -> None: function get_log_levels_dict (line 107) | def get_log_levels_dict(): function get_logger (line 111) | def get_logger(name: Optional[str] = None) -> logging.Logger: function get_verbosity (line 125) | def get_verbosity() -> int: function set_verbosity (line 145) | def set_verbosity(verbosity: int) -> None: function set_verbosity_info (line 164) | def set_verbosity_info(): function set_verbosity_warning (line 169) | def set_verbosity_warning(): function set_verbosity_debug (line 174) | def set_verbosity_debug(): function set_verbosity_error (line 179) | def set_verbosity_error(): function disable_default_handler (line 184) | def disable_default_handler() -> None: function enable_default_handler (line 193) | def enable_default_handler() -> None: function add_handler (line 202) | def add_handler(handler: logging.Handler) -> None: function remove_handler (line 211) | def remove_handler(handler: logging.Handler) -> None: function disable_propagation (line 220) | def disable_propagation() -> None: function enable_propagation (line 229) | def enable_propagation() -> None: function enable_explicit_format (line 239) | def enable_explicit_format() -> None: function reset_format (line 254) | def reset_format() -> None: function warning_advice (line 266) | def warning_advice(self, *args, **kwargs): class EmptyTqdm (line 280) | class EmptyTqdm: method __init__ (line 283) | def __init__(self, *args, **kwargs): # pylint: disable=unused-argument method __iter__ (line 286) | def __iter__(self): method __getattr__ (line 289) | def __getattr__(self, _): method __enter__ (line 297) | def __enter__(self): method __exit__ (line 300) | def __exit__(self, type_, value, traceback): class _tqdm_cls (line 304) | class _tqdm_cls: method __call__ (line 305) | def __call__(self, *args, **kwargs): method set_lock (line 311) | def set_lock(self, *args, **kwargs): method get_lock (line 316) | def get_lock(self): function is_progress_bar_enabled (line 324) | def is_progress_bar_enabled() -> bool: function enable_progress_bar (line 330) | def enable_progress_bar(): function disable_progress_bar (line 336) | def disable_progress_bar(): FILE: 6DoF/diffusers/utils/outputs.py function is_tensor (line 27) | def is_tensor(x): class BaseOutput (line 40) | class BaseOutput(OrderedDict): method __post_init__ (line 54) | def __post_init__(self): method __delitem__ (line 73) | def __delitem__(self, *args, **kwargs): method setdefault (line 76) | def setdefault(self, *args, **kwargs): method pop (line 79) | def pop(self, *args, **kwargs): method update (line 82) | def update(self, *args, **kwargs): method __getitem__ (line 85) | def __getitem__(self, k): method __setattr__ (line 92) | def __setattr__(self, name, value): method __setitem__ (line 98) | def __setitem__(self, key, value): method to_tuple (line 104) | def to_tuple(self) -> Tuple[Any]: FILE: 6DoF/diffusers/utils/pil_utils.py function pt_to_pil (line 25) | def pt_to_pil(images): function numpy_to_pil (line 35) | def numpy_to_pil(images): FILE: 6DoF/diffusers/utils/testing_utils.py function torch_all_close (line 64) | def torch_all_close(a, b, *args, **kwargs): function print_tensor_test (line 72) | def print_tensor_test(tensor, filename="test_corrections.txt", expected_... function get_tests_dir (line 87) | def get_tests_dir(append_path=None): function parse_flag_from_env (line 108) | def parse_flag_from_env(key, default=False): function floats_tensor (line 128) | def floats_tensor(shape, scale=1.0, rng=None, name=None): function slow (line 144) | def slow(test_case): function nightly (line 154) | def nightly(test_case): function require_torch (line 164) | def require_torch(test_case): function require_torch_2 (line 171) | def require_torch_2(test_case): function require_torch_gpu (line 180) | def require_torch_gpu(test_case): function skip_mps (line 187) | def skip_mps(test_case): function require_flax (line 192) | def require_flax(test_case): function require_compel (line 199) | def require_compel(test_case): function require_onnxruntime (line 207) | def require_onnxruntime(test_case): function require_note_seq (line 214) | def require_note_seq(test_case): function require_torchsde (line 221) | def require_torchsde(test_case): function load_numpy (line 228) | def load_numpy(arry: Union[str, np.ndarray], local_path: Optional[str] =... function load_pt (line 255) | def load_pt(url: str): function load_image (line 262) | def load_image(image: Union[str, PIL.Image.Image]) -> PIL.Image.Image: function preprocess_image (line 293) | def preprocess_image(image: PIL.Image, batch_size: int): function export_to_gif (line 303) | def export_to_gif(image: List[PIL.Image.Image], output_gif_path: str = N... function export_to_video (line 318) | def export_to_video(video_frames: List[np.ndarray], output_video_path: s... function load_hf_numpy (line 335) | def load_hf_numpy(path) -> np.ndarray: function pytest_addoption_shared (line 350) | def pytest_addoption_shared(parser): function pytest_terminal_summary_main (line 369) | def pytest_terminal_summary_main(tr, id): function run_test_in_subprocess (line 499) | def run_test_in_subprocess(test_case, target_func, inputs=None, timeout=... class CaptureLogger (line 542) | class CaptureLogger: method __init__ (line 563) | def __init__(self, logger): method __enter__ (line 569) | def __enter__(self): method __exit__ (line 573) | def __exit__(self, *exc): method __repr__ (line 577) | def __repr__(self): function enable_full_determinism (line 581) | def enable_full_determinism(): function disable_full_determinism (line 599) | def disable_full_determinism(): FILE: 6DoF/diffusers/utils/torch_utils.py function maybe_allow_in_graph (line 32) | def maybe_allow_in_graph(cls): function randn_tensor (line 36) | def randn_tensor( function is_compiled_module (line 80) | def is_compiled_module(module): FILE: 6DoF/pipeline_zero1to3.py class CCProjection (line 67) | class CCProjection(ModelMixin, ConfigMixin): method __init__ (line 69) | def __init__(self, in_channel=772, out_channel=768): method forward (line 75) | def forward(self, x): class CLIPProjection (line 78) | class CLIPProjection(ModelMixin, ConfigMixin): method __init__ (line 80) | def __init__(self, in_channel, out_channel): method forward (line 87) | def forward(self, x): class CNLayernorm (line 91) | class CNLayernorm(ModelMixin, ConfigMixin): method __init__ (line 93) | def __init__(self, in_channel, eps): method forward (line 98) | def forward(self, x): class Zero1to3StableDiffusionPipeline (line 101) | class Zero1to3StableDiffusionPipeline(DiffusionPipeline): method __init__ (line 130) | def __init__( method enable_vae_slicing (line 227) | def enable_vae_slicing(self): method disable_vae_slicing (line 236) | def disable_vae_slicing(self): method enable_vae_tiling (line 243) | def enable_vae_tiling(self): method disable_vae_tiling (line 252) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 259) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 284) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 313) | def _execution_device(self): method _encode_prompt (line 330) | def _encode_prompt( method CLIP_preprocess (line 468) | def CLIP_preprocess(self, x): method _encode_image (line 486) | def _encode_image(self, image, device, num_images_per_prompt, do_class... method run_safety_checker (line 594) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 604) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 612) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 629) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 651) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_img_latents (line 668) | def prepare_img_latents(self, image, batch_size, dtype, device, genera... method __call__ (line 736) | def __call__( FILE: 6DoF/train_eschernet.py function image_grid (line 72) | def image_grid(imgs, rows, cols): function log_validation (line 83) | def log_validation(validation_dataloader, vae, image_encoder, feature_ex... function parse_args (line 210) | def parse_args(input_args=None): function _encode_image (line 494) | def _encode_image(feature_extractor, image_encoder, image, device, dtype... function main (line 507) | def main(args): FILE: 6DoF/unet_2d_condition.py class UNet2DConditionOutput (line 55) | class UNet2DConditionOutput(BaseOutput): class UNet2DConditionModel (line 67) | class UNet2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoade... method __init__ (line 158) | def __init__( method attn_processors (line 574) | def attn_processors(self) -> Dict[str, AttentionProcessor]: method set_attn_processor (line 597) | def set_attn_processor(self, processor: Union[AttentionProcessor, Dict... method set_default_attn_processor (line 631) | def set_default_attn_processor(self): method set_attention_slice (line 637) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 702) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 706) | def forward( FILE: CaPE.py class CaPE_6DoF (line 7) | class CaPE_6DoF: method cape_embed (line 8) | def cape_embed(self, f, P): method attn_with_CaPE (line 18) | def attn_with_CaPE(self, f1, f2, p1, p2): function euler_to_matrix (line 40) | def euler_to_matrix(alpha, beta, gamma, x, y, z): function random_6dof_pose (line 49) | def random_6dof_pose(B, T): class CaPE_4DoF (line 96) | class CaPE_4DoF: method rotate_every_two (line 97) | def rotate_every_two(self, x): method cape (line 103) | def cape(self, x, p): method cape_embed (line 109) | def cape_embed(self, qq, kk, p1, p2): method attn_with_CaPE (line 132) | def attn_with_CaPE(self, f1, f2, p1, p2): function random_4dof_pose (line 152) | def random_4dof_pose(B, T): function look_at (line 161) | def look_at(origin, target, up): function pose_4dof2matrix (line 173) | def pose_4dof2matrix(pose_4dof): function pose_matrix24dof (line 189) | def pose_matrix24dof(pose_matrix): FILE: eval_eschernet.py function parse_args (line 36) | def parse_args(input_args=None): function get_archimedean_spiral (line 141) | def get_archimedean_spiral(sphere_radius, num_steps=250): function look_at (line 169) | def look_at(origin, target, up): function main (line 180) | def main(args): FILE: metrics/eval_2D_NVS.py function calc_2D_metrics (line 15) | def calc_2D_metrics(pred_np, gt_np): FILE: metrics/eval_3D_GSO.py function cartesian_to_spherical (line 10) | def cartesian_to_spherical(xyz): function get_pose (line 19) | def get_pose(target_RT): function trimesh_to_open3d (line 25) | def trimesh_to_open3d(src): function normalize_mesh (line 35) | def normalize_mesh(vertices): function capture_screenshots (line 47) | def capture_screenshots(mesh_rec_o3d, cam_param, render_param, img_name): function vis_3D_rec (line 63) | def vis_3D_rec(GT_DIR, REC_DIR, method_name): FILE: metrics/metrics.py function chamfer (line 6) | def chamfer(gt_points, rec_points): function compute_iou (line 20) | def compute_iou(mesh_pr, mesh_gt): FILE: scripts/blender_script_mvs.py function sample_point_on_sphere (line 100) | def sample_point_on_sphere(radius: float) -> Tuple[float, float, float]: function sample_spherical (line 109) | def sample_spherical(radius=3.0, maxz=3.0, minz=0.): function sample_spherical (line 119) | def sample_spherical(radius_min=1.5, radius_max=2.0, maxz=1.6, minz=-0.75): function randomize_camera (line 130) | def randomize_camera(): function set_camera_location (line 137) | def set_camera_location(elevation, azimuth, distance): function set_camera_location_old (line 152) | def set_camera_location_old(elevation, azimuth, distance): function randomize_lighting (line 165) | def randomize_lighting() -> None: function reset_lighting (line 173) | def reset_lighting() -> None: function reset_scene (line 180) | def reset_scene() -> None: function load_object (line 198) | def load_object(object_path: str) -> None: function scene_bbox (line 210) | def scene_bbox(single_obj=None, ignore_matrix=False): function scene_root_objects (line 227) | def scene_root_objects(): function scene_meshes (line 233) | def scene_meshes(): function get_3x4_RT_matrix_from_blender (line 239) | def get_3x4_RT_matrix_from_blender(cam): function normalize_scene (line 272) | def normalize_scene(): function save_images (line 286) | def save_images(object_file: str) -> None: function download_object (line 342) | def download_object(object_url: str) -> str: FILE: scripts/objaverse_filter.py function filter_zero123_views (line 11) | def filter_zero123_views(path): function move_invalid_views (line 34) | def move_invalid_views(path, invalid_ids, invalid_path):