# from tabnanny import check
import torch
import pathlib

path = r"/workspaces/vits_web_demo/export/model/D_second.pth"
model_path = pathlib.Path(path)

assert model_path.exists(), "model path does not exist"

checkpoint = torch.load(str(model_path), map_location='cpu')

state_file = checkpoint['model']
iteration = checkpoint['iteration']

out_path = model_path.parent / pathlib.Path("19_"+str(iteration)+'_demo'+'.pth')
out_path = str(out_path)
torch.save({'model': state_file,
            'iteration': iteration,
            'optimizer': None,
            'learning_rate': None}, out_path)