Hugging Face
Models
Datasets
Spaces
Buckets
new
Docs
Enterprise
Pricing
Website
Tasks
HuggingChat
Collections
Languages
Organizations
Community
Blog
Posts
Daily Papers
Learn
Discord
Forum
GitHub
Solutions
Team & Enterprise
Hugging Face PRO
Enterprise Support
Inference Providers
Inference Endpoints
Storage Buckets
Log In
Sign Up
YuhaiW
/
prism-jepa-red-cube-arx
like
0
Robotics
PyTorch
world-model
jepa
lewm
prism
arx
visuomotor
License:
apache-2.0
Model card
Files
Files and versions
xet
Community
Copy to bucket
new
main
prism-jepa-red-cube-arx
74.8 MB
Ctrl+K
Ctrl+K
1 contributor
History:
2 commits
YuhaiW
red_cube PRISM-JEPA deploy bundle: LeWM WM + PRISM prior + self-contained inference + README
8f017a6
verified
30 days ago
.gitattributes
Safe
1.52 kB
initial commit
30 days ago
README.md
5.12 kB
red_cube PRISM-JEPA deploy bundle: LeWM WM + PRISM prior + self-contained inference + README
30 days ago
arx_inference_demo.py
15 kB
red_cube PRISM-JEPA deploy bundle: LeWM WM + PRISM prior + self-contained inference + README
30 days ago
jepa.py
Safe
5.4 kB
red_cube PRISM-JEPA deploy bundle: LeWM WM + PRISM prior + self-contained inference + README
30 days ago
lewm_red_cube_epoch_100_object.ckpt
pickle
Detected Pickle imports (39)
"torch.nn.modules.activation.SiLU"
,
"torch.FloatStorage"
,
"transformers.models.vit.modeling_vit.ViTEmbeddings"
,
"transformers.models.vit.modeling_vit.ViTSelfAttention"
,
"torch.nn.modules.linear.Identity"
,
"transformers.models.vit.modeling_vit.ViTEncoder"
,
"torch.nn.modules.dropout.Dropout"
,
"torch._utils._rebuild_parameter"
,
"collections.OrderedDict"
,
"torch.nn.modules.normalization.LayerNorm"
,
"transformers.models.vit.modeling_vit.ViTPatchEmbeddings"
,
"torch.nn.modules.container.ModuleList"
,
"transformers.models.vit.modeling_vit.ViTAttention"
,
"transformers.models.vit.modeling_vit.ViTOutput"
,
"transformers.models.vit.modeling_vit.ViTModel"
,
"torch._C._nn.gelu"
,
"__builtin__.set"
,
"module.ARPredictor"
,
"torch.LongStorage"
,
"module.Transformer"
,
"torch.nn.modules.batchnorm.BatchNorm1d"
,
"transformers.models.vit.modeling_vit.ViTIntermediate"
,
"torch.nn.modules.activation.GELU"
,
"module.ConditionalBlock"
,
"module.MLP"
,
"module.FeedForward"
,
"torch.nn.modules.conv.Conv2d"
,
"module.Attention"
,
"module.Embedder"
,
"transformers.models.vit.modeling_vit.ViTSelfOutput"
,
"torch.nn.modules.conv.Conv1d"
,
"jepa.JEPA"
,
"torch.nn.modules.linear.Linear"
,
"transformers.models.vit.modeling_vit.ViTLayer"
,
"torch.nn.modules.activation.Softmax"
,
"transformers.models.vit.configuration_vit.ViTConfig"
,
"torch.nn.modules.container.Sequential"
,
"transformers.activations.GELUActivation"
,
"torch._utils._rebuild_tensor_v2"
How to fix it?
72.4 MB
xet
red_cube PRISM-JEPA deploy bundle: LeWM WM + PRISM prior + self-contained inference + README
30 days ago
module.py
Safe
9.18 kB
red_cube PRISM-JEPA deploy bundle: LeWM WM + PRISM prior + self-contained inference + README
30 days ago
prior_head.py
Safe
2.42 kB
red_cube PRISM-JEPA deploy bundle: LeWM WM + PRISM prior + self-contained inference + README
30 days ago
prior_head_red_cube.pt
pickle
Detected Pickle imports (3)
"collections.OrderedDict"
,
"torch._utils._rebuild_tensor_v2"
,
"torch.FloatStorage"
What is a pickle import?
2.36 MB
xet
red_cube PRISM-JEPA deploy bundle: LeWM WM + PRISM prior + self-contained inference + README
30 days ago