Downloads · 30 days
10
24% of all-time downloads
clip_vit_emb_neuronx
optimum/clip_vit_emb_neuronx
clip_vit_emb_neuronx is a machine learning model from optimum. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
All-time downloads
42
Public
Repo size
625 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.neuron625 MB · 99%
At a glance
- License
- apache-2.0
- Model type
- clip
- Access
- Public
- Created
- Apr 7, 2025
- Updated
- Apr 7, 2025
- SHA
- c94b95cf
Model card
From the Hugging Face model README
INFERENTIA ONLY
Exported with:
from optimum.neuron import NeuronModelForSentenceTransformers
# [Compile]
model_id = "sentence-transformers/clip-ViT-B-32"
# configs for compiling model
input_shapes = {
"num_channels": 3,
"height": 224,
"width": 224,
"text_batch_size": 3,
"image_batch_size": 1,
"sequence_length": 64,
}
neuron_model = NeuronModelForSentenceTransformers.from_pretrained(
model_id, subfolder="0_CLIPModel", export=True, dynamic_batch_size=False, **input_shapes
)
# Save locally or upload to the HuggingFace Hub
save_directory = "clip_emb_neuron/"
neuron_model.save_pretrained(save_directory)
# Upload to the HuggingFace Hub
neuron_model.push_to_hub(
"clip_emb_neuron/", repository_id="optimum/clip_vit_emb_neuronx" # Replace with your HF Hub repo id
)
- Type
- clip
- License
- apache-2.0
- Created
- Apr 7, 2025
- Updated
- Apr 7, 2025