Update self-contained hftrainer artifact with text encoder weights
Browse files- clip.safetensors +3 -0
- model_index.json +11 -2
- momask_config.json +10 -3
clip.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:431adb635d7a35aaaf5b70acf8738712c6f9ae30d08e17698c6f505c9a686fb8
|
| 3 |
+
size 353470708
|
model_index.json
CHANGED
|
@@ -10,17 +10,26 @@
|
|
| 10 |
"t2m_transformer": "t2m_trans.safetensors",
|
| 11 |
"res_transformer": "res_trans.safetensors",
|
| 12 |
"length_estimator": "length_est.safetensors",
|
|
|
|
| 13 |
"mean": "Mean.npy",
|
| 14 |
"std": "Std.npy"
|
| 15 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
"external_components": {
|
| 17 |
"clip": {
|
| 18 |
"name": "ViT-B/32",
|
| 19 |
-
"stored_in_artifact":
|
|
|
|
| 20 |
}
|
| 21 |
},
|
| 22 |
"api": {
|
| 23 |
"from_pretrained": "hftrainer.models.motion.momask.MoMaskBundle.from_pretrained",
|
| 24 |
"from_config": "hftrainer.models.motion.momask.MoMaskBundle.from_config"
|
| 25 |
}
|
| 26 |
-
}
|
|
|
|
| 10 |
"t2m_transformer": "t2m_trans.safetensors",
|
| 11 |
"res_transformer": "res_trans.safetensors",
|
| 12 |
"length_estimator": "length_est.safetensors",
|
| 13 |
+
"clip": "clip.safetensors",
|
| 14 |
"mean": "Mean.npy",
|
| 15 |
"std": "Std.npy"
|
| 16 |
},
|
| 17 |
+
"components": {
|
| 18 |
+
"clip": {
|
| 19 |
+
"name": "ViT-B/32",
|
| 20 |
+
"stored_in_artifact": true,
|
| 21 |
+
"path": "clip.safetensors"
|
| 22 |
+
}
|
| 23 |
+
},
|
| 24 |
"external_components": {
|
| 25 |
"clip": {
|
| 26 |
"name": "ViT-B/32",
|
| 27 |
+
"stored_in_artifact": true,
|
| 28 |
+
"path": "clip.safetensors"
|
| 29 |
}
|
| 30 |
},
|
| 31 |
"api": {
|
| 32 |
"from_pretrained": "hftrainer.models.motion.momask.MoMaskBundle.from_pretrained",
|
| 33 |
"from_config": "hftrainer.models.motion.momask.MoMaskBundle.from_config"
|
| 34 |
}
|
| 35 |
+
}
|
momask_config.json
CHANGED
|
@@ -1,7 +1,15 @@
|
|
| 1 |
{
|
| 2 |
"model_type": "momask",
|
|
|
|
| 3 |
"clip_version": "ViT-B/32",
|
| 4 |
"has_length_estimator": true,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
"config": {
|
| 6 |
"vq": {
|
| 7 |
"nb_code": 512,
|
|
@@ -42,6 +50,5 @@
|
|
| 42 |
"input_size": 512,
|
| 43 |
"output_size": 50
|
| 44 |
}
|
| 45 |
-
}
|
| 46 |
-
|
| 47 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
"model_type": "momask",
|
| 3 |
+
"format": "hftrainer-momask-v1",
|
| 4 |
"clip_version": "ViT-B/32",
|
| 5 |
"has_length_estimator": true,
|
| 6 |
+
"components": {
|
| 7 |
+
"clip": {
|
| 8 |
+
"name": "ViT-B/32",
|
| 9 |
+
"stored_in_artifact": true,
|
| 10 |
+
"path": "clip.safetensors"
|
| 11 |
+
}
|
| 12 |
+
},
|
| 13 |
"config": {
|
| 14 |
"vq": {
|
| 15 |
"nb_code": 512,
|
|
|
|
| 50 |
"input_size": 512,
|
| 51 |
"output_size": 50
|
| 52 |
}
|
| 53 |
+
}
|
| 54 |
+
}
|
|
|