Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|