1 contributor

History: 7 commits

jackdent

Upload diffusion_module.pt

d11304e verified 2 months ago

.gitattributes

1.52 kB

initial commit 3 months ago
README.md

1.89 kB

Update README.md 3 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 2 months ago
diffusion_module.pt
Detected Pickle imports (418)
- "__torch__.torch.nn.modules.linear.___torch_mangle_1845.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1834.ModuleList",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1656.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1524.Sequential",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1490.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1760.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1874.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1639.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1575.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1728.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1552.EinMix",
- "__torch__.torch.nn.modules.container.___torch_mangle_1521.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1584.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1488.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1643.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1859.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1686.LayerNorm",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1581.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1522.ReLU",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1833.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1622.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1735.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1677.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1569.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1811.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1635.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1548.ModuleList",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1694.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1605.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1770.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1777.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1599.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1561.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1609.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1835.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1623.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1616.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1624.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1494.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1689.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1708.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1721.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1704.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1732.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1593.Linear",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1504.TransitionLayer_v1",
- "__torch__.torch.nn.modules.container.___torch_mangle_1749.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1839.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1852.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1873.ModuleList",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1596.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1571.LocalDiffusionTransformer",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1549.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1794.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1670.LayerNorm",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1663.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1710.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1875.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1680.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1773.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1861.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1712.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1805.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1534.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1724.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1532.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1665.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1675.AdaLayerNorm_v2",
- "__torch__.chai.model.modules.af3_diffusion_transformer.DenseDiffusionTransformer",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1697.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1810.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1780.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1806.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1858.EinMix",
- "__torch__.chai.model.modules.af3_atom_attention_encoder_blocked.___torch_mangle_1529.AtomPairUpdateBlockedBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1520.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1668.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1693.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1646.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1701.Sequential",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1627.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1783.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1648.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1775.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1513.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1808.Linear",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1872.LocalAttentionPairBiasBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1880.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1830.LayerNorm",
- "__torch__.torch.nn.modules.container.___torch_mangle_1492.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1642.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1530.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1650.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1536.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1750.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1690.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1586.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1766.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1645.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1702.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1821.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1831.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1594.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1600.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1489.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1882.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1853.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1816.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1598.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1657.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1667.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1828.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1832.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1610.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1864.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1589.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1539.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1519.ReLU",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1681.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1752.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1736.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1628.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1854.ModuleList",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1761.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1629.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1826.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1814.LayerNorm",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1759.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1774.Linear",
- "__torch__.chai.model.modules.af3_atom_attention_encoder_blocked.___torch_mangle_1579.AtomAttentionBlockedEncoder",
- "__torch__.torch.nn.modules.container.___torch_mangle_1717.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1851.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1725.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1862.AdaLayerNorm_v2",
- "__torch__.einops.layers.torch.___torch_mangle_1857.Rearrange",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1771.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1499.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1719.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1608.Linear",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1860.LocalAttentionPairBiasBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1678.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1786.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1660.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1841.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1722.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1877.LocalDiffusionTransformer",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1692.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1791.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.container.___torch_mangle_1797.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1871.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1487.LayerNorm",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1560.LocalAttentionPairBiasBlock",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1679.ConditionedTransitionBlock_v1",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1809.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1619.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1742.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1842.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1637.Sequential",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1700.Sigmoid",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1541.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1838.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1801.DiffusionTransformerBlock",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1785.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1501.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1502.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1767.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1652.Sigmoid",
- "__torch__.torch.nn.modules.activation.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1559.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1883.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1588.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1638.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1720.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1779.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1738.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1745.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1782.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1784.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1493.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1757.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1662.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1881.Sequential",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1535.ConditionedTransitionBlock_v1",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1743.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1763.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1837.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1651.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1740.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1753.DiffusionTransformerBlock",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1807.ConditionedTransitionBlock_v1",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1691.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1822.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1684.Sigmoid",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1798.LayerNorm",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1576.LayerNorm",
- "__torch__.torch.nn.modules.container.___torch_mangle_1781.Sequential",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1568.LayerNorm",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1601.AdaLayerNorm_v2",
- "__torch__.chai.model.modules.af3_atom_transformer.___torch_mangle_1878.AtomTransformer",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1546.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1603.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1699.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1744.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1709.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1716.Sigmoid",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1705.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.container.___torch_mangle_1570.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1698.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1734.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1503.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1755.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1543.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1647.ConditionedTransitionBlock_v1",
- "__torch__.einops.layers.torch.___torch_mangle_1563.Rearrange",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1498.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1855.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1615.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1654.LayerNorm",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1823.ConditionedTransitionBlock_v1",
- "__torch__.einops.layers.torch.___torch_mangle_1564.EinMix",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1707.AdaLayerNorm_v2",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1554.LocalAttentionPairBiasBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1523.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1577.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1695.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1509.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1820.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1723.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1580.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1604.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1815.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1565.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1756.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1868.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1611.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1636.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1518.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1812.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1788.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1525.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1803.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1824.Linear",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1866.LocalAttentionPairBiasBlock",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1574.ReLU",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1540.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1696.Linear",
- "__torch__.chai.model.modules.af3_fourier_emb.FourierEmbed",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1661.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1595.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1819.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1741.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1765.Sequential",
- "__torch__.chai.model.modules.af3_atom_transformer.___torch_mangle_1572.AtomTransformer",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1847.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1544.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1621.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1630.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1506.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1578.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1632.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1533.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1682.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1840.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1553.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1538.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1844.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1653.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1776.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1649.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1825.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1703.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1614.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1747.Linear",
- "__torch__.chai.model.modules.af3_attention_pair_bias_block.___torch_mangle_1566.LocalAttentionPairBiasBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1583.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1796.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1495.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1711.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1602.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1505.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1730.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1547.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1846.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1879.LayerNorm",
- "__torch__.torch.nn.modules.container.___torch_mangle_1669.Sequential",
- "__torch__.torch.nn.modules.container.___torch_mangle_1829.Sequential",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1562.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1850.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1876.Sequential",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1787.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1768.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1813.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1792.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1802.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1658.Linear",
- "torch._utils._rebuild_tensor_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1731.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1550.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1790.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1491.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1512.LayerNorm",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1591.LayerNorm",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1727.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1848.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1558.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1612.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1869.Rearrange",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1625.DiffusionTransformerBlock",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1500.TransitionLayer_v1",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1748.Sigmoid",
- "torch.FloatStorage",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1597.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1751.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1827.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1739.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1497.LayerNorm",
- "__torch__.torch.nn.modules.container.___torch_mangle_1517.Sequential",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1496.TransitionLayer_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1664.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1804.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1590.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1818.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1655.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1551.Rearrange",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1688.Linear",
- "__torch__.chai.model.modules.af3_atom_attention_decoder.AtomAttentionDecoder",
- "__torch__.chai.model.blocks3.transition_layer.___torch_mangle_1508.TransitionLayer_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1726.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1640.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1706.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1620.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1676.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1659.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1764.Sigmoid",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1836.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1795.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1631.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1758.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1515.LayerNorm",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1633.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1671.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1718.LayerNorm",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1587.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1531.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1799.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1641.DiffusionTransformerBlock",
- "collections.OrderedDict",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1617.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1545.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1843.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1573.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1528.Sequential",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1849.AdaLayerNorm_v2",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1769.DiffusionTransformerBlock",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1817.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.container.___torch_mangle_1685.Sequential",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1673.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1674.Linear",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1606.LayerNorm",
- "__torch__.torch.nn.modules.container.___torch_mangle_1511.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1618.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1527.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1557.Rearrange",
- "__torch__.torch.nn.modules.normalization.___torch_mangle_1884.LayerNorm",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1746.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1516.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1510.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1672.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1856.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1729.AdaLayerNorm_v2",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1537.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1800.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1514.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1863.Rearrange",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1607.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1555.Linear",
- "__torch__.torch.nn.modules.activation.___torch_mangle_1526.ReLU",
- "__torch__.torch.nn.modules.container.___torch_mangle_1733.Sequential",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1542.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1683.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1772.Linear",
- "__torch__.chai.model.modules.af3_diffusion_transformer.___torch_mangle_1737.DiffusionTransformerBlock",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1762.Linear",
- "__torch__.einops.layers.torch.___torch_mangle_1870.EinMix",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1865.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1507.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1556.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1714.Linear",
- "__torch__.chai.model.modules.af3_diffusion_module.DiffusionModule",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1634.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1687.Linear",
- "__torch__.chai.model.blocks3.cond_transition.___torch_mangle_1585.ConditionedTransitionBlock_v1",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1789.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1867.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1666.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1582.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1754.Linear",
- "__torch__.torch.nn.modules.container.___torch_mangle_1567.ModuleList",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1715.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1613.Linear",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1793.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1644.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1592.Linear",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1626.Linear",
- "__torch__.chai.model.modules.af3_diffusion_conditioning.DiffusionConditioning",
- "__torch__.chai.model.blocks3.ada_layernorm.___torch_mangle_1713.AdaLayerNorm_v2",
- "__torch__.torch.nn.modules.linear.___torch_mangle_1778.Linear",
- "torch.LongStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.DoubleStorage",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "collections.OrderedDict",
- "torch.LongStorage",
- "torch.BoolStorage",
- "torch.jit._pickle.restore_type_tag",
- "torch._utils._rebuild_tensor_v2"
How to fix it?
477 MB
LFS

Upload diffusion_module.pt 2 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 2 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 2 months ago

Detected Pickle imports (120)

Detected Pickle imports (418)

Detected Pickle imports (67)

Detected Pickle imports (77)