File size: 400 Bytes
16de183
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import torch

def model_information(path):
    model_data = torch.load(path, map_location="cpu")

    print(f"Loaded model from {path}")

    data = model_data

    epochs = data.get("info", "None")
    sr = data.get("sr", "None")
    f0 = data.get("f0", "None")
    version = data.get("version", "None")

    return(f"Epochs: {epochs}\nSampling rate: {sr}\nPitch guidance: {f0}\nVersion: {version}")