necknet

from transformers import AutoModel

model = AutoModel.from_pretrained("han-xudong/necknet", trust_remote_code=True)
x = torch.zeros((1, 6))  # Example input: batch size of 1, 6D motion
output = model(x)

# Example code to load onnx
import onnxruntime as ort
import numpy as np
from huggingface_hub import hf_hub_download

onnx_model_path = hf_hub_download("han-xudong/necknet", filename="model.onnx")
ort_session = ort.InferenceSession(onnx_model_path)

# Example input
x = np.zeros((1, 6), dtype=np.float32)  # Batch size of 1, 6D motion
output = ort_session.run(None, {"motion": x})