↓ 5 callersFunctiongeneric_param_init_fn_(module: nn.Module, init_fn_, n_layers: int, d_model: Optional[int]=None, init_div_is_residual: Union[int, flo
llava/model/language_model/mpt/param_init_fns.py:28
↓ 5 callersMethodget_dummy_inputs(self, device, seed=0, height=64, width=64, num_images=1)
diffusers/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py:271
↓ 5 callersMethodget_sd_image(self, seed=0, shape=(4, 3, 512, 512), fp16=False)
diffusers/tests/models/autoencoders/test_models_vae.py:806
↓ 5 callersFunctiononnx_export(
model,
model_args: tuple,
output_path: Path,
ordered_input_names,
output_names,
dyna
diffusers/scripts/convert_stable_diffusion_controlnet_to_onnx.py:187
↓ 5 callersFunctiononnx_export(
model,
model_args: tuple,
output_path: Path,
ordered_input_names,
output_names,
dyna
diffusers/scripts/convert_stable_diffusion_checkpoint_to_onnx.py:31
↓ 4 callersMethod__init__(
self,
channels,
num_res_blocks: int,
hidden_size,
hidden_dropout,
diffusers/src/diffusers/models/unets/uvit_2d.py:308
↓ 4 callersMethod_combiner""" Combines a latent iamge img_vae of shape (B, C, H, W) and a CLIP-embedded image img_clip of shape (B, 1, clip_img_dim) into a sin
diffusers/src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:790