sparanoid commited on
Commit
e8eee51
·
unverified ·
1 Parent(s): a8d37ff

feat: use latest model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import infer_tool
9
  convert_cnt = [0]
10
  dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
  # model_name = "83_epochs.pth"
12
- model_name = "mg_1255_epochs_v0.0.6.pth"
13
  config_name = "milky_green.json"
14
  net_g_ms, hubert_soft, feature_input, hps_ms = infer_tool.load_model(f"{model_name}", f"configs/{config_name}")
15
 
 
9
  convert_cnt = [0]
10
  dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
  # model_name = "83_epochs.pth"
12
+ model_name = "mg_1324_epochs_v0.0.8.pth"
13
  config_name = "milky_green.json"
14
  net_g_ms, hubert_soft, feature_input, hps_ms = infer_tool.load_model(f"{model_name}", f"configs/{config_name}")
15