wsj1995

/

VAE

like 1

• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_11.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_184.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_187.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_6.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_18.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_186.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_121.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_174.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_204.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_162.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_231.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_75.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_12.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_122.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_88.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_152.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_212.Conv2d",
• "collections.OrderedDict",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_157.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_137.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_141.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_230.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.ImageEmbedding",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_14.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_54.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_176.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_86.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_98.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_53.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_59.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_60.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_2.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_109.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_116.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_158.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_96.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_27.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_62.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_24.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_36.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_221.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_232.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_139.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.TimestepEmbedding",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_165.Conv2d",
• "__torch__.torch.nn.modules.container.ModuleDict",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_99.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_143.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_31.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_136.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_8.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_164.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_56.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_21.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_67.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_203.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_129.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_173.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_178.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_66.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_103.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_1.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_89.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_90.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_123.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_83.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_216.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_50.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_159.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_73.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_224.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_111.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_195.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_113.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_177.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_114.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_34.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_37.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_43.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_117.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_57.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_110.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_169.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_10.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_91.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_105.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_51.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_25.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_115.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_78.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_30.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_39.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_215.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_223.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_166.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_219.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_227.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_13.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_94.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.ImageUnembedding",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_22.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_100.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_45.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_104.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.ConvUNetVAE",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_179.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_85.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_168.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_194.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_154.ConditionalGroupNorm",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_180.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_134.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_9.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_79.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_142.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_171.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_49.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_102.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_149.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_197.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_16.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_118.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_107.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_68.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_181.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_106.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_112.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_131.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_47.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_196.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_58.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_65.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_120.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_209.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_225.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_41.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_108.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_84.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_5.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_130.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_144.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_128.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_160.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_151.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_7.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_20.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_147.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_126.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_233.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_132.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_76.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_153.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_77.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_95.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_138.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_125.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_33.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_23.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_205.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_200.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_150.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_69.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_229.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_35.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_124.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_210.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_188.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_183.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_198.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_64.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_185.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_156.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_55.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_208.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_92.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_133.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_146.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_101.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_190.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_213.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_167.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_140.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_172.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_175.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_226.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_155.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_163.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_46.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_217.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_127.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_228.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_97.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_71.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_17.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_206.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_207.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_218.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_29.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_202.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_87.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_42.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_19.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_145.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_191.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_72.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_189.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_38.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_199.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_4.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_192.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_214.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_211.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_63.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_40.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_48.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_74.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_170.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_193.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_26.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_82.ConditionalGroupNorm",
• "torch.FloatStorage",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_135.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.GroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_161.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_0.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_61.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_28.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_32.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_81.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_93.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_70.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_15.ConvResblock",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_222.ConditionalGroupNorm",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_80.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_201.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_148.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_3.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_52.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_182.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_220.Conv2d",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_44.Affine",
• "__torch__.dalle_runner_api.model_infra.modules.public_diff_vae.___torch_mangle_119.Affine",
• "torch._utils._rebuild_tensor_v2",
• "torch.FloatStorage",
• "collections.OrderedDict",
• "torch.DoubleStorage",
• "torch.LongStorage",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "torch.jit._pickle.restore_type_tag",
• "torch.FloatStorage"

How to fix it?