Siratish commited on
Commit
e6c6994
·
1 Parent(s): d4c21bd
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -76,14 +76,14 @@ cache_schedule = {
76
  'ff': presets[default_preset]['ff'][:]
77
  }
78
 
79
- config = tomli.load(open(os.path.join(files("f5_tts").joinpath("infer/examples/basic"), "basic.toml"), "rb"))
80
 
81
  model = config.get("model", "F5TTS_v1_Base")
82
  ckpt_file = config.get("ckpt_file", "")
83
  vocab_file = config.get("vocab_file", "")
84
 
85
  model_cfg = OmegaConf.load(
86
- config.get("model_cfg", str(files("f5_tts").joinpath(f"configs/{model}.yaml")))
87
  )
88
  model_cls = get_class(f"f5_tts.model.{model_cfg.model.backbone}")
89
  model_arc = model_cfg.model.arch
 
76
  'ff': presets[default_preset]['ff'][:]
77
  }
78
 
79
+ config = tomli.load(open(os.path.join("f5_tts/infer/examples/basic", "basic.toml"), "rb"))
80
 
81
  model = config.get("model", "F5TTS_v1_Base")
82
  ckpt_file = config.get("ckpt_file", "")
83
  vocab_file = config.get("vocab_file", "")
84
 
85
  model_cfg = OmegaConf.load(
86
+ config.get("model_cfg", f"f5_tts/configs/{model}.yaml")
87
  )
88
  model_cls = get_class(f"f5_tts.model.{model_cfg.model.backbone}")
89
  model_arc = model_cfg.model.arch