(
self,
resolution=256,
num_channels=3,
hidden_channels=128,
channel_mult=(1, 2, 2, 4, 6),
num_res_blocks=2,
attn_resolutions=(),
no_attn_mid_block=True,
z_channels=64,
num_embeddings=8192,
quantized_embed_dim=64,
dropout=0.0,
resample_with_conv=True,
commitment_cost=0.25
)
| 60 | model_type = "vqgan" |
| 61 | |
| 62 | def __init__( |
| 63 | self, |
| 64 | resolution=256, |
| 65 | num_channels=3, |
| 66 | hidden_channels=128, |
| 67 | channel_mult=(1, 2, 2, 4, 6), |
| 68 | num_res_blocks=2, |
| 69 | attn_resolutions=(), |
| 70 | no_attn_mid_block=True, |
| 71 | z_channels=64, |
| 72 | num_embeddings=8192, |
| 73 | quantized_embed_dim=64, |
| 74 | dropout=0.0, |
| 75 | resample_with_conv=True, |
| 76 | commitment_cost=0.25 |
| 77 | ): |
| 78 | self.resolution = resolution |
| 79 | self.num_channels = num_channels |
| 80 | self.hidden_channels = hidden_channels |
| 81 | self.channel_mult = channel_mult |
| 82 | self.num_res_blocks = num_res_blocks |
| 83 | self.attn_resolutions = attn_resolutions |
| 84 | self.no_attn_mid_block = no_attn_mid_block |
| 85 | self.z_channels = z_channels |
| 86 | self.num_embeddings = num_embeddings |
| 87 | self.quantized_embed_dim = quantized_embed_dim |
| 88 | self.dropout = dropout |
| 89 | self.resample_with_conv = resample_with_conv |
| 90 | self.commitment_cost = commitment_cost |
| 91 | |
| 92 | @classmethod |
| 93 | def get_default_config(cls, updates=None): |
nothing calls this directly
no outgoing calls
no test coverage detected