Models
Datasets
Spaces
Posts
Docs
Enterprise
Pricing
Log In
Sign Up

chaidiscovery
/

chai-1

Model card Files Files and versions Community

1 contributor

History: 6 commits

jackdent's picture

Upload token_embedder.pt

8ca69c6 verified 4 months ago

.gitattributes

1.52 kB

initial commit 4 months ago
README.md

1.89 kB

Update README.md 4 months ago
confidence_head.pt
Detected Pickle imports (120)
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1902.TransitionLayer_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1947.Linear",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1984.AttentionPairBiasBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1937.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1971.Linear",
- "torch.FloatStorage",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1922.Linear",
- "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1948.TriangleAttentionUpdate_v1",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1924.LayerNorm",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1961.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1966.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1918.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1938.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1945.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1898.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1943.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1974.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1894.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1940.LayerNorm",
- "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1919.TriangleMultiplicativeUpdateFastv2r",
- "__torch__.einops.layers.torch.___torch_mangle_1956.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1967.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1942.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1887.LayerNorm",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1899.LayerNorm",
- "__torch__.einops.layers.torch.___torch_mangle_1931.EinMix",
- "__torch__.torch.nn.modules.container.___torch_mangle_1986.ModuleList",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1897.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1911.LayerNorm",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1965.LayerNorm",
- "__torch__.chai.model.blocks3.triangle_attention.TriangleAttentionUpdate_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1921.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1951.Linear",
- "__torch__.chai.model.modules.af3_confidence_head.ConfidenceHead",
- "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1960.PairformerBlock",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1909.AttentionPairBiasBlock",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1949.LayerNorm",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1964.TransitionLayer_v1",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1928.LayerNorm",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1970.LayerNorm",
- "collections.OrderedDict",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1963.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1916.Linear",
- "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1973.TriangleAttentionUpdate_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1962.Linear",
- "__torch__.chai.model.primitives_fast.___torch_mangle_1983.GatedSelfAttention",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1892.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1957.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1885.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1915.LayerNorm",
- "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1985.PairformerBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1913.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1929.LayerNorm",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1891.LayerNorm",
- "__torch__.chai.model.blocks3.triangle_attention.___torch_mangle_1923.TriangleAttentionUpdate_v1",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1952.TransitionLayer_v1",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1904.LayerNorm",
- "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1895.TriangleMultiplicativeUpdateFastv2r",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1905.Linear",
- "__torch__.chai.model.primitives_fast.___torch_mangle_1908.GatedSelfAttention",
- "__torch__.einops.layers.torch.___torch_mangle_1932.EinMix",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1920.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1972.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1950.Linear",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1959.AttentionPairBiasBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1976.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1896.LayerNorm",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1927.TransitionLayer_v1",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1978.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1941.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1975.Linear",
- "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1969.TriangleMultiplicativeUpdateFastv2r",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1917.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1900.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1886.Linear",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1934.AttentionPairBiasBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1925.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1926.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1946.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1989.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1901.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1906.EinMix",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1914.TransitionLayer_v1",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1977.TransitionLayer_v1",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1890.TransitionLayer_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1889.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1912.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1988.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1953.LayerNorm",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1903.LayerNorm",
- "__torch__.einops.layers.torch.___torch_mangle_1907.EinMix",
- "__torch__.chai.model.modules.triangular_multiplicative_update_fast.___torch_mangle_1944.TriangleMultiplicativeUpdateFastv2r",
- "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1910.PairformerBlock",
- "__torch__.chai.model.primitives_fast.___torch_mangle_1933.GatedSelfAttention",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1936.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1888.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1930.Linear",
- "torch._utils._rebuild_tensor_v2",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1954.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1955.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1987.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1981.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1980.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1893.Linear",
- "__torch__.chai.model.primitives_fast.___torch_mangle_1958.GatedSelfAttention",
- "__torch__.einops.layers.torch.___torch_mangle_1982.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1968.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1979.LayerNorm",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1939.TransitionLayer_v1",
- "__torch__.chai.model.modules.af3_pairformer.___torch_mangle_1935.PairformerBlock",
- "torch.LongStorage",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch.LongStorage",
- "torch.BoolStorage",
- "torch.BFloat16Storage",
- "torch.jit._pickle.restore_type_tag",
- "torch._utils._rebuild_tensor_v2"
How to fix it?
55.3 MB
LFS

