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.
|