Spaces:
Running
on
Zero
Running
on
Zero
Tune generation type for GGML
Browse files
app.py
CHANGED
@@ -44,6 +44,9 @@ def try_ggml_model(model: outetts.Models, backend: outetts.Backend, quantization
|
|
44 |
local_dir=os.path.join(helpers.get_cache_dir(), "gguf"),
|
45 |
local_files_only=False
|
46 |
)
|
|
|
|
|
|
|
47 |
return outetts.ModelConfig(
|
48 |
model_path=model_path,
|
49 |
tokenizer_path=f"OuteAI/{model.value}",
|
@@ -54,6 +57,7 @@ def try_ggml_model(model: outetts.Models, backend: outetts.Backend, quantization
|
|
54 |
dtype=None,
|
55 |
additional_model_config={},
|
56 |
audio_codec_path=None,
|
|
|
57 |
**model_config
|
58 |
)
|
59 |
|
|
|
44 |
local_dir=os.path.join(helpers.get_cache_dir(), "gguf"),
|
45 |
local_files_only=False
|
46 |
)
|
47 |
+
generation_type = outetts.GenerationType.CHUNKED
|
48 |
+
if model_config['interface_version'] == outetts.InterfaceVersion.V3:
|
49 |
+
generation_type = outetts.GenerationType.GUIDED_WORDS
|
50 |
return outetts.ModelConfig(
|
51 |
model_path=model_path,
|
52 |
tokenizer_path=f"OuteAI/{model.value}",
|
|
|
57 |
dtype=None,
|
58 |
additional_model_config={},
|
59 |
audio_codec_path=None,
|
60 |
+
generation_type=generation_type,
|
61 |
**model_config
|
62 |
)
|
63 |
|