id: 1
unit: def main()
file: training/train_muse.py
start line: 0
end line: 0
size: 543 LOC
McCabe index: 92
number of parameters: 0

id: 2
unit: def main()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 270 LOC
McCabe index: 46
number of parameters: 0

id: 3
unit: def main()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 189 LOC
McCabe index: 24
number of parameters: 0

id: 4
unit: def main()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 170 LOC
McCabe index: 19
number of parameters: 0

id: 5
unit: def from_pretrained()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 124 LOC
McCabe index: 19
number of parameters: 4

id: 6
unit: def rename_flax_dict()
file: scripts/convert_maskgit_vqgan.py
start line: 0
end line: 0
size: 99 LOC
McCabe index: 22
number of parameters: 1

id: 7
unit: def main()
file: scripts/gen_sdxl_synthetic_dataset.py
start line: 0
end line: 0
size: 87 LOC
McCabe index: 8
number of parameters: 0

id: 8
unit: def __call__()
file: training/data.py
start line: 0
end line: 0
size: 76 LOC
McCabe index: 46
number of parameters: 2

id: 9
unit: def generate_and_save_images_coco()
file: scripts/calculate_fid.py
start line: 0
end line: 0
size: 75 LOC
McCabe index: 7
number of parameters: 1

id: 10
unit: def chart()
file: benchmark/muse_chart.py
start line: 0
end line: 0
size: 65 LOC
McCabe index: 6
number of parameters: 4

id: 11
unit: def mask_or_random_replace_tokens()
file: training/train_muse.py
start line: 0
end line: 0
size: 57 LOC
McCabe index: 16
number of parameters: 5

id: 12
unit: def load_flax_weights_in_pytorch_model()
file: scripts/convert_maskgit_vqgan.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 15
number of parameters: 2

id: 13
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 51 LOC
McCabe index: 10
number of parameters: 2

id: 14
unit: def table()
file: benchmark/muse_table.py
start line: 0
end line: 0
size: 49 LOC
McCabe index: 3
number of parameters: 3

id: 15
unit: def generate_and_log()
file: scripts/log_inpainting_images.py
start line: 0
end line: 0
size: 45 LOC
McCabe index: 6
number of parameters: 1

id: 16
unit: def muse_benchmark()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 42 LOC
McCabe index: 2
number of parameters: 5

id: 17
unit: def generate_and_log()
file: scripts/log_generations_wandb.py
start line: 0
end line: 0
size: 38 LOC
McCabe index: 8
number of parameters: 1

id: 18
unit: def sd_benchmark()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 37 LOC
McCabe index: 2
number of parameters: 3

id: 19
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 35 LOC
McCabe index: 3
number of parameters: 3

id: 20
unit: def load_state_dict()
file: muse/training_utils.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 13
number of parameters: 2

id: 21
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 4
number of parameters: 2

id: 22
unit: def load_state_dict()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 31 LOC
McCabe index: 13
number of parameters: 2

id: 23
unit: def register_to_config()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 13
number of parameters: 1

id: 24
unit: def save_checkpoint()
file: training/train_muse.py
start line: 0
end line: 0
size: 30 LOC
McCabe index: 10
number of parameters: 4

id: 25
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 4
number of parameters: 2

id: 26
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 4
number of parameters: 3

id: 27
unit: def __init__()
file: training/data.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 1
number of parameters: 0

id: 28
unit: def offline_ema()
file: scripts/compute_offline_ema.py
start line: 0
end line: 0
size: 29 LOC
McCabe index: 9
number of parameters: 1

id: 29
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 6
number of parameters: 4

id: 30
unit: def __call__()
file: muse/pipeline_muse.py
start line: 0
end line: 0
size: 28 LOC
McCabe index: 2
number of parameters: 18

id: 31
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 3
number of parameters: 2

id: 32
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 3
number of parameters: 2

id: 33
unit: def generate_images()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 6
number of parameters: 4

id: 34
unit: def convert_imagenet_to_wds()
file: scripts/convert_imagenet_to_wds.py
start line: 0
end line: 0
size: 27 LOC
McCabe index: 5
number of parameters: 3

