(
self,
*,
ch=128,
ch_mult=(1, 2, 4, 8),
num_res_blocks=2,
dropout=0.0,
in_channels=3, # in_channels: raw img channels
z_channels,
using_sa=True,
using_mid_sa=True,
)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected