SYMBOL INDEX (183 symbols across 13 files)

FILE: python_coreml_stable_diffusion/activation_quantization.py
  function get_coreml_inputs (line 68) | def get_coreml_inputs(sample_inputs):
  function convert_to_coreml (line 77) | def convert_to_coreml(torchscript_module, sample_inputs):
  function unet_data_loader (line 92) | def unet_data_loader(data_dir, device='cpu', calibration_nsamples=None):
  function quantize_module_config (line 121) | def quantize_module_config(module_name):
  function quantize_cumulative_config (line 141) | def quantize_cumulative_config(skip_conv_layers, skip_einsum_layers):
  function quantize (line 173) | def quantize(model, config, calibration_data):
  function get_quantizable_modules (line 205) | def get_quantizable_modules(unet):
  function recipe_overrides_for_inference_speedup (line 217) | def recipe_overrides_for_inference_speedup(conv_layers, skipped_conv):
  function recipe_overrides_for_quality (line 231) | def recipe_overrides_for_quality(conv_layers, skipped_conv):
  function register_input_log_hook (line 244) | def register_input_log_hook(unet, inputs):
  function generate_calibration_data (line 258) | def generate_calibration_data(pipe, args, calibration_dir):
  function register_input_preprocessing_hook (line 282) | def register_input_preprocessing_hook(pipe):
  function prepare_pipe (line 300) | def prepare_pipe(pipe, unet):
  function run_pipe (line 310) | def run_pipe(pipe):
  function get_reference_pipeline (line 320) | def get_reference_pipeline(model_version):
  function main (line 342) | def main(args):

FILE: python_coreml_stable_diffusion/attention.py
  function softmax (line 11) | def softmax(x, dim):
  function split_einsum (line 24) | def split_einsum(q, k, v, mask, heads, dim_head):
  function split_einsum_v2 (line 77) | def split_einsum_v2(q, k, v, mask, heads, dim_head):
  function original (line 147) | def original(q, k, v, mask, heads, dim_head):

