guangyil commited on
Commit
bc6a876
·
verified ·
1 Parent(s): 13efdd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -36,6 +36,8 @@ spkr_model = spkr_model_init()
36
  spkr_model.model.to("cuda")
37
  spkr_model.device = "cuda"
38
  model, tokenizer, tokenizer_voila, model_type = load_model("maitrix-org/Voila-chat", "maitrix-org/Voila-Tokenizer")
 
 
39
  default_ref_file = "examples/character_ref_emb_demo.pkl"
40
  default_ref_name = "Homer Simpson"
41
  million_voice_ref_file = hf_hub_download(repo_id="maitrix-org/Voila-million-voice", filename="character_ref_emb_chunk0.pkl", repo_type="dataset")
 
36
  spkr_model.model.to("cuda")
37
  spkr_model.device = "cuda"
38
  model, tokenizer, tokenizer_voila, model_type = load_model("maitrix-org/Voila-chat", "maitrix-org/Voila-Tokenizer")
39
+ model = model.to("cuda")
40
+ tokenizer_voila.to("cuda")
41
  default_ref_file = "examples/character_ref_emb_demo.pkl"
42
  default_ref_name = "Homer Simpson"
43
  million_voice_ref_file = hf_hub_download(repo_id="maitrix-org/Voila-million-voice", filename="character_ref_emb_chunk0.pkl", repo_type="dataset")