1.07 GB

Ctrl+K

1 contributor

History: 1 commit

bmrayan

Add diff_1h_eval checkpoints

527ad8f verified about 1 month ago

config.yaml

1.76 kB
Add diff_1h_eval checkpoints about 1 month ago
lewm_diff_epoch_1_object.ckpt
Detected Pickle imports (43)
- "le_wm.models.diffusion_predictor._Attention",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "torch.nn.modules.activation.GELU",
- "module.Transformer",
- "module.ConditionalBlock",
- "module.FeedForward",
- "torch.nn.modules.conv.Conv1d",
- "torch.nn.modules.activation.Softmax",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "le_wm.models.diffusion_predictor._FeedForward",
- "torch.nn.modules.activation.SiLU",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "transformers.activations.GELUActivation",
- "module.MLP",
- "transformers.models.vit.configuration_vit.ViTConfig",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "torch._utils._rebuild_parameter",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "collections.OrderedDict",
- "__builtin__.set",
- "torch._utils._rebuild_tensor_v2",
- "torch.FloatStorage",
- "module.Attention",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "transformers.models.vit.modeling_vit.ViTModel",
- "torch.nn.modules.linear.Identity",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "torch.nn.modules.dropout.Dropout",
- "module.Embedder",
- "module.ARPredictor",
- "le_wm.models.diffusion_predictor.DiffusionPredictor",
- "torch.nn.modules.batchnorm.BatchNorm1d",
- "le_wm.models.diffusion_predictor._ConditionalBlock",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.container.Sequential",
- "jepa.JEPA",
- "torch.nn.modules.linear.Linear",
- "torch.nn.modules.conv.Conv2d",
- "torch.nn.modules.container.ModuleList",
- "torch._C._nn.gelu",
- "torch.LongStorage",
- "transformers.models.vit.modeling_vit.ViTLayer"
How to fix it?
117 MB
xet

Add diff_1h_eval checkpoints about 1 month ago
lewm_diff_epoch_2_object.ckpt
Detected Pickle imports (43)
- "torch.nn.modules.conv.Conv2d",
- "torch.nn.modules.activation.Softmax",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "module.ARPredictor",
- "torch.nn.modules.conv.Conv1d",
- "collections.OrderedDict",
- "torch.nn.modules.activation.SiLU",
- "module.ConditionalBlock",
- "__builtin__.set",
- "torch._utils._rebuild_parameter",
- "torch.LongStorage",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "torch.nn.modules.container.Sequential",
- "le_wm.models.diffusion_predictor._ConditionalBlock",
- "torch._C._nn.gelu",
- "torch.nn.modules.linear.Identity",
- "jepa.JEPA",
- "module.Transformer",
- "torch.nn.modules.dropout.Dropout",
- "le_wm.models.diffusion_predictor._Attention",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "module.Attention",
- "transformers.models.vit.configuration_vit.ViTConfig",
- "module.MLP",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.normalization.LayerNorm",
- "le_wm.models.diffusion_predictor.DiffusionPredictor",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.vit.modeling_vit.ViTModel",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "torch.nn.modules.linear.Linear",
- "module.FeedForward",
- "torch.nn.modules.batchnorm.BatchNorm1d",
- "module.Embedder",
- "transformers.activations.GELUActivation",
- "le_wm.models.diffusion_predictor._FeedForward"
How to fix it?
117 MB
xet

Add diff_1h_eval checkpoints about 1 month ago
lewm_diff_epoch_3_object.ckpt
Detected Pickle imports (43)
- "transformers.models.vit.configuration_vit.ViTConfig",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "torch.nn.modules.batchnorm.BatchNorm1d",
- "torch.nn.modules.conv.Conv1d",
- "module.Transformer",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "torch.LongStorage",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.models.vit.modeling_vit.ViTModel",
- "torch.nn.modules.activation.Softmax",
- "torch._utils._rebuild_tensor_v2",
- "module.ARPredictor",
- "torch.nn.modules.container.ModuleList",
- "le_wm.models.diffusion_predictor.DiffusionPredictor",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "module.Embedder",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "le_wm.models.diffusion_predictor._FeedForward",
- "collections.OrderedDict",
- "le_wm.models.diffusion_predictor._ConditionalBlock",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.linear.Identity",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "le_wm.models.diffusion_predictor._Attention",
- "torch._utils._rebuild_parameter",
- "module.MLP",
- "torch.nn.modules.dropout.Dropout",
- "torch._C._nn.gelu",
- "module.Attention",
- "jepa.JEPA",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "torch.nn.modules.linear.Linear",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "module.ConditionalBlock",
- "__builtin__.set",
- "transformers.activations.GELUActivation",
- "module.FeedForward",
- "torch.nn.modules.conv.Conv2d"
How to fix it?
117 MB
xet

Add diff_1h_eval checkpoints about 1 month ago
lewm_diff_epoch_4_object.ckpt
Detected Pickle imports (43)
- "transformers.models.vit.configuration_vit.ViTConfig",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "torch.nn.modules.batchnorm.BatchNorm1d",
- "torch.nn.modules.conv.Conv1d",
- "module.Transformer",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "torch.LongStorage",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.models.vit.modeling_vit.ViTModel",
- "torch.nn.modules.activation.Softmax",
- "torch._utils._rebuild_tensor_v2",
- "module.ARPredictor",
- "torch.nn.modules.container.ModuleList",
- "le_wm.models.diffusion_predictor.DiffusionPredictor",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "module.Embedder",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "le_wm.models.diffusion_predictor._FeedForward",
- "collections.OrderedDict",
- "le_wm.models.diffusion_predictor._ConditionalBlock",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.linear.Identity",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "le_wm.models.diffusion_predictor._Attention",
- "torch._utils._rebuild_parameter",
- "module.MLP",
- "torch.nn.modules.dropout.Dropout",
- "torch._C._nn.gelu",
- "module.Attention",
- "jepa.JEPA",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "torch.nn.modules.linear.Linear",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "module.ConditionalBlock",
- "__builtin__.set",
- "transformers.activations.GELUActivation",
- "module.FeedForward",
- "torch.nn.modules.conv.Conv2d"
How to fix it?
117 MB
xet

