revert: upload unmodified spacekaren/chatterbox-turbo-webgpu models (our INT64 fix was breaking them)
Browse files- .gitattributes +3 -0
- README.md +52 -0
- onnx/conditional_decoder_q4f16.onnx +2 -2
- onnx/embed_tokens_q4f16.onnx +2 -2
- onnx/language_model_q4f16.onnx +2 -2
- onnx/speech_encoder_q4f16.onnx +2 -2
.gitattributes
CHANGED
|
@@ -37,3 +37,6 @@ onnx/conditional_decoder_q4f16.onnx_data filter=lfs diff=lfs merge=lfs -text
|
|
| 37 |
onnx/embed_tokens_q4f16.onnx_data filter=lfs diff=lfs merge=lfs -text
|
| 38 |
onnx/language_model_q4f16.onnx_data filter=lfs diff=lfs merge=lfs -text
|
| 39 |
onnx/speech_encoder_q4f16.onnx_data filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
onnx/embed_tokens_q4f16.onnx_data filter=lfs diff=lfs merge=lfs -text
|
| 38 |
onnx/language_model_q4f16.onnx_data filter=lfs diff=lfs merge=lfs -text
|
| 39 |
onnx/speech_encoder_q4f16.onnx_data filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
onnx/conditional_decoder.onnx_data filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
onnx/embed_tokens.onnx_data filter=lfs diff=lfs merge=lfs -text
|
| 42 |
+
onnx/speech_encoder.onnx_data filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
base_model: ResembleAI/chatterbox-turbo-ONNX
|
| 4 |
+
tags:
|
| 5 |
+
- text-to-speech
|
| 6 |
+
- tts
|
| 7 |
+
- onnx
|
| 8 |
+
- webgpu
|
| 9 |
+
- transformers.js
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# Chatterbox Turbo - WebGPU Compatible
|
| 13 |
+
|
| 14 |
+
This is a WebGPU-compatible version of [ResembleAI/chatterbox-turbo-ONNX](https://huggingface.co/ResembleAI/chatterbox-turbo-ONNX).
|
| 15 |
+
|
| 16 |
+
## Changes from Original
|
| 17 |
+
|
| 18 |
+
The original model contains `int64` Cast operations and tensors that WebGPU cannot execute.
|
| 19 |
+
This version converts all `int64` operations to `int32`, enabling direct WebGPU inference.
|
| 20 |
+
|
| 21 |
+
### Modifications Made:
|
| 22 |
+
- **conditional_decoder**: 521 Cast nodes inserted (376 Shape/Range ops)
|
| 23 |
+
- **speech_encoder**: 350 Cast nodes inserted (243 Shape/Range ops)
|
| 24 |
+
- **language_model**: 3 Cast nodes inserted
|
| 25 |
+
- **embed_tokens**: 1 Cast node inserted
|
| 26 |
+
|
| 27 |
+
## Usage with Transformers.js
|
| 28 |
+
|
| 29 |
+
```javascript
|
| 30 |
+
import { AutoModel, AutoProcessor } from '@huggingface/transformers';
|
| 31 |
+
|
| 32 |
+
const model = await AutoModel.from_pretrained('spacekaren/chatterbox-turbo-webgpu', {
|
| 33 |
+
device: 'webgpu',
|
| 34 |
+
dtype: 'q4f16',
|
| 35 |
+
});
|
| 36 |
+
|
| 37 |
+
const processor = await AutoProcessor.from_pretrained('spacekaren/chatterbox-turbo-webgpu');
|
| 38 |
+
```
|
| 39 |
+
|
| 40 |
+
## Model Size
|
| 41 |
+
|
| 42 |
+
- **Total**: ~539 MB (q4f16 quantization)
|
| 43 |
+
- Same architecture as original, just int64→int32 conversion
|
| 44 |
+
|
| 45 |
+
## License
|
| 46 |
+
|
| 47 |
+
MIT (same as original)
|
| 48 |
+
|
| 49 |
+
## Credits
|
| 50 |
+
|
| 51 |
+
- Original model: [ResembleAI/chatterbox-turbo-ONNX](https://huggingface.co/ResembleAI/chatterbox-turbo-ONNX)
|
| 52 |
+
- Conversion script: [local.core/scripts/convert_int64_to_int32.py](https://github.com/anthropics/lama)
|
onnx/conditional_decoder_q4f16.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1a0d7f7c574dc40d6f5649229e8203bc181b75f119a4c80e97d2ca2defd53593
|
| 3 |
+
size 2398356
|
onnx/embed_tokens_q4f16.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d59c34a799c5dc5d5220bb1b70d1bd5e9b689db7340cd9e4a73b8a8c904b841a
|
| 3 |
+
size 2524
|
onnx/language_model_q4f16.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fd3f7756161d6e03f2b10212170c8f4364f568cf21ac660eead8121a09690b8a
|
| 3 |
+
size 276817
|
onnx/speech_encoder_q4f16.onnx
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:57986f2f254d415edc627b469dc67f5c09016e2873df15db7cda1579005aec9f
|
| 3 |
+
size 1220677
|