| import torch |
| from datetime import datetime |
|
|
|
|
| def prettify_date(date_str): |
| date_time_obj = datetime.strptime(date_str, "%Y-%m-%dT%H:%M:%S.%f") |
| return date_time_obj.strftime("%Y-%m-%d %H:%M:%S") |
|
|
|
|
| def model_information(path): |
| model_data = torch.load(path, map_location="cpu") |
|
|
| print(f"Loaded model from {path}") |
|
|
| epochs = model_data.get("epoch", "None") |
| steps = model_data.get("step", "None") |
| sr = model_data.get("sr", "None") |
| f0 = model_data.get("f0", "None") |
| version = model_data.get("version", "None") |
| creation_date = model_data.get("creation_date", "None") |
| model_hash = model_data.get("model_hash", "None") |
|
|
| pitch_guidance = "True" if f0 == 1 else "False" |
|
|
| return ( |
| f"Epochs: {epochs}\n" |
| f"Steps: {steps}\n" |
| f"RVC Version: {version}\n" |
| f"Sampling Rate: {sr}\n" |
| f"Pitch Guidance: {pitch_guidance}\n" |
| f"Creation Date: {prettify_date(creation_date)}\n" |
| f"Hash (ID): {model_hash}" |
| ) |
|
|