FILE: python_coreml_stable_diffusion/chunk_mlprogram.py
  function _verify_output_correctness_of_chunks (line 34) | def _verify_output_correctness_of_chunks(full_model,
  function _load_prog_from_mlmodel (line 82) | def _load_prog_from_mlmodel(model):
  function _get_op_idx_split_location (line 100) | def _get_op_idx_split_location(prog: Program):
  function _get_first_chunk_outputs (line 131) | def _get_first_chunk_outputs(block, op_idx):
  function _add_fp32_casts (line 150) | def _add_fp32_casts(block, boundary_vars):
  function _make_first_chunk_prog (line 161) | def _make_first_chunk_prog(prog, op_idx):
  function _make_second_chunk_prog (line 176) | def _make_second_chunk_prog(prog, op_idx):
  function _legacy_model_chunking (line 234) | def _legacy_model_chunking(args):
  function main (line 342) | def main(args):

FILE: python_coreml_stable_diffusion/controlnet.py
  class ControlNetConditioningEmbedding (line 15) | class ControlNetConditioningEmbedding(nn.Module):
    method __init__ (line 17) | def __init__(
    method forward (line 37) | def forward(self, conditioning):
  class ControlNetModel (line 49) | class ControlNetModel(ModelMixin, ConfigMixin):
    method __init__ (line 52) | def __init__(
    method get_num_residuals (line 191) | def get_num_residuals(self):
    method forward (line 199) | def forward(

FILE: python_coreml_stable_diffusion/coreml_model.py
  function _macos_version (line 23) | def _macos_version():
  class CoreMLModel (line 36) | class CoreMLModel:
    method __init__ (line 40) | def __init__(self, model_path, compute_unit, sources='packages', optim...
    method _verify_inputs (line 97) | def _verify_inputs(self, **kwargs):
    method __call__ (line 118) | def __call__(self, **kwargs):
  function get_resource_type (line 126) | def get_resource_type(resources_dir: str) -> str:
  function _load_mlpackage (line 155) | def _load_mlpackage(submodule_name,
  function _load_mlpackage_controlnet (line 206) | def _load_mlpackage_controlnet(mlpackages_dir, model_version, compute_un...
  function get_available_compute_units (line 224) | def get_available_compute_units():

FILE: python_coreml_stable_diffusion/layer_norm.py
  class LayerNormANE (line 11) | class LayerNormANE(nn.Module):
    method __init__ (line 18) | def __init__(self,
    method _reset_parameters (line 46) | def _reset_parameters(self):
    method forward (line 51) | def forward(self, inputs):

FILE: python_coreml_stable_diffusion/mixed_bit_compression_apply.py
  function main (line 23) | def main(args):

FILE: python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py
  function fake_linear_quantize (line 65) | def fake_linear_quantize(val, axis=-1, mode='LINEAR', dtype=types.int8):
  function fake_palettize (line 139) | def fake_palettize(module, nbits, in_ngroups=1, out_ngroups=1):
  function restore_weight (line 189) | def restore_weight(module, value):
  function get_palettizable_modules (line 194) | def get_palettizable_modules(unet, min_size=PALETTIZE_MIN_SIZE):
  function fake_int8_quantize (line 205) | def fake_int8_quantize(module):
  function fake_nbits_palette (line 216) | def fake_nbits_palette(module, nbits):
  function fake_palette_from_recipe (line 226) | def fake_palette_from_recipe(module, recipe):
  function run_pipe (line 248) | def run_pipe(pipe):
  function benchmark_signal_integrity (line 280) | def benchmark_signal_integrity(pipe,
  function descending_psnr_order (line 329) | def descending_psnr_order(results):
  function simulate_quant_fn (line 336) | def simulate_quant_fn(ref_pipe, quantization_to_simulate):
  function build_recipe (line 352) | def build_recipe(results, sizes, psnr_threshold, default_nbits):
  function plot (line 376) | def plot(results, args):
  function main (line 436) | def main(args):

FILE: python_coreml_stable_diffusion/multilingual_projection.py
  function main (line 15) | def main(args):

FILE: python_coreml_stable_diffusion/pipeline.py
  class CoreMLStableDiffusionPipeline (line 47) | class CoreMLStableDiffusionPipeline(DiffusionPipeline):
    method __init__ (line 52) | def __init__(
    method _encode_prompt (line 123) | def _encode_prompt(self,
    method run_controlnet (line 259) | def run_controlnet(self,
    method run_safety_checker (line 286) | def run_safety_checker(self, image):
    method decode_latents (line 313) | def decode_latents(self, latents):
    method prepare_latents (line 322) | def prepare_latents(self,
    method prepare_control_cond (line 346) | def prepare_control_cond(self,
    method check_inputs (line 359) | def check_inputs(self, prompt, height, width, callback_steps):
    method prepare_extra_step_kwargs (line 384) | def prepare_extra_step_kwargs(self, eta):
    method _get_add_time_ids (line 398) | def _get_add_time_ids(self, original_size, crops_coords_top_left, targ...
    method __call__ (line 403) | def __call__(
  function get_available_schedulers (line 592) | def get_available_schedulers():
  function get_coreml_pipe (line 607) | def get_coreml_pipe(pytorch_pipe,
  function get_image_path (line 700) | def get_image_path(args, **override_kwargs):
  function prepare_controlnet_cond (line 717) | def prepare_controlnet_cond(image_path, height, width):
  function main (line 724) | def main(args):

FILE: python_coreml_stable_diffusion/torch2coreml.py
  function _get_coreml_inputs (line 49) | def _get_coreml_inputs(sample_inputs, args):
  function compute_psnr (line 59) | def compute_psnr(a, b):
  function report_correctness (line 80) | def report_correctness(original_outputs, final_outputs, log_prefix):
  function _get_out_path (line 99) | def _get_out_path(args, submodule_name):
  function _convert_to_coreml (line 105) | def _convert_to_coreml(submodule_name, torchscript_module, sample_inputs,
  function _get_deployment_target (line 147) | def _get_deployment_target(target_string):
  function quantize_weights (line 182) | def quantize_weights(args):
  function _quantize_weights (line 206) | def _quantize_weights(out_path, model_name, nbits):
  function _compile_coreml_model (line 231) | def _compile_coreml_model(source_model_path, output_dir, final_name):
  function _download_t5_model (line 251) | def _download_t5_model(args, t5_save_path):
  function bundle_resources_for_swift_cli (line 271) | def bundle_resources_for_swift_cli(args):
  function patched_make_causal_mask (line 363) | def patched_make_causal_mask(input_ids_shape, dtype, device, past_key_va...
  function convert_text_encoder (line 379) | def convert_text_encoder(text_encoder, tokenizer, submodule_name, args):
  function modify_coremltools_torch_frontend_badbmm (line 500) | def modify_coremltools_torch_frontend_badbmm():
  function convert_vae_decoder (line 548) | def convert_vae_decoder(pipe, args):
  function convert_vae_decoder_sd3 (line 644) | def convert_vae_decoder_sd3(args):
  function convert_vae_encoder (line 700) | def convert_vae_encoder(pipe, args):
  function convert_unet (line 799) | def convert_unet(pipe, args, model_name=None):
  function convert_mmdit (line 1053) | def convert_mmdit(args):
  function convert_safety_checker (line 1119) | def convert_safety_checker(pipe, args):
  function _get_controlnet_base_model (line 1312) | def _get_controlnet_base_model(controlnet_model_version):
  function convert_controlnet (line 1317) | def convert_controlnet(pipe, args):
  function get_pipeline (line 1485) | def get_pipeline(args):
  function main (line 1520) | def main(args):
  function parser_spec (line 1603) | def parser_spec():

FILE: python_coreml_stable_diffusion/unet.py
  class AttentionImplementations (line 33) | class AttentionImplementations(Enum):
  class Einsum (line 45) | class Einsum(nn.Module):
    method __init__ (line 46) | def __init__(self, heads, dim_head):
    method forward (line 51) | def forward(self, q, k, v, mask):
  class CrossAttention (line 62) | class CrossAttention(nn.Module):
    method __init__ (line 65) | def __init__(self, query_dim, context_dim=None, heads=8, dim_head=64):
    method forward (line 87) | def forward(self, hidden_states, context=None, mask=None):
  function linear_to_conv2d_map (line 121) | def linear_to_conv2d_map(state_dict, prefix, local_metadata, strict,
  function correct_for_bias_scale_order_inversion (line 132) | def correct_for_bias_scale_order_inversion(state_dict, prefix, local_met...
  class LayerNormANE (line 141) | class LayerNormANE(LayerNormANE):
    method __init__ (line 143) | def __init__(self, *args, **kwargs):
  class CrossAttnUpBlock2D (line 151) | class CrossAttnUpBlock2D(nn.Module):
    method __init__ (line 153) | def __init__(
    method forward (line 207) | def forward(self,
  class UpBlock2D (line 228) | class UpBlock2D(nn.Module):
    method __init__ (line 230) | def __init__(
    method forward (line 266) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None):
  class CrossAttnDownBlock2D (line 282) | class CrossAttnDownBlock2D(nn.Module):
    method __init__ (line 284) | def __init__(
    method forward (line 336) | def forward(self, hidden_states, temb=None, encoder_hidden_states=None):
  class DownBlock2D (line 353) | class DownBlock2D(nn.Module):
    method __init__ (line 355) | def __init__(
    method forward (line 389) | def forward(self, hidden_states, temb=None):
  class ResnetBlock2D (line 406) | class ResnetBlock2D(nn.Module):
    method __init__ (line 408) | def __init__(
    method forward (line 470) | def forward(self, x, temb):
  class Upsample2D (line 492) | class Upsample2D(nn.Module):
    method __init__ (line 494) | def __init__(self, channels):
    method forward (line 498) | def forward(self, x):
  class Downsample2D (line 503) | class Downsample2D(nn.Module):
    method __init__ (line 505) | def __init__(self, channels):
    method forward (line 509) | def forward(self, x):
  class SpatialTransformer (line 513) | class SpatialTransformer(nn.Module):
    method __init__ (line 515) | def __init__(
    method forward (line 553) | def forward(self, hidden_states, context=None):
  class BasicTransformerBlock (line 566) | class BasicTransformerBlock(nn.Module):
    method __init__ (line 568) | def __init__(self, dim, n_heads, d_head, context_dim=None, gated_ff=Tr...
    method forward (line 586) | def forward(self, hidden_states, context=None):
  class FeedForward (line 594) | class FeedForward(nn.Module):
    method __init__ (line 596) | def __init__(self, dim, dim_out=None, mult=4, glu=False):
    method forward (line 605) | def forward(self, hidden_states):
  class GEGLU (line 609) | class GEGLU(nn.Module):
    method __init__ (line 611) | def __init__(self, dim_in, dim_out):
    method forward (line 615) | def forward(self, hidden_states):
  function get_activation (line 620) | def get_activation(act_fn):
  class TimestepEmbedding (line 630) | class TimestepEmbedding(nn.Module):
    method __init__ (line 631) | def __init__(
    method forward (line 665) | def forward(self, sample, condition=None):
  class Timesteps (line 685) | class Timesteps(nn.Module):
    method __init__ (line 687) | def __init__(self, num_channels, flip_sin_to_cos, downscale_freq_shift):
    method forward (line 693) | def forward(self, timesteps):
  function get_timestep_embedding (line 703) | def get_timestep_embedding(
  class UNetMidBlock2DCrossAttn (line 731) | class UNetMidBlock2DCrossAttn(nn.Module):
    method __init__ (line 733) | def __init__(
    method forward (line 789) | def forward(self, hidden_states, temb=None, encoder_hidden_states=None):
  class UNet2DConditionModel (line 798) | class UNet2DConditionModel(ModelMixin, ConfigMixin):
    method __init__ (line 801) | def __init__(
    method forward (line 975) | def forward(
  class UNet2DConditionModelXL (line 1051) | class UNet2DConditionModelXL(UNet2DConditionModel):
    method forward (line 1055) | def forward(
  function get_down_block (line 1155) | def get_down_block(
  function get_up_block (line 1201) | def get_up_block(
  function calculate_conv2d_output_shape (line 1247) | def calculate_conv2d_output_shape(in_h, in_w, conv2d_layer):

FILE: tests/test_stable_diffusion.py
  class TestStableDiffusionForTextToImage (line 40) | class TestStableDiffusionForTextToImage(unittest.TestCase):
    method setUpClass (line 54) | def setUpClass(cls):
    method tearDownClass (line 65) | def tearDownClass(cls):
    method test_torch_to_coreml_conversion (line 70) | def test_torch_to_coreml_conversion(self):
    method test_end_to_end_image_generation_speed (line 95) | def test_end_to_end_image_generation_speed(self):
    method test_image_to_prompt_clip_score (line 117) | def test_image_to_prompt_clip_score(self):
    method test_safety_checker_efficacy (line 157) | def test_safety_checker_efficacy(self):
    method test_swift_cli_image_generation (line 170) | def test_swift_cli_image_generation(self):
    method _init_coreml_pipe (line 218) | def _init_coreml_pipe(self, compute_unit):
    method _coreml_text_to_image_with_compute_unit (line 239) | def _coreml_text_to_image_with_compute_unit(self, compute_unit):
  function _reset_seed (line 289) | def _reset_seed():
  function _get_test_artifacts_dir (line 296) | def _get_test_artifacts_dir(args):
  function _extend_parser (line 306) | def _extend_parser(parser):