Feature Extraction
Transformers
Safetensors
AuriStream
audio
speech
language-model
auristream
custom_code
Instructions to use TuKoResearch/AuriStream100M_20Pred_BigAudioDataset_500k-randinit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use TuKoResearch/AuriStream100M_20Pred_BigAudioDataset_500k-randinit with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="TuKoResearch/AuriStream100M_20Pred_BigAudioDataset_500k-randinit", trust_remote_code=True)# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("TuKoResearch/AuriStream100M_20Pred_BigAudioDataset_500k-randinit", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload AuriStream checkpoint
Browse files- README.md +1 -1
- config.json +5 -3
- model.safetensors +2 -2
README.md
CHANGED
|
@@ -14,7 +14,7 @@ library_name: transformers
|
|
| 14 |
|
| 15 |
This model predicts cochlear tokens from a tokenizer such as [WavCochCausalV8192](https://huggingface.co/TuKoResearch/WavCochCausalV8192).
|
| 16 |
|
| 17 |
-
|
| 18 |
|
| 19 |
## Model Details
|
| 20 |
|
|
|
|
| 14 |
|
| 15 |
This model predicts cochlear tokens from a tokenizer such as [WavCochCausalV8192](https://huggingface.co/TuKoResearch/WavCochCausalV8192).
|
| 16 |
|
| 17 |
+
Native training step-zero initialization for the 100M 20-prediction model. This exactly uses origin seed `11101994` and historical source commit `fad735c90bd44b4bcfd6bdf9429edbb640d2ebf5`, matching the start of [W&B run 3g8jxx2k](https://wandb.ai/NaudioGPT/auristream/runs/3g8jxx2k). The weights are untrained FP32 values produced before XLA/FSDP wrapping.
|
| 18 |
|
| 19 |
## Model Details
|
| 20 |
|
config.json
CHANGED
|
@@ -8,7 +8,7 @@
|
|
| 8 |
"AutoModel": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel",
|
| 9 |
"AutoModelForCausalLM": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel"
|
| 10 |
},
|
| 11 |
-
"torch_dtype": "
|
| 12 |
"transformers_version": "4.40.0",
|
| 13 |
"vocab_size": 8192,
|
| 14 |
"n_embd": 768,
|
|
@@ -18,6 +18,8 @@
|
|
| 18 |
"dropout": 0.0,
|
| 19 |
"bias": false,
|
| 20 |
"input_conv_kernel_size": 0,
|
| 21 |
-
"random_init_seed":
|
| 22 |
-
"random_init_source": "
|
|
|
|
|
|
|
| 23 |
}
|
|
|
|
| 8 |
"AutoModel": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel",
|
| 9 |
"AutoModelForCausalLM": "TuKoResearch/AuriStream-base--modeling_auristream.AuriStreamModel"
|
| 10 |
},
|
| 11 |
+
"torch_dtype": "float32",
|
| 12 |
"transformers_version": "4.40.0",
|
| 13 |
"vocab_size": 8192,
|
| 14 |
"n_embd": 768,
|
|
|
|
| 18 |
"dropout": 0.0,
|
| 19 |
"bias": false,
|
| 20 |
"input_conv_kernel_size": 0,
|
| 21 |
+
"random_init_seed": 11101994,
|
| 22 |
+
"random_init_source": "native-training-step-zero",
|
| 23 |
+
"random_init_source_commit": "fad735c90bd44b4bcfd6bdf9429edbb640d2ebf5",
|
| 24 |
+
"random_init_checkpoint_iteration": 0
|
| 25 |
}
|
model.safetensors
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:d2d67219043868773f06dfcb001a680333e05260a05e0c8929ac33ca73f711e1
|
| 3 |
+
size 868309144
|