44.6 GB

Ctrl+K

1 contributor

History: 17 commits

This model has 17 files scanned as unsafe.

bi0s

Upload joint_diffusion.ckpt with huggingface_hub

50890ee verified 2 months ago

protein_ckpt
Upload protein_ckpt/cath_ema_260.ckpt with huggingface_hub 2 months ago
.gitattributes

1.52 kB
initial commit 4 months ago
epoch=3799_val_rmsd=0.00038.ckpt
Detected Pickle imports (14)
- "typing.Any",
- "torch._utils._rebuild_tensor_v2",
- "omegaconf.listconfig.ListConfig",
- "collections.defaultdict",
- "omegaconf.base.ContainerMetadata",
- "torch.FloatStorage",
- "__builtin__.dict",
- "__builtin__.long",
- "functools.partial",
- "omegaconf.base.Metadata",
- "torch.optim.adam.Adam",
- "omegaconf.nodes.AnyNode",
- "collections.OrderedDict",
- "__builtin__.list"
How to fix it?
22.4 MB
xet

Upload epoch=3799_val_rmsd=0.00038.ckpt with huggingface_hub 2 months ago
geom_kl_16.ckpt
Detected Pickle imports (46)
- "torch._C._nn.gelu",
- "tabasco.models.components.positional_encoder.SinusoidEncoding",
- "tabasco.models.components.transformer_module.TransformerModule",
- "omegaconf.listconfig.ListConfig",
- "omegaconf.base.ContainerMetadata",
- "tabasco.flow.interpolate.SDEMetricInterpolant",
- "torch.nn.modules.activation.MultiheadAttention",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.transformer.TransformerDecoderLayer",
- "tabasco.models.components.transformer.TransformerBlock",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.functional.relu",
- "torch._utils._rebuild_tensor_v2",
- "torch.Size",
- "torch.optim.adam.Adam",
- "tabasco.models.components.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.dropout.Dropout",
- "tabasco.flow.interpolate.DiscreteInterpolant",
- "torch._utils._rebuild_parameter",
- "tabasco.models.flow_model.FlowMatchingModel",
- "omegaconf.base.Metadata",
- "tabasco.models.components.attention.AttentionBlock",
- "collections.defaultdict",
- "tabasco.models.components.attention.Attention",
- "tabasco.flow.time_factor.InverseTimeFactor",
- "torch.distributions.beta.Beta",
- "tabasco.models.components.transformer.Transformer",
- "torch.nn.modules.loss.MSELoss",
- "torch.nn.modules.normalization.LayerNorm",
- "__builtin__.long",
- "__builtin__.dict",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.distributions.dirichlet.Dirichlet",
- "tabasco.models.components.transition.Transition",
- "__builtin__.list",
- "torch.nn.modules.sparse.Embedding",
- "__builtin__.set",
- "omegaconf.nodes.AnyNode",
- "tabasco.sample.noise_schedule.SampleNoiseSchedule",
- "functools.partial",
- "typing.Any"
How to fix it?
78.6 MB
xet

Upload geom_kl_16.ckpt with huggingface_hub 4 months ago
geom_kl_32.ckpt
Detected Pickle imports (46)
- "torch._C._nn.gelu",
- "tabasco.models.components.positional_encoder.SinusoidEncoding",
- "tabasco.models.components.transformer_module.TransformerModule",
- "omegaconf.listconfig.ListConfig",
- "omegaconf.base.ContainerMetadata",
- "tabasco.flow.interpolate.SDEMetricInterpolant",
- "torch.nn.modules.activation.MultiheadAttention",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.transformer.TransformerDecoderLayer",
- "tabasco.models.components.transformer.TransformerBlock",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.functional.relu",
- "torch._utils._rebuild_tensor_v2",
- "torch.Size",
- "torch.optim.adam.Adam",
- "tabasco.models.components.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.dropout.Dropout",
- "tabasco.flow.interpolate.DiscreteInterpolant",
- "torch._utils._rebuild_parameter",
- "tabasco.models.flow_model.FlowMatchingModel",
- "omegaconf.base.Metadata",
- "tabasco.models.components.attention.AttentionBlock",
- "collections.defaultdict",
- "tabasco.models.components.attention.Attention",
- "tabasco.flow.time_factor.InverseTimeFactor",
- "torch.distributions.beta.Beta",
- "tabasco.models.components.transformer.Transformer",
- "torch.nn.modules.loss.MSELoss",
- "torch.nn.modules.normalization.LayerNorm",
- "__builtin__.long",
- "__builtin__.dict",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.distributions.dirichlet.Dirichlet",
- "tabasco.models.components.transition.Transition",
- "__builtin__.list",
- "torch.nn.modules.sparse.Embedding",
- "__builtin__.set",
- "omegaconf.nodes.AnyNode",
- "tabasco.sample.noise_schedule.SampleNoiseSchedule",
- "functools.partial",
- "typing.Any"
How to fix it?
78.7 MB
xet

