Views
No views yet
| Model | Description | Size |
|---|---|---|
skeleton.safetensors | Skeleton prediction model (articulation-xl_quantization_256) | ~1.4 GB |
skin.safetensors | Skinning weights model (articulation-xl) | ~4.1 GB |
1from safetensors.torch import load_file
2
3# Load skeleton model
4skeleton_weights = load_file("skeleton.safetensors")
5
6# Load skin model
7skin_weights = load_file("skin.safetensors").ckpt loading code:1# Original (ckpt)
2# checkpoint = torch.load("model.ckpt")
3# state_dict = checkpoint['state_dict']
4
5# With safetensors
6from safetensors.torch import load_file
7state_dict = load_file("model.safetensors")1import torch
2from safetensors.torch import save_file
3
4checkpoint = torch.load("model.ckpt", map_location='cpu', weights_only=False)
5state_dict = checkpoint['state_dict']
6save_file(state_dict, "model.safetensors")1@article{unirig2024,
2 title={UniRig: A Unified Framework for 3D Character Rigging},
3 author={VAST AI Research},
4 journal={arXiv preprint arXiv:2405.02986},
5 year={2024}
6}