id: 35
unit: def wurst_benchmark()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 2
number of parameters: 2

id: 36
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 6
number of parameters: 5

id: 37
unit: def create_model_and_benchmark()
file: scripts/benchmark_models.py
start line: 0
end line: 0
size: 26 LOC
McCabe index: 3
number of parameters: 1

id: 38
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 8
number of parameters: 4

id: 39
unit: def flatten_omega_conf()
file: training/train_muse.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 11
number of parameters: 2

id: 40
unit: def flatten_omega_conf()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 25 LOC
McCabe index: 11
number of parameters: 2

id: 41
unit: def sdxl_benchmark()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 4
number of parameters: 4

id: 42
unit: def ssd_1b_benchmark()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 4
number of parameters: 4

id: 43
unit: def load_state_dict()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 5
number of parameters: 2

id: 44
unit: def sd_turbo_benchmark()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 2
number of parameters: 3

id: 45
unit: def sdxl_turbo_benchmark()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 2
number of parameters: 3

id: 46
unit: def generate_and_save_images_flickr_8k()
file: scripts/calculate_fid.py
start line: 0
end line: 0
size: 23 LOC
McCabe index: 3
number of parameters: 1

id: 47
unit: def generate2()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 2
number of parameters: 0

id: 48
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 3
number of parameters: 3

id: 49
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 5
number of parameters: 5

id: 50
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 5
number of parameters: 4

id: 51
unit: def lcm_benchmark()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 2
number of parameters: 3

id: 52
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 5
number of parameters: 3

id: 53
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 5
number of parameters: 3

id: 54
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 7
number of parameters: 6

id: 55
unit: def sdxl_synthetic_dataset_map()
file: training/data.py
start line: 0
end line: 0
size: 21 LOC
McCabe index: 8
number of parameters: 1

id: 56
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 4

id: 57
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 2

id: 58
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 4

id: 59
unit: def __call__()
file: muse/pipeline_muse.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 1
number of parameters: 20

id: 60
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 3
number of parameters: 6

id: 61
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 7

id: 62
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 9
number of parameters: 6

id: 63
unit: def step()
file: training/optimizer.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 6
number of parameters: 2

id: 64
unit: def main()
file: scripts/convert_coco_to_wds.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 5
number of parameters: 0

id: 65
unit: def token_probability_distributions_per_percent_masked_bucket()
file: muse/training_utils.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 5
number of parameters: 3

id: 66
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 0

id: 67
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 0

id: 68
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 6
number of parameters: 4

id: 69
unit: def __init__()
file: training/data.py
start line: 0
end line: 0
size: 19 LOC
McCabe index: 1
number of parameters: 0

id: 70
unit: def sinusoidal_encode()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 3

id: 71
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 5

id: 72
unit: def attention()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 5

id: 73
unit: def __init__()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 2

id: 74
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 5

id: 75
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 5

id: 76
unit: def sinusoidal_enocde()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 3

id: 77
unit: def attention()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 2
number of parameters: 5

id: 78
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 3

id: 79
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 4
number of parameters: 2

id: 80
unit: def inpainting_validation_data()
file: training/train_muse.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 0

id: 81
unit: def group_by_keys_nothrow()
file: training/data.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 14
number of parameters: 5

id: 82
unit: def distribute_shards()
file: scripts/calculate_fid.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 3

id: 83
unit: def get_captions()
file: scripts/gen_sdxl_synthetic_dataset.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 5
number of parameters: 1

id: 84
unit: def distribute_shards()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 6
number of parameters: 3

id: 85
unit: def __init__()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 4
number of parameters: 3

id: 86
unit: def __init__()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 4
number of parameters: 3

id: 87
unit: def register_to_config()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 8
number of parameters: 2

id: 88
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 0

id: 89
unit: def validate_model()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 2
number of parameters: 5

id: 90
unit: def ds_clean_map()
file: training/data.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 1

id: 91
unit: def config_from_legacy_kwargs()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 6
number of parameters: 1

