chaidiscovery

/

chai-1

like 16

Follow

Chai Discovery 10

• "__torch__.chai.model.modules.af3_attention_pair_bias_block.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_48.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_70.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_82.EinMix",
• "__torch__.torch.nn.modules.container.___torch_mangle_53.Sequential",
• "__torch__.torch.nn.modules.activation.___torch_mangle_89.ReLU",
• "__torch__.torch.nn.modules.linear.___torch_mangle_94.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_86.EinMix",
• "__torch__.chai.model.modules.af3_atom_transformer.AtomTransformer",
• "__torch__.torch.nn.modules.linear.___torch_mangle_77.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_64.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_90.Sequential",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_71.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.container.___torch_mangle_85.ModuleList",
• "__torch__.torch.nn.modules.linear.___torch_mangle_68.Linear",
• "torch.FloatStorage",
• "__torch__.torch.nn.modules.linear.___torch_mangle_91.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.AdaLayerNorm_v2",
• "__torch__.chai.model.modules.af3_diffusion_transformer.LocalDiffusionTransformer",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_84.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.normalization.LayerNorm",
• "__torch__.torch.nn.modules.activation.___torch_mangle_47.ReLU",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.nn.modules.container.ModuleList",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_63.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_60.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_66.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_56.Linear",
• "__torch__.einops.layers.torch.Rearrange",
• "__torch__.torch.nn.modules.linear.___torch_mangle_83.Linear",
• "__torch__.chai.model.af3.token_input_emb.TokenInputEmbedding",
• "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_69.ConditionedTransitionBlock_v1",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_74.AdaLayerNorm_v2",
• "__torch__.chai.model.modules.af3_input_embedder.AF3InputEmbedder",
• "__torch__.torch.nn.modules.linear.___torch_mangle_57.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_80.AdaLayerNorm_v2",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_59.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_50.Linear",
• "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_78.LocalAttentionPairBiasBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_55.Linear",
• "__torch__.chai.model.modules.af3_atom_attention_encoder_blocked.AtomPairUpdateBlockedBlock",
• "__torch__.torch.nn.modules.linear.___torch_mangle_62.Linear",
• "__torch__.einops.layers.torch.EinMix",
• "__torch__.torch.nn.modules.activation.ReLU",
• "collections.OrderedDict",
• "__torch__.torch.nn.modules.linear.___torch_mangle_79.Linear",
• "__torch__.chai.model.modules.af3_atom_attention_encoder_blocked.AtomAttentionBlockedEncoder",
• "__torch__.torch.nn.modules.linear.___torch_mangle_92.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_61.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_72.Linear",
• "__torch__.chai.model.blocks3.cond_transition.ConditionedTransitionBlock_v1",
• "__torch__.torch.nn.modules.linear.___torch_mangle_67.Linear",
• "__torch__.torch.nn.modules.container.___torch_mangle_87.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_54.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_76.EinMix",
• "__torch__.torch.nn.modules.linear.___torch_mangle_58.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_88.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_52.Linear",
• "__torch__.torch.nn.modules.linear.___torch_mangle_73.Linear",
• "__torch__.torch.nn.modules.activation.___torch_mangle_51.ReLU",
• "__torch__.einops.layers.torch.___torch_mangle_75.Rearrange",
• "__torch__.torch.nn.modules.linear.___torch_mangle_44.Linear",
• "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_65.AdaLayerNorm_v2",
• "__torch__.torch.nn.modules.linear.___torch_mangle_93.Linear",
• "__torch__.einops.layers.torch.___torch_mangle_81.Rearrange",
• "__torch__.torch.nn.modules.container.___torch_mangle_46.Sequential",
• "__torch__.torch.nn.modules.container.___torch_mangle_49.Sequential",
• "__torch__.torch.nn.modules.linear.___torch_mangle_45.Linear",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "torch.LongStorage",
• "torch.LongStorage",
• "torch.BFloat16Storage",
• "torch.jit._pickle.restore_type_tag",
• "torch._utils._rebuild_tensor_v2",
• "collections.OrderedDict",
• "torch.BoolStorage"

How to fix it?