Upload geom_kl_32.ckpt with huggingface_hub 4 months ago
geom_kl_8.ckpt
Detected Pickle imports (46)
- "torch._C._nn.gelu",
- "tabasco.models.components.positional_encoder.SinusoidEncoding",
- "tabasco.models.components.transformer_module.TransformerModule",
- "omegaconf.listconfig.ListConfig",
- "omegaconf.base.ContainerMetadata",
- "tabasco.flow.interpolate.SDEMetricInterpolant",
- "torch.nn.modules.activation.MultiheadAttention",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.transformer.TransformerDecoderLayer",
- "tabasco.models.components.transformer.TransformerBlock",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.functional.relu",
- "torch._utils._rebuild_tensor_v2",
- "torch.Size",
- "torch.optim.adam.Adam",
- "tabasco.models.components.positional_encoder.TimeFourierEncoding",
- "torch.nn.modules.dropout.Dropout",
- "tabasco.flow.interpolate.DiscreteInterpolant",
- "torch._utils._rebuild_parameter",
- "tabasco.models.flow_model.FlowMatchingModel",
- "omegaconf.base.Metadata",
- "tabasco.models.components.attention.AttentionBlock",
- "collections.defaultdict",
- "tabasco.models.components.attention.Attention",
- "tabasco.flow.time_factor.InverseTimeFactor",
- "torch.distributions.beta.Beta",
- "tabasco.models.components.transformer.Transformer",
- "torch.nn.modules.loss.MSELoss",
- "torch.nn.modules.normalization.LayerNorm",
- "__builtin__.long",
- "__builtin__.dict",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.distributions.dirichlet.Dirichlet",
- "tabasco.models.components.transition.Transition",
- "__builtin__.list",
- "torch.nn.modules.sparse.Embedding",
- "__builtin__.set",
- "omegaconf.nodes.AnyNode",
- "tabasco.sample.noise_schedule.SampleNoiseSchedule",
- "functools.partial",
- "typing.Any"
How to fix it?
78.6 MB
xet

Upload geom_kl_8.ckpt with huggingface_hub 4 months ago
joint_diffusion.ckpt
Detected Pickle imports (33)
- "src.tabasco.models.denoisers.dit.DiT",
- "torch.nn.modules.container.ModuleList",
- "src.tabasco.flow.interpolate.FlowMatchingInterpolant",
- "omegaconf.dictconfig.DictConfig",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.activation.SiLU",
- "src.tabasco.models.denoisers.dit.DiTBlock",
- "omegaconf.base.Metadata",
- "src.tabasco.models.denoisers.dit.Mlp",
- "src.tabasco.models.denoisers.dit.TimestepEmbedder",
- "torch.nn.modules.activation.MultiheadAttention",
- "functools.partial",
- "torch.nn.modules.linear.Linear",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.container.Sequential",
- "__builtin__.dict",
- "__builtin__.set",
- "collections.OrderedDict",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.nn.modules.linear.Identity",
- "torch.device",
- "torch.FloatStorage",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.activation.GELU",
- "typing.Any",
- "torch.nn.modules.sparse.Embedding",
- "src.tabasco.models.denoisers.dit.FinalLayer",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.dropout.Dropout",
- "collections.defaultdict",
- "src.tabasco.models.denoisers.dit.LabelEmbedder"
How to fix it?
9.16 GB
xet

Upload joint_diffusion.ckpt with huggingface_hub 2 months ago
joint_kl_16.ckpt

22.4 MB
xet

Upload joint_kl_16.ckpt with huggingface_hub 4 months ago
joint_kl_32.ckpt

22.5 MB
xet