id: 92
unit: def forward()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 3

id: 93
unit: def forward()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 3

id: 94
unit: def input_ids_to_masked_buckets()
file: muse/training_utils.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 3
number of parameters: 3

id: 95
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 9

id: 96
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 4
number of parameters: 3

id: 97
unit: def __init__()
file: training/data.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 3
number of parameters: 4

id: 98
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 3

id: 99
unit: def __init__()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 2

id: 100
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 2

id: 101
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 9

id: 102
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 2
number of parameters: 4

id: 103
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 3
number of parameters: 3

id: 104
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 4
number of parameters: 3

id: 105
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 4
number of parameters: 3

id: 106
unit: def unfused_rms_norm()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 4

id: 107
unit: def __init__()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 9

id: 108
unit: def main()
file: scripts/calculate_fid.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 5
number of parameters: 1

id: 109
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 4

id: 110
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 4

id: 111
unit: def __init__()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 3

id: 112
unit: def pixel_entropy_per_percent_masked_bucket()
file: muse/training_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 3

id: 113
unit: def image_entropy_per_percent_masked_bucket()
file: muse/training_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 3

id: 114
unit: def _get_model_file()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 115
unit: def num_parameters()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 12
number of parameters: 3

id: 116
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 4

id: 117
unit: def step()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 2

id: 118
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 119
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 0

id: 120
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 121
unit: def validation_masks_to_latent_tensors()
file: training/train_muse.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 1

id: 122
unit: def save_checkpoint()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 3
number of parameters: 4

id: 123
unit: def get_aesthetic_score()
file: training/data.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 6
number of parameters: 1

id: 124
unit: def main()
file: benchmark/muse_chart.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 125
unit: def _init_weights()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 9
number of parameters: 2

id: 126
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 3

id: 127
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 5

id: 128
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 3

id: 129
unit: def forward()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 2

id: 130
unit: def compute_distances()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 2

id: 131
unit: def get_soft_code()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 4

id: 132
unit: def get_soft_code()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 4

id: 133
unit: def __init__()
file: muse/training_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 134
unit: def step()
file: muse/training_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 4
number of parameters: 2

id: 135
unit: def state_dict()
file: muse/training_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 1

id: 136
unit: def cross_entropy_per_percent_masked_bucket()
file: muse/training_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 6

id: 137
unit: def to_json_string()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 5
number of parameters: 1

id: 138
unit: def get_soft_code()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 4

id: 139
unit: def compute_distances()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 2

id: 140
unit: def get_soft_code()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 2
number of parameters: 4

id: 141
unit: def __init__()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 142
unit: def state_dict()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 1

id: 143
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 144
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 3
number of parameters: 2

id: 145
unit: def _init_weights()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 9
number of parameters: 2

id: 146
unit: def generate()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 147
unit: def generate2()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 148
unit: def log_token_probability_distributions()
file: training/train_muse.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 5

id: 149
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 2

id: 150
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 3

id: 151
unit: def _configure_library_root_logger()
file: muse/logging.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 0

id: 152
unit: def save_pretrained()
file: muse/training_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 2

id: 153
unit: def get_decay()
file: muse/training_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 2

id: 154
unit: def get_mask_chedule()
file: muse/sampling.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 2

id: 155
unit: def _load_state_dict_into_model()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 2

id: 156
unit: def lr_lambda()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 1

id: 157
unit: def Normalize()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 3

id: 158
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 3

id: 159
unit: def save_pretrained()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 2

id: 160
unit: def get_decay()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 3
number of parameters: 2

id: 161
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 2

id: 162
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 2

id: 163
unit: def get_vq_model_class()
file: training/train_muse.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 5
number of parameters: 1

id: 164
unit: def soft_target_cross_entropy()
file: training/train_muse.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 3

id: 165
unit: def generate_images()
file: training/train_muse.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 166
unit: def generate_inpainting_images()
file: training/train_muse.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 0

id: 167
unit: def soft_target_cross_entropy()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 2
number of parameters: 3

