bnewton-genmedlabs commited on
Commit
c7d0055
·
verified ·
1 Parent(s): 787073e

Add PyTorch 2.1.0 compatible models with mobile optimizations

Browse files
.gitattributes CHANGED
@@ -34,3 +34,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  fp16/xtts_infer_fp16.ts filter=lfs diff=lfs merge=lfs -text
 
 
 
 
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  fp16/xtts_infer_fp16.ts filter=lfs diff=lfs merge=lfs -text
37
+ pytorch21/xtts_mobile_fp16_aggressive.ts filter=lfs diff=lfs merge=lfs -text
38
+ pytorch21/xtts_mobile_fp16_mobile.ts filter=lfs diff=lfs merge=lfs -text
39
+ pytorch21/xtts_mobile_fp32_cpu.ts filter=lfs diff=lfs merge=lfs -text
pytorch21/manifest.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "pytorch_version": "2.8.0",
3
+ "export_type": "torchscript",
4
+ "mobile_compatible": true,
5
+ "note": "Exported with PyTorch 2.8.0 but compatible with LibTorch 2.1.0 on mobile",
6
+ "variants": {
7
+ "fp32_cpu": {
8
+ "file": "xtts_mobile_fp32_cpu.ts",
9
+ "size_mb": 11.79,
10
+ "dtype": "float32",
11
+ "optimization": "none",
12
+ "description": "Full precision CPU model"
13
+ },
14
+ "fp16_mobile": {
15
+ "file": "xtts_mobile_fp16_mobile.ts",
16
+ "size_mb": 5.89,
17
+ "dtype": "float16",
18
+ "optimization": "mobile",
19
+ "description": "Half precision with mobile optimizations",
20
+ "recommended": true
21
+ },
22
+ "fp16_aggressive": {
23
+ "file": "xtts_mobile_fp16_aggressive.ts",
24
+ "size_mb": 5.89,
25
+ "dtype": "float16",
26
+ "optimization": "mobile_aggressive",
27
+ "description": "Half precision with aggressive optimizations"
28
+ }
29
+ },
30
+ "compatibility": {
31
+ "android": {
32
+ "min_version": "LibTorch 2.1.0",
33
+ "gradle": "implementation 'org.pytorch:pytorch_android_lite:2.1.0'"
34
+ },
35
+ "ios": {
36
+ "min_version": "LibTorch 2.1.0",
37
+ "pod": "pod 'LibTorch-Lite', '~> 2.1.0'"
38
+ },
39
+ "react_native": {
40
+ "min_version": "pytorch-mobile 2.1.0",
41
+ "package": "pytorch-mobile"
42
+ }
43
+ },
44
+ "usage": {
45
+ "android": "Module module = Module.load(assetFilePath(context, 'xtts_mobile_fp16_mobile.ts'));",
46
+ "ios": "TorchModule *module = [[TorchModule alloc] initWithFileAtPath:modelPath];",
47
+ "react_native": "const model = await torch.jit._loadForMobile(modelPath);"
48
+ }
49
+ }
pytorch21/xtts_mobile_fp16_aggressive.ts ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9bce70a283e2f0f4cd958d8ae23f04e384c6eb5b97cd00f421e5e5fc1aab0a75
3
+ size 6175671
pytorch21/xtts_mobile_fp16_mobile.ts ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e8814d14b023b81147589f024a13e30e101098c9109a365bad88a746ac7716f
3
+ size 6175469
pytorch21/xtts_mobile_fp32_cpu.ts ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44c82e61749e64618417c7535ce8bdaea9f0befa18749b0fc098fb564b465519
3
+ size 12367772