Upload joint_kl_32.ckpt with huggingface_hub 4 months ago
joint_kl_4.ckpt
Detected Pickle imports (14)
- "__builtin__.dict",
- "torch._utils._rebuild_tensor_v2",
- "torch.optim.adam.Adam",
- "omegaconf.base.Metadata",
- "typing.Any",
- "__builtin__.list",
- "collections.defaultdict",
- "omegaconf.listconfig.ListConfig",
- "omegaconf.nodes.AnyNode",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "functools.partial",
- "__builtin__.long",
- "omegaconf.base.ContainerMetadata"
How to fix it?
22.4 MB
xet

Upload joint_kl_4.ckpt with huggingface_hub 4 months ago
joint_kl_8.ckpt
Detected Pickle imports (14)
- "__builtin__.dict",
- "torch._utils._rebuild_tensor_v2",
- "torch.optim.adam.Adam",
- "omegaconf.base.Metadata",
- "typing.Any",
- "__builtin__.list",
- "collections.defaultdict",
- "omegaconf.listconfig.ListConfig",
- "omegaconf.nodes.AnyNode",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "functools.partial",
- "__builtin__.long",
- "omegaconf.base.ContainerMetadata"
How to fix it?
22.4 MB
xet

Upload joint_kl_8.ckpt with huggingface_hub 4 months ago
last.ckpt
Detected Pickle imports (33)
- "torch.nn.modules.linear.Identity",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.activation.MultiheadAttention",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.container.ModuleList",
- "src.tabasco.models.denoisers.dit.DiTBlock",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.activation.SiLU",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.dropout.Dropout",
- "torch.device",
- "torch._utils._rebuild_parameter",
- "src.tabasco.models.denoisers.dit.LabelEmbedder",
- "omegaconf.base.Metadata",
- "typing.Any",
- "collections.defaultdict",
- "src.tabasco.models.denoisers.dit.FinalLayer",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.normalization.LayerNorm",
- "src.tabasco.models.denoisers.dit.TimestepEmbedder",
- "src.tabasco.models.denoisers.dit.DiT",
- "src.tabasco.models.denoisers.dit.Mlp",
- "__builtin__.dict",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.nn.modules.sparse.Embedding",
- "src.tabasco.flow.interpolate.FlowMatchingInterpolant",
- "__builtin__.set",
- "omegaconf.nodes.AnyNode",
- "functools.partial",
- "omegaconf.dictconfig.DictConfig"
How to fix it?
5.49 GB
xet

Upload folder using huggingface_hub 4 months ago
ldm-epoch@2245-step@80856-val_qmof150_all_checks@0.6154.ckpt
Detected Pickle imports (33)
- "torch.nn.modules.linear.Identity",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.activation.MultiheadAttention",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.container.ModuleList",
- "src.tabasco.models.denoisers.dit.DiTBlock",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.activation.SiLU",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.dropout.Dropout",
- "torch.device",
- "torch._utils._rebuild_parameter",
- "src.tabasco.models.denoisers.dit.LabelEmbedder",
- "omegaconf.base.Metadata",
- "typing.Any",
- "collections.defaultdict",
- "src.tabasco.models.denoisers.dit.FinalLayer",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.normalization.LayerNorm",
- "src.tabasco.models.denoisers.dit.TimestepEmbedder",
- "src.tabasco.models.denoisers.dit.DiT",
- "src.tabasco.models.denoisers.dit.Mlp",
- "__builtin__.dict",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.nn.modules.sparse.Embedding",
- "src.tabasco.flow.interpolate.FlowMatchingInterpolant",
- "__builtin__.set",
- "omegaconf.nodes.AnyNode",
- "functools.partial",
- "omegaconf.dictconfig.DictConfig"
How to fix it?
5.49 GB
xet

Upload folder using huggingface_hub 4 months ago
ldm-epoch@29-step@137370-val_qm9_valid_rate@1.0000.ckpt
Detected Pickle imports (33)
- "src.tabasco.models.denoisers.dit.Mlp",
- "torch.nn.modules.activation.MultiheadAttention",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.sparse.Embedding",
- "src.tabasco.flow.interpolate.FlowMatchingInterpolant",
- "__builtin__.set",
- "typing.Any",
- "src.tabasco.models.denoisers.dit.FinalLayer",
- "omegaconf.base.Metadata",
- "src.tabasco.models.denoisers.dit.DiTBlock",
- "src.tabasco.models.denoisers.dit.TimestepEmbedder",
- "torch.nn.modules.activation.GELU",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "functools.partial",
- "torch.nn.modules.container.Sequential",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.linear.Identity",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.device",
- "omegaconf.dictconfig.DictConfig",
- "src.tabasco.models.denoisers.dit.DiT",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "collections.defaultdict",
- "torch.nn.modules.activation.SiLU",
- "__builtin__.dict",
- "src.tabasco.models.denoisers.dit.LabelEmbedder",
- "collections.OrderedDict",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.dropout.Dropout"
How to fix it?
1.56 GB
xet

