Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,8 @@ def spkr_model_init():
|
|
| 33 |
return spkr_model
|
| 34 |
|
| 35 |
spkr_model = spkr_model_init()
|
| 36 |
-
spkr_model.to("cuda")
|
|
|
|
| 37 |
model, tokenizer, tokenizer_voila, model_type = load_model("maitrix-org/Voila-chat", "maitrix-org/Voila-Tokenizer")
|
| 38 |
default_ref_file = "examples/character_ref_emb_demo.pkl"
|
| 39 |
default_ref_name = "Homer Simpson"
|
|
|
|
| 33 |
return spkr_model
|
| 34 |
|
| 35 |
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"
|