hans00 commited on
Commit
1098b3c
·
unverified ·
1 Parent(s): 03fd9ee

Tune generation type for GGML

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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