Upload ldm-epoch@29-step@137370-val_qm9_valid_rate@1.0000.ckpt with huggingface_hub 4 months ago
ldm-epoch@3428-step@123444-val_qmof150_all_checks@0.5510.ckpt
Detected Pickle imports (33)
- "src.tabasco.models.denoisers.dit.Mlp",
- "torch.nn.modules.activation.MultiheadAttention",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.sparse.Embedding",
- "src.tabasco.flow.interpolate.FlowMatchingInterpolant",
- "__builtin__.set",
- "typing.Any",
- "src.tabasco.models.denoisers.dit.FinalLayer",
- "omegaconf.base.Metadata",
- "src.tabasco.models.denoisers.dit.DiTBlock",
- "src.tabasco.models.denoisers.dit.TimestepEmbedder",
- "torch.nn.modules.activation.GELU",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "functools.partial",
- "torch.nn.modules.container.Sequential",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.linear.Identity",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.device",
- "omegaconf.dictconfig.DictConfig",
- "src.tabasco.models.denoisers.dit.DiT",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "collections.defaultdict",
- "torch.nn.modules.activation.SiLU",
- "__builtin__.dict",
- "src.tabasco.models.denoisers.dit.LabelEmbedder",
- "collections.OrderedDict",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.dropout.Dropout"
How to fix it?
5.49 GB
xet

Upload folder using huggingface_hub 4 months ago
ldm-epoch@3459-step@124560-val_qmof150_all_checks@0.5278.ckpt
Detected Pickle imports (33)
- "torch.nn.modules.linear.Identity",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.activation.MultiheadAttention",
- "collections.OrderedDict",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.container.ModuleList",
- "src.tabasco.models.denoisers.dit.DiTBlock",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.activation.SiLU",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.dropout.Dropout",
- "torch.device",
- "torch._utils._rebuild_parameter",
- "src.tabasco.models.denoisers.dit.LabelEmbedder",
- "omegaconf.base.Metadata",
- "typing.Any",
- "collections.defaultdict",
- "src.tabasco.models.denoisers.dit.FinalLayer",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.normalization.LayerNorm",
- "src.tabasco.models.denoisers.dit.TimestepEmbedder",
- "src.tabasco.models.denoisers.dit.DiT",
- "src.tabasco.models.denoisers.dit.Mlp",
- "__builtin__.dict",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "torch.nn.modules.sparse.Embedding",
- "src.tabasco.flow.interpolate.FlowMatchingInterpolant",
- "__builtin__.set",
- "omegaconf.nodes.AnyNode",
- "functools.partial",
- "omegaconf.dictconfig.DictConfig"
How to fix it?
5.49 GB
xet

Upload folder using huggingface_hub 4 months ago
ldm-epoch@79-step@366320-val_qm9_valid_rate@0.9990.ckpt
Detected Pickle imports (33)
- "src.tabasco.models.denoisers.dit.Mlp",
- "torch.nn.modules.activation.MultiheadAttention",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.sparse.Embedding",
- "src.tabasco.flow.interpolate.FlowMatchingInterpolant",
- "__builtin__.set",
- "typing.Any",
- "src.tabasco.models.denoisers.dit.FinalLayer",
- "omegaconf.base.Metadata",
- "src.tabasco.models.denoisers.dit.DiTBlock",
- "src.tabasco.models.denoisers.dit.TimestepEmbedder",
- "torch.nn.modules.activation.GELU",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "functools.partial",
- "torch.nn.modules.container.Sequential",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.linear.Identity",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.device",
- "omegaconf.dictconfig.DictConfig",
- "src.tabasco.models.denoisers.dit.DiT",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "collections.defaultdict",
- "torch.nn.modules.activation.SiLU",
- "__builtin__.dict",
- "src.tabasco.models.denoisers.dit.LabelEmbedder",
- "collections.OrderedDict",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.dropout.Dropout"
How to fix it?
1.56 GB
xet