Add diff_1h_eval checkpoints about 1 month ago
lewm_diff_epoch_5_object.ckpt
Detected Pickle imports (43)
- "transformers.models.vit.configuration_vit.ViTConfig",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "torch.nn.modules.batchnorm.BatchNorm1d",
- "torch.nn.modules.conv.Conv1d",
- "module.Transformer",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "torch.LongStorage",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.models.vit.modeling_vit.ViTModel",
- "torch.nn.modules.activation.Softmax",
- "torch._utils._rebuild_tensor_v2",
- "module.ARPredictor",
- "torch.nn.modules.container.ModuleList",
- "le_wm.models.diffusion_predictor.DiffusionPredictor",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "module.Embedder",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "le_wm.models.diffusion_predictor._FeedForward",
- "collections.OrderedDict",
- "le_wm.models.diffusion_predictor._ConditionalBlock",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.linear.Identity",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "le_wm.models.diffusion_predictor._Attention",
- "torch._utils._rebuild_parameter",
- "module.MLP",
- "torch.nn.modules.dropout.Dropout",
- "torch._C._nn.gelu",
- "module.Attention",
- "jepa.JEPA",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "torch.nn.modules.linear.Linear",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "module.ConditionalBlock",
- "__builtin__.set",
- "transformers.activations.GELUActivation",
- "module.FeedForward",
- "torch.nn.modules.conv.Conv2d"
How to fix it?
117 MB
xet

Add diff_1h_eval checkpoints about 1 month ago
lewm_diff_epoch_6_object.ckpt
Detected Pickle imports (43)
- "transformers.models.vit.configuration_vit.ViTConfig",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "torch.nn.modules.batchnorm.BatchNorm1d",
- "torch.nn.modules.conv.Conv1d",
- "module.Transformer",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "torch.LongStorage",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.models.vit.modeling_vit.ViTModel",
- "torch.nn.modules.activation.Softmax",
- "torch._utils._rebuild_tensor_v2",
- "module.ARPredictor",
- "torch.nn.modules.container.ModuleList",
- "le_wm.models.diffusion_predictor.DiffusionPredictor",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "torch.nn.modules.activation.SiLU",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "module.Embedder",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "le_wm.models.diffusion_predictor._FeedForward",
- "collections.OrderedDict",
- "le_wm.models.diffusion_predictor._ConditionalBlock",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.linear.Identity",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "le_wm.models.diffusion_predictor._Attention",
- "torch._utils._rebuild_parameter",
- "module.MLP",
- "torch.nn.modules.dropout.Dropout",
- "torch._C._nn.gelu",
- "module.Attention",
- "jepa.JEPA",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "torch.nn.modules.linear.Linear",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "module.ConditionalBlock",
- "__builtin__.set",
- "transformers.activations.GELUActivation",
- "module.FeedForward",
- "torch.nn.modules.conv.Conv2d"
How to fix it?
117 MB
xet

Add diff_1h_eval checkpoints about 1 month ago
lewm_diff_epoch_7_object.ckpt
Detected Pickle imports (43)
- "torch.nn.modules.conv.Conv2d",
- "torch.nn.modules.activation.Softmax",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "module.ARPredictor",
- "torch.nn.modules.conv.Conv1d",
- "collections.OrderedDict",
- "torch.nn.modules.activation.SiLU",
- "module.ConditionalBlock",
- "__builtin__.set",
- "torch._utils._rebuild_parameter",
- "torch.LongStorage",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "torch.nn.modules.container.Sequential",
- "le_wm.models.diffusion_predictor._ConditionalBlock",
- "torch._C._nn.gelu",
- "torch.nn.modules.linear.Identity",
- "jepa.JEPA",
- "module.Transformer",
- "torch.nn.modules.dropout.Dropout",
- "le_wm.models.diffusion_predictor._Attention",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "module.Attention",
- "transformers.models.vit.configuration_vit.ViTConfig",
- "module.MLP",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.normalization.LayerNorm",
- "le_wm.models.diffusion_predictor.DiffusionPredictor",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.vit.modeling_vit.ViTModel",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "torch.nn.modules.linear.Linear",
- "module.FeedForward",
- "torch.nn.modules.batchnorm.BatchNorm1d",
- "module.Embedder",
- "transformers.activations.GELUActivation",
- "le_wm.models.diffusion_predictor._FeedForward"
How to fix it?
117 MB
xet

Add diff_1h_eval checkpoints about 1 month ago
lewm_diff_weights.ckpt
Detected Pickle imports (4)
- "collections.OrderedDict",
- "torch.LongStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.FloatStorage"
What is a pickle import?
257 MB
xet

Add diff_1h_eval checkpoints about 1 month ago

Detected Pickle imports (43)

Detected Pickle imports (43)

Detected Pickle imports (43)

Detected Pickle imports (43)

Detected Pickle imports (43)

Detected Pickle imports (43)

Detected Pickle imports (43)

Detected Pickle imports (4)