Upload confidence_head.pt 4 months ago
feature_embedding.pt
Detected Pickle imports (67)
- "torch.LongStorage",
- "__torch__.torch.nn.modules.container.___torch_mangle_21.ModuleDict",
- "__torch__.torch.nn.modules.container.ModuleDict",
- "__torch__.chai.model.embedding.identity.___torch_mangle_10.Identity",
- "__torch__.chai.model.embedding.identity.___torch_mangle_1.Identity",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_7.OneHot",
- "__torch__.torch.nn.modules.sparse.Embedding",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_13.OneHot",
- "__torch__.chai.model.embedding.one_hot.OneHot",
- "__torch__.chai.model.embedding.identity.___torch_mangle_9.Identity",
- "__torch__.torch.nn.modules.container.___torch_mangle_43.ModuleDict",
- "__torch__.torch.nn.modules.linear.___torch_mangle_37.Linear",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_16.OneHot",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_15.OneHot",
- "__torch__.chai.model.embedding.identity.___torch_mangle_4.Identity",
- "__torch__.chai.model.embedding.rbf.RBF",
- "__torch__.chai.model.embedding.identity.___torch_mangle_25.Identity",
- "__torch__.torch.nn.modules.container.___torch_mangle_34.Sequential",
- "__torch__.torch.nn.modules.container.___torch_mangle_31.ModuleDict",
- "__torch__.chai.model.embedding.identity.___torch_mangle_22.Identity",
- "torch.FloatStorage",
- "__torch__.torch.nn.modules.linear.___torch_mangle_35.Linear",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_0.OneHot",
- "__torch__.chai.model.embedding.feature_embedding.FeatureEmbedding",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_23.OneHot",
- "torch._utils._rebuild_tensor_v2",
- "__torch__.chai.model.embedding.identity.Identity",
- "__torch__.chai.model.embedding.identity.___torch_mangle_8.Identity",
- "__torch__.torch.nn.modules.container.___torch_mangle_5.ModuleDict",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_11.OneHot",
- "__torch__.chai.model.embedding.identity.___torch_mangle_24.Identity",
- "__torch__.torch.nn.modules.linear.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_42.Sequential",
- "__torch__.chai.model.embedding.esm.ESMEmbedding",
- "__torch__.torch.nn.modules.container.___torch_mangle_38.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_39.Linear",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_17.OneHot",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_18.OneHot",
- "__torch__.torch.nn.modules.container.___torch_mangle_14.ModuleDict",
- "__torch__.torch.nn.modules.container.___torch_mangle_40.Sequential",
- "__torch__.torch.nn.modules.container.___torch_mangle_32.ModuleDict",
- "__torch__.torch.nn.modules.container.Sequential",
- "collections.OrderedDict",
- "__torch__.torch.nn.modules.container.___torch_mangle_36.Sequential",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_26.OneHot",
- "__torch__.torch.nn.modules.linear.___torch_mangle_41.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_27.ModuleDict",
- "__torch__.chai.model.embedding.identity.___torch_mangle_29.Identity",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_28.OneHot",
- "__torch__.chai.model.embedding.rbf.___torch_mangle_20.RBF",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_3.OneHot",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_19.OneHot",
- "__torch__.chai.model.embedding.one_hot.___torch_mangle_12.OneHot",
- "__torch__.chai.model.embedding.identity.___torch_mangle_30.Identity",
- "__torch__.torch.nn.modules.linear.___torch_mangle_33.Linear",
- "__torch__.chai.model.embedding.identity.___torch_mangle_6.Identity",
- "__torch__.chai.model.embedding.identity.___torch_mangle_2.Identity",
- "__torch__.chai.model.embedding.outer_sum.OuterSumProj",
- "torch.LongStorage",
- "torch.FloatStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.DoubleStorage",
- "collections.OrderedDict",
- "torch.jit._pickle.restore_type_tag",
- "torch.BFloat16Storage",
- "collections.OrderedDict",
- "torch._utils._rebuild_tensor_v2"
How to fix it?
4.59 MB
LFS

Upload feature_embedding.pt 4 months ago
token_embedder.pt
Detected Pickle imports (77)
- "__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?
6.29 MB
LFS

Upload token_embedder.pt 4 months ago