id: 168
unit: def ds_clean_upscaled_map()
file: training/data.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 1
number of parameters: 1

id: 169
unit: def take_up_to()
file: scripts/gen_sdxl_synthetic_dataset.py
start line: 0
end line: 0
size: 11 LOC
McCabe index: 4
number of parameters: 2

id: 170
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 3

id: 171
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 2

id: 172
unit: def forward()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 2

id: 173
unit: def forward()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 2

id: 174
unit: def _get_default_logging_level()
file: muse/logging.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 0

id: 175
unit: def get_parameter_device()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 1

id: 176
unit: def get_parameter_dtype()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 1

id: 177
unit: def from_pretrained()
file: muse/pipeline_muse.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 178
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 5

id: 179
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 180
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 4

id: 181
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 182
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 1
number of parameters: 0

id: 183
unit: def log_pixel_entropy()
file: training/train_muse.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 5

id: 184
unit: def log_image_entropy()
file: training/train_muse.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 5

id: 185
unit: def log_cross_entropy()
file: training/train_muse.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 8

id: 186
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 187
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 2

id: 188
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 3
number of parameters: 3

id: 189
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 4

id: 190
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 2

id: 191
unit: def _reset_library_root_logger()
file: muse/logging.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 0

id: 192
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 193
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 3

id: 194
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 5

id: 195
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 3

id: 196
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 197
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 2

id: 198
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 2
number of parameters: 3

id: 199
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 200
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 201
unit: def __init__()
file: training/optimizer.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 4
number of parameters: 5

id: 202
unit: def image_transform()
file: training/data.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 2

id: 203
unit: def submit()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 204
unit: def _upload_thread_entrypoint()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 205
unit: def main()
file: benchmark/muse_table.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 206
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 207
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 208
unit: def __init__()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 4

id: 209
unit: def forward()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 210
unit: def __init__()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 4

id: 211
unit: def average_by_buckets()
file: muse/training_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 212
unit: def sigmoid_schedule()
file: muse/sampling.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 5

id: 213
unit: def from_config()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 5

id: 214
unit: def get_linear_schedule_with_warmup()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 4

id: 215
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 216
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 217
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 2

id: 218
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 219
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 3

id: 220
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 221
unit: def encode()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 222
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 223
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 224
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 4
number of parameters: 2

id: 225
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 4

id: 226
unit: def encode()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 227
unit: def to()
file: muse/pipeline_muse.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 228
unit: def to_pil_image()
file: muse/pipeline_muse.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 229
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 2
number of parameters: 3

id: 230
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 231
unit: def __init__()
file: training/data.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 232
unit: def convert()
file: scripts/convert_maskgit_vqgan.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 233
unit: def __init__()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 4

id: 234
unit: def measure_max_memory_allocated()
file: benchmark/muse_perf.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 235
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 3

id: 236
unit: def Norm()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 237
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 238
unit: def forward()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 239
unit: def forward()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 240
unit: def encode()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 241
unit: def forward()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 242
unit: def encode()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 243
unit: def enable_full_determinism()
file: muse/training_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 1

id: 244
unit: def restore()
file: muse/training_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 245
unit: def _load_pretrained_model()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 246
unit: def save_config()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 5

id: 247
unit: def lr_lambda()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 248
unit: def get_scheduler()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 249
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 2

id: 250
unit: def restore()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 251
unit: def __init__()
file: muse/pipeline_muse.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 252
unit: def prob_mask_like()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 3

id: 253
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 2

id: 254
unit: def validate_model()
file: training/train_muse.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 0

id: 255
unit: def get_vq_model_class()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 1

id: 256
unit: def __init__()
file: scripts/calculate_fid.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 2
number of parameters: 3

id: 257
unit: def unfused_layer_norm()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 6

id: 258
unit: def forward()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 2

id: 259
unit: def get_code()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 260
unit: def get_code()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 261
unit: def from_pretrained()
file: muse/training_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 262
unit: def to()
file: muse/training_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 3

id: 263
unit: def top_k()
file: muse/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 264
unit: def mask_by_random_topk()
file: muse/sampling.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 4

