chaidiscovery

/

chai-1

like 16

Follow

Chai Discovery 10

• "__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?