Pushkar Patel
commited on
use float16 instead of bfloat16 as we are inferencing on cpu
Browse files
README.md
CHANGED
|
@@ -1,11 +1,11 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
language:
|
| 4 |
-
- en
|
| 5 |
tags:
|
| 6 |
-
- text-to-speech
|
| 7 |
base_model:
|
| 8 |
-
- sesame/csm-1b
|
| 9 |
pipeline_tag: text-to-speech
|
| 10 |
---
|
| 11 |
|
|
@@ -75,7 +75,7 @@ model_args = ModelArgs(
|
|
| 75 |
audio_vocab_size=2051,
|
| 76 |
audio_num_codebooks=32,
|
| 77 |
)
|
| 78 |
-
model = Model(model_args).to(device=device, dtype=torch.
|
| 79 |
loaded = load_file(model_path)
|
| 80 |
model.load_state_dict(loaded)
|
| 81 |
|
|
@@ -155,4 +155,4 @@ By using this model, you agree to comply with all applicable laws and ethical gu
|
|
| 155 |
---
|
| 156 |
|
| 157 |
## Authors
|
| 158 |
-
Johan Schalkwyk, Ankit Kumar, Dan Lyth, Sefik Emre Eskimez, Zack Hodari, Cinjon Resnick, Ramon Sanabria, Raven Jiang, and the Sesame team.
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
language:
|
| 4 |
+
- en
|
| 5 |
tags:
|
| 6 |
+
- text-to-speech
|
| 7 |
base_model:
|
| 8 |
+
- sesame/csm-1b
|
| 9 |
pipeline_tag: text-to-speech
|
| 10 |
---
|
| 11 |
|
|
|
|
| 75 |
audio_vocab_size=2051,
|
| 76 |
audio_num_codebooks=32,
|
| 77 |
)
|
| 78 |
+
model = Model(model_args).to(device=device, dtype=torch.float16)
|
| 79 |
loaded = load_file(model_path)
|
| 80 |
model.load_state_dict(loaded)
|
| 81 |
|
|
|
|
| 155 |
---
|
| 156 |
|
| 157 |
## Authors
|
| 158 |
+
Johan Schalkwyk, Ankit Kumar, Dan Lyth, Sefik Emre Eskimez, Zack Hodari, Cinjon Resnick, Ramon Sanabria, Raven Jiang, and the Sesame team.
|