id: 265
unit: def save_pretrained()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 266
unit: def get_constant_schedule_with_warmup()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 3

id: 267
unit: def get_code()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 268
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 269
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 2

id: 270
unit: def get_code()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 271
unit: def from_pretrained()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 272
unit: def to()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 3

id: 273
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 274
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 6

id: 275
unit: def log_grad_norm()
file: training/train_muse.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 3

id: 276
unit: def log_grad_norm()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 3

id: 277
unit: def replace_person_token()
file: training/data.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 1

id: 278
unit: def tokenize()
file: training/data.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 279
unit: def tokenize()
file: training/data.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 280
unit: def __exit__()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 3
number of parameters: 4

id: 281
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 282
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 283
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 284
unit: def FeedForward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 285
unit: def __init__()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 286
unit: def get_codebook_entry()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 287
unit: def get_logger()
file: muse/logging.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 288
unit: def enable_explicit_format()
file: muse/logging.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 0

id: 289
unit: def warning_advice()
file: muse/logging.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 290
unit: def __call__()
file: muse/logging.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 291
unit: def get_codebook_entry()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 292
unit: def decode()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 293
unit: def decode_code()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 294
unit: def set_seed()
file: muse/training_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 295
unit: def pow()
file: muse/sampling.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 296
unit: def fn_recursive_set_mem_eff()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 3
number of parameters: 1

id: 297
unit: def _find_mismatched_keys()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 298
unit: def __init__()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 299
unit: def lr_lambda()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 1

id: 300
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 301
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 302
unit: def __init__()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 303
unit: def forward()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 304
unit: def get_codebook_entry()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 305
unit: def get_code()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 306
unit: def forward()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 307
unit: def __init__()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 308
unit: def get_codebook_entry()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 309
unit: def get_code()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 310
unit: def get_config()
file: training/train_muse.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 311
unit: def reset()
file: training/train_muse.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 312
unit: def update()
file: training/train_muse.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 313
unit: def get_config()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 314
unit: def reset()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 315
unit: def update()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 3

id: 316
unit: def tarfile_to_samples_nothrow()
file: training/data.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 317
unit: def get_tar_file_name()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 1

id: 318
unit: def _set_gradient_checkpointing()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 319
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 320
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 321
unit: def forward()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 322
unit: def decode_code()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 323
unit: def get_soft_code()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 324
unit: def get_code()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 325
unit: def disable_default_handler()
file: muse/logging.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 326
unit: def enable_default_handler()
file: muse/logging.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 327
unit: def add_handler()
file: muse/logging.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 328
unit: def remove_handler()
file: muse/logging.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 329
unit: def reset_format()
file: muse/logging.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 0

id: 330
unit: def __getattr__()
file: muse/logging.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 331
unit: def set_lock()
file: muse/logging.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 3

id: 332
unit: def get_code()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 333
unit: def forward()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 334
unit: def copy_to()
file: muse/training_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 335
unit: def linear_schedule()
file: muse/sampling.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 336
unit: def enable_gradient_checkpointing()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 337
unit: def __setattr__()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 3

id: 338
unit: def __setitem__()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 3

id: 339
unit: def _dict_from_json_file()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 340
unit: def decode()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 341
unit: def decode_code()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 342
unit: def decode()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 343
unit: def decode_code()
file: muse/modeling_taming_vqgan.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 344
unit: def copy_to()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 345
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 346
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 347
unit: def create_custom_forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 348
unit: def filter_keys()
file: training/data.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 1

id: 349
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 350
unit: def __init__()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 351
unit: def decode()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 352
unit: def get_verbosity()
file: muse/logging.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 353
unit: def set_verbosity()
file: muse/logging.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 354
unit: def disable_propagation()
file: muse/logging.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 355
unit: def enable_propagation()
file: muse/logging.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 356
unit: def get_lock()
file: muse/logging.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 357
unit: def is_progress_bar_enabled()
file: muse/logging.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 358
unit: def enable_progress_bar()
file: muse/logging.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 359
unit: def disable_progress_bar()
file: muse/logging.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 360
unit: def compute_distances()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 361
unit: def store()
file: muse/training_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 362
unit: def gumbel_noise()
file: muse/sampling.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 363
unit: def disable_gradient_checkpointing()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 1

