Pushkar Patel commited on
Commit
53bc958
·
unverified ·
1 Parent(s): 0348cb7

use float16 instead of bfloat16 as we are inferencing on cpu

Browse files
Files changed (1) hide show
  1. README.md +5 -5
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.bfloat16)
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.