Upload ldm-epoch@79-step@366320-val_qm9_valid_rate@0.9990.ckpt with huggingface_hub 4 months ago
protein_diffusion.ckpt
Detected Pickle imports (33)
- "src.tabasco.models.denoisers.dit.Mlp",
- "torch.nn.modules.activation.MultiheadAttention",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.sparse.Embedding",
- "src.tabasco.flow.interpolate.FlowMatchingInterpolant",
- "__builtin__.set",
- "typing.Any",
- "src.tabasco.models.denoisers.dit.FinalLayer",
- "omegaconf.base.Metadata",
- "src.tabasco.models.denoisers.dit.DiTBlock",
- "src.tabasco.models.denoisers.dit.TimestepEmbedder",
- "torch.nn.modules.activation.GELU",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "functools.partial",
- "torch.nn.modules.container.Sequential",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.linear.Identity",
- "torch.optim.adamw.AdamW",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.device",
- "omegaconf.dictconfig.DictConfig",
- "src.tabasco.models.denoisers.dit.DiT",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear",
- "collections.defaultdict",
- "torch.nn.modules.activation.SiLU",
- "__builtin__.dict",
- "src.tabasco.models.denoisers.dit.LabelEmbedder",
- "collections.OrderedDict",
- "omegaconf.base.ContainerMetadata",
- "torch.nn.modules.dropout.Dropout"
How to fix it?
6.13 GB
xet

Upload protein_diffusion.ckpt with huggingface_hub 4 months ago
protein_kl_32.ckpt
Detected Pickle imports (46)
- "tabasco.models.components.transformer_module.TransformerModule",
- "torch.nn.modules.activation.SiLU",
- "tabasco.sample.noise_schedule.SampleNoiseSchedule",
- "torch._C._nn.gelu",
- "tabasco.models.components.attention.AttentionBlock",
- "collections.defaultdict",
- "omegaconf.base.ContainerMetadata",
- "tabasco.flow.interpolate.SDEMetricInterpolant",
- "tabasco.models.components.transformer.Transformer",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.distributions.dirichlet.Dirichlet",
- "torch.nn.modules.transformer.TransformerDecoderLayer",
- "__builtin__.long",
- "torch._utils._rebuild_parameter",
- "tabasco.models.components.positional_encoder.TimeFourierEncoding",
- "omegaconf.nodes.AnyNode",
- "torch.nn.modules.activation.MultiheadAttention",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.loss.MSELoss",
- "torch.nn.modules.loss.CrossEntropyLoss",
- "torch.nn.modules.sparse.Embedding",
- "__builtin__.list",
- "tabasco.flow.interpolate.DiscreteInterpolant",
- "torch.distributions.beta.Beta",
- "tabasco.models.components.transition.Transition",
- "omegaconf.listconfig.ListConfig",
- "torch.nn.modules.dropout.Dropout",
- "__builtin__.set",
- "tabasco.models.flow_model.FlowMatchingModel",
- "torch._utils._rebuild_tensor_v2",
- "functools.partial",
- "tabasco.flow.time_factor.InverseTimeFactor",
- "collections.OrderedDict",
- "__builtin__.dict",
- "torch.nn.functional.relu",
- "torch.Size",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.linear.Linear",
- "tabasco.models.components.attention.Attention",
- "typing.Any",
- "tabasco.models.components.positional_encoder.SinusoidEncoding",
- "tabasco.models.components.transformer.TransformerBlock",
- "torch.optim.adam.Adam",
- "torch.FloatStorage",
- "omegaconf.base.Metadata",
- "torch.nn.modules.linear.NonDynamicallyQuantizableLinear"
How to fix it?
19.3 MB
xet

Upload protein_kl_32.ckpt with huggingface_hub 4 months ago
qmof_kl_32.ckpt

20.1 MB
xet

Upload qmof_kl_32.ckpt with huggingface_hub 4 months ago

Detected Pickle imports (14)

Detected Pickle imports (46)

Detected Pickle imports (46)

Detected Pickle imports (46)

Detected Pickle imports (33)

Detected Pickle imports (14)

Detected Pickle imports (14)

Detected Pickle imports (33)

Detected Pickle imports (33)

Detected Pickle imports (33)

Detected Pickle imports (33)

Detected Pickle imports (33)

Detected Pickle imports (33)

Detected Pickle imports (33)

Detected Pickle imports (46)