id: 364
unit: def enable_xformers_memory_efficient_attention()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 365
unit: def disable_xformers_memory_efficient_attention()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 1

id: 366
unit: def to_json_file()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 367
unit: def compute_distances()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 368
unit: def store()
file: muse/modeling_ema.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 369
unit: def save_pretrained()
file: muse/pipeline_muse.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 370
unit: def __init__()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 371
unit: def benchmark_torch_function()
file: scripts/benchmark_models.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 3

id: 372
unit: def chunk()
file: scripts/log_inpainting_images.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 373
unit: def chunk()
file: scripts/log_generations_wandb.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 374
unit: def generate()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 375
unit: def set_use_memory_efficient_attention_xformers()
file: muse/modeling_transformer_v2.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 376
unit: def calc_same_pad()
file: muse/modeling_maskgit_vqgan.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 5

id: 377
unit: def _get_library_name()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 378
unit: def _get_library_root_logger()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 379
unit: def get_log_levels_dict()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 380
unit: def set_verbosity_info()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 381
unit: def set_verbosity_warning()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 382
unit: def set_verbosity_debug()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 383
unit: def set_verbosity_error()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 384
unit: def __init__()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 3

id: 385
unit: def __iter__()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 386
unit: def __enter__()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 387
unit: def __exit__()
file: muse/logging.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 388
unit: def _norm()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 389
unit: def __init__()
file: muse/modeling_paella_vq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 390
unit: def log()
file: muse/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 391
unit: def gumbel_sample()
file: muse/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 392
unit: def cosine_schedule()
file: muse/sampling.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 393
unit: def __init__()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 394
unit: def is_gradient_checkpointing()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 1

id: 395
unit: def set_use_memory_efficient_attention_xformers()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 396
unit: def device()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 397
unit: def dtype()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 398
unit: def __delitem__()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 399
unit: def setdefault()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 400
unit: def pop()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 401
unit: def update()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 402
unit: def load_config()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 403
unit: def _get_init_keys()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 404
unit: def __repr__()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 405
unit: def config()
file: muse/modeling_utils.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 406
unit: def get_constant_schedule()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 407
unit: def get_cosine_schedule_with_warmup()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 408
unit: def get_cosine_with_hard_restarts_schedule_with_warmup()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 409
unit: def get_polynomial_decay_schedule_with_warmup()
file: muse/lr_schedulers.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 410
unit: def set_use_memory_efficient_attention_xformers()
file: muse/modeling_movq.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 411
unit: def uniform()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 412
unit: def make_attention_mask()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 413
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 414
unit: def set_use_memory_efficient_attention_xformers()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 415
unit: def forward()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 416
unit: def _set_gradient_checkpointing()
file: muse/modeling_transformer.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 417
unit: def get_loss_weight()
file: training/train_muse.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 418
unit: def __init__()
file: training/train_muse.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 419
unit: def __init__()
file: training/train_maskgit_imagenet.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 420
unit: def get_orig_size()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 421
unit: def train_dataset()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 422
unit: def train_dataloader()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 423
unit: def eval_dataset()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 424
unit: def eval_dataloader()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 425
unit: def train_dataset()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 426
unit: def train_dataloader()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 427
unit: def eval_dataset()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 428
unit: def eval_dataloader()
file: training/data.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 429
unit: def __len__()
file: scripts/calculate_fid.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 430
unit: def __getitem__()
file: scripts/calculate_fid.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 431
unit: def format_shard_number()
file: scripts/calculate_fid.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 432
unit: def format_shard_number()
file: scripts/gen_sdxl_synthetic_dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 433
unit: def split_list()
file: scripts/gen_sdxl_synthetic_dataset.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 434
unit: def format_shard_number()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 435
unit: def __enter__()
file: scripts/pre_encode.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1