| { | |
| "architectures": [ | |
| "MLEForAnimeLineExtraction" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_mle.MLEConfig", | |
| "AutoModel": "modeling_mle.MLEForAnimeLineExtraction" | |
| }, | |
| "custom_pipelines": { | |
| "image-to-image": { | |
| "impl": "MangaPipe.MangaLineExtractionPipe", | |
| "pt": [ | |
| "AutoModel" | |
| ], | |
| "tf": [], | |
| "type": "image" | |
| } | |
| }, | |
| "batch_norm_eps": 0.001, | |
| "block_kernel_size": 3, | |
| "block_patch_size": 24, | |
| "block_stride_size": 4, | |
| "hidden_act": "leaky_relu", | |
| "in_channels": 1, | |
| "last_hidden_channels": 16, | |
| "model_type": "mle", | |
| "negative_slope": 0.2, | |
| "num_decoder_layers": [ | |
| 7, | |
| 5, | |
| 3, | |
| 2, | |
| 2 | |
| ], | |
| "num_encoder_layers": [ | |
| 2, | |
| 3, | |
| 5, | |
| 7, | |
| 12 | |
| ], | |
| "torch_dtype": "bfloat16", | |
| "transformers_version": "4.37.2", | |
| "upsample_ratio": 